Java:如何在编译时找到方法的调用者?

Question

我有一个Java问题，我找不到答案:我有一个类a 包含一个方法a()，应该只从类扩展类B调用。目前，我能找到的最佳解决方案如下:

Class callerClass = Reflection.getCallerClass(2);
if (!B.class.isAssignableFrom(callerClass))
    throw new InvalidMethodCallException("A.a", B.class.getName(), callerClass.getName());

InvalidMethodCallException是我自己的例外。

然而，我对我的解决方案并不完全满意，因为调用是在运行时进行的。相反，我想在 a() 中声明只有 B 可以访问它，但要在编译时检查这一点，因为反射是一个昂贵的过程。

你知道正确的方法吗?

谢谢你的帮助。

编辑:附加信息

一个更简单的例子是，我不希望用户在Edge类中调用方法connect(port1, port2)，而是从Graph类中调用方法addEdge(edge, port1, port2)，因为图还必须记住它内部有哪些边。

这是我的代码从类Graph:

        public void addEdge (Edge edge, Port firstPort, Port secondPort)
            throws DuplicateEdgeInGraphException, AttachedNodeNotInGraphException, GraphDescriptionModelException
        {
            // Must not appear in the graph
            if (this.edges.contains(edge))
                throw new DuplicateEdgeInGraphException(edge);
            // The nodes attached to the ports must be in the graph
            if (!this.nodes.contains(firstPort.getNode()))
                throw new AttachedNodeNotInGraphException(firstPort);
            if (!this.nodes.contains(secondPort.getNode()))
                throw new AttachedNodeNotInGraphException(secondPort);
            // We connect the edge and add it
            edge.connect(firstPort, secondPort);
            this.edges.add(edge);
        }

类Edge:

        public void connect (Port portOne, Port portTwo)
            throws InvalidMethodCallException, WrongPortsDirectionsException, WrongPortsSizesException, GraphDescriptionModelException
        {
            // We check the reserved method call
            Class callerClass = Reflection.getCallerClass(2);
            if (!Graph.class.isAssignableFrom(callerClass))
                throw new InvalidMethodCallException("Edge.connect", Graph.class.getName(), callerClass.getName());
            // We check the directions of the ports
            if ((portOne.isInputPort() && !portOne.isInputOutputPort() && !portTwo.isOutputPort())
            || (portTwo.isInputPort() && !portTwo.isInputOutputPort() && !portOne.isOutputPort())
            || (portOne.isOutputPort() && !portOne.isInputOutputPort() && !portTwo.isInputPort())
            || (portTwo.isOutputPort() && !portTwo.isInputOutputPort() && !portOne.isInputPort()))
                throw new WrongPortsDirectionsException(portOne, portTwo);
            // We check the sizes of the ports
            if ((portOne.isInputPort() && !portOne.isInputOutputPort() && portOne.getSize() < portTwo.getSize())
            || (portTwo.isInputPort() && !portTwo.isInputOutputPort() && portTwo.getSize() < portOne.getSize())
            || (portOne.isOutputPort() && !portOne.isInputOutputPort() && portOne.getSize() > portTwo.getSize())
            || (portTwo.isOutputPort() && !portTwo.isInputOutputPort() && portTwo.getSize() > portOne.getSize())
            || (portOne.isInputOutputPort() && portTwo.isInputOutputPort() && portOne.getSize() != portTwo.getSize()))
                throw new WrongPortsSizesException(portOne, portTwo);
            // We avoid similar edges
            for (Edge portOneEdge : portOne.getConnectedEdges())
                if (portOneEdge.getOppositePort(portOne).equals(portTwo))
                    throw new SimilarEdgeAlreadyExistsException(portOneEdge);
            // Ports' attributes
            portOne.getConnectedEdges().add(this);
            portTwo.getConnectedEdges().add(this);
            // Attributes
            this.portOne = portOne;
            this.portTwo = portTwo;
        }

希望这有助于理解问题。

编辑:谢谢大家!

Answer 1

您可以查看注释处理工具。它允许您在编译时进行注释，并在编译代码时产生警告/错误。

Answer 2

我不认为有一个干净的方法来做到这一点。请考虑另一种设计。

如果你希望类A只服务于类B和它的子类，定义类A为B的内部类，并给它一个受保护的访问修饰符。