基于SpringBoot打造一个通用CLI命令系统

@ShellComponent
public class ExecCommand {

    @ShellMethod(key = "exec", value = "执行远程服务命令")
    public String executeCommand(
            @ShellOption(value = {"", "service"}, help = "服务名称") String serviceName,
            @ShellOption(value = "--args", help = "命令参数", arity = 100) String[] args) {

        // 构建请求并发送到服务端
        CommandRequest request = new CommandRequest(serviceName, Arrays.asList(args));
        return httpClient.post("/cli", request);
    }
}

> exec userService --args list
user1, user2, user3

> exec roleService --args users admin
role1, role2

> exec systemService --args status
系统正常运行

@RestController
@RequestMapping("/cli")
public class CliController {

    @Autowired
    private ApplicationContext applicationContext;

    @PostMapping
    public String execute(@RequestBody CommandRequest request) {
        String serviceName = request.getService();
        String[] args = request.getArgs().toArray(new String[0]);

        // 动态获取 Service Bean
        Object serviceBean = applicationContext.getBean(serviceName);

        // 执行命令
        if (serviceBean instanceof CommandHandler handler) {
            return handler.handle(args);
        }

        return "服务未找到";
    }
}

public interface CommandHandler {
    String handle(String[] args);
    default String getDescription() { return "命令描述"; }
    default String getUsage() { return "使用说明"; }
}

@Service("userService")
public class UserService implements CommandHandler {

    @Override
    public String handle(String[] args) {
        if (args.length == 0) return getUsage();

        switch (args[0]) {
            case "list":
                return listUsers(args.length > 1 ? args[1] : null);
            case "get":
                return getUser(args[1]);
            default:
                return "未知命令: " + args[0];
        }
    }

    private String listUsers(String type) {
        // 实现获取用户列表逻辑
        return "用户列表...";
    }
}

cli:
  allowed-services:
    - userService
    - roleService
    - systemService

logger.info("CLI请求 - 服务: {}, 参数: {}, 来源: {}",
    serviceName, Arrays.toString(args), httpRequest.getRemoteAddr());

# 查看系统状态
exec systemService --args status

# 重启服务
exec serviceManager --args restart userService

# 查看日志
exec logService --args tail 100 error

# 查看用户详情
exec userService --args get 123

# 测试接口
exec testService --args simulate /api/orders

# 清理缓存
exec cacheService --args clear all

# 批量导入用户
exec userService --args import users.csv

# 批量更新角色
exec roleService --args batchUpdate role-mapping.json

private String formatResponse(String data) {
    try {
        Object json = objectMapper.readValue(data, Object.class);
        return objectMapper.writerWithDefaultPrettyPrinter()
                          .writeValueAsString(json);
    } catch (Exception e) {
        return data;
    }
}

exec script --args commands.txt