重构java代码

Question

好吧，猜猜这个问题看起来很像：

What is the best way to replace or substitute if..else if..else trees in programs?

考虑关闭这个问题！

---

我想重构如下所示的代码：

String input; // input from client socket.
if (input.equals(x)) {
  doX();
} else if (input.equals(y)) {
  doY();
} else {
  unknown_command();
}

它是检查来自套接字的输入以执行某些操作的代码，但我不喜欢 if else 构造，因为每次将新命令添加到服务器（代码）时，都必须添加一个新的 if else很丑。此外，在删除命令时，if else 也必须修改。

Answer 1

在Map<String, Command> 中收集这些命令，其中Command 是带有execute() 方法的interface。

Map<String, Command> commands = new HashMap<String, Command>();
// Fill it with concrete Command implementations with `x`, `y` and so on as keys.

// Then do:
Command command = commands.get(input);
if (command != null) {
    command.execute();
} else {
    // unknown command.
}

为了更进一步，您可以考虑通过扫描实现特定接口（在本例中为Command）的类或类路径中的特定注释来动态填充映射。 Google Reflections 可能会对此有所帮助。

更新（来自 cmets）您也可以考虑将 answer of Instantsoup 与我的回答结合起来。在buildExecutor() 方法中，首先从Map 中获取命令，如果Map 中不存在该命令，则尝试加载关联的类并将其放入Map。有点懒加载。这比在我的答案中扫描整个类路径并在 Instantsoup 的答案中每次都创建它更有效。

Answer 2

一种方法可能是拥有一个接口ICommand，它是一个命令的通用合同，例如：

public interface ICommand {
    /** @param context The command's execution context */
    public void execute(final Object context);
    public String getKeyword();
}

然后您可以使用 Java 的 SPI 机制自动发现您的各种实现并将它们注册到 Map<String,ICommand> 中，然后执行 knownCommandsMap.get(input).execute(ctx) 或类似的操作。

这实际上使您能够将服务与命令实现分离，从而有效地使这些实现可插拔。

向 SPI 注册实现类是通过添加一个名为 ICommand 类的完全限定名称的文件来完成的（因此，如果它在包 dummy 中，则该文件将在您的类路径中为 META-INF/dummy.ICommand），然后您'将它们加载并注册为：

final ServiceLoader<ICommand> spi = ServiceLoader.load(ICommand.class);
for(final ICommand commandImpl : spi)
    knownCommandsMap.put(commandImpl.getKeyword(), commandImpl);

Answer 3

接口、工厂和一点反射怎么样？您仍然需要处理错误输入的异常，但您总是需要这样做。使用此方法，您只需为新输入添加一个新的 Executor 实现。

public class ExecutorFactory
{
    public static Executor buildExecutor(String input) throws Exception
    {
        Class<Executor> forName = (Class<Executor>) Class.forName(input);
        return (Executor) executorClass.newInstance();
    }
}

public interface Executor
{
    public void execute();
}


public class InputA implements Executor
{
    public void execute()
    {
        // do A stuff
    }
}

public class InputB implements Executor
{
    public void execute()
    {
        // do B stuff
    }
}

你的代码示例就变成了

String input;
ExecutorFactory.buildExecutor(input).execute();

Answer 4

在枚举类上构建命令模式可以减少一些样板代码。假设input.equals(x)中的x是“XX”，input.equals(y)中的y是“YY”

enum Commands {
   XX {
     public void execute() { doX(); }        
   },
   YY {
     public void execute() { doY(); }        
   };

   public abstract void execute();
}

String input = ...; // Get it from somewhere

try {
  Commands.valueOf(input).execute();
}
catch(IllegalArgumentException e) {
   unknown_command();
}

Answer 5

您说您正在处理来自套接字的输入。输入多少？它有多复杂？它的结构如何？

根据这些问题的答案，您最好编写一个语法并让解析器生成器（例如，ANTLR）生成输入处理代码。