java - 如何用 Java 编写自定义 Protobuf CodeGenerator

Question

我正在尝试为内部专有编程语言编写自定义代码生成器。我想我可以使用 protoc 插件指南用 Java 编写生成器。我的 main() 做了这样的事情:

public static void main(String[] args) throws IOException {
    CodeGenerator gen = new CodeGenerator();
    PluginProtos.CodeGeneratorRequest codeGeneratorRequest = PluginProtos.CodeGeneratorRequest.parseFrom(args[0].getBytes());
    codeGeneratorRequest.getProtoFileList().forEach(gen::handleFile);
    // get the response and do something with it 
    //PluginProtos.CodeGeneratorResponse response = PluginProtos.CodeGeneratorResponse.newBuilder().build();
    //response.writeTo(System.out);
}

(显然我才刚刚开始；想在实际编写生成逻辑之前先让一些东西工作起来)

问题是:如何使用我的插件调用带有 --plugin 参数的 protoc 以我的自定义语言生成代码？我尝试编写一个 shell 脚本来执行此操作:

#!/bin/bash
java -cp ./codegen.jar CodeGeneratorMain "$@"

我尝试像这样调用 protoc:protoc --plugin=protoc-gen-code --code_out=./build hello.proto 然而，当我运行它时，我得到了这个错误:

Exception in thread "main" java.lang.ArrayIndexOutOfBoundsException: 0 at CodeGeneratorMain.main(CodeGeneratorMain.java:12) --code_out: protoc-gen-code: Plugin failed with status code 1.

就好像它根本没有在标准输入上传递 CodeGeneratorRequest 一样。我将如何验证？我做错了什么吗？

Answer 1

因此，在阅读并重新阅读文档后，我意识到我犯了一个非常愚蠢的错误:protoc 通过 stdin 不是 通过 argv 传递解析的输入。这意味着如果我改变这个: PluginProtos.CodeGeneratorRequest codeGeneratorRequest = PluginProtos.CodeGeneratorRequest.parseFrom(args[0].getBytes()); 到这个: PluginProtos.CodeGeneratorRequest codeGeneratorRequest = PluginProtos.CodeGeneratorRequest .parseFrom(System.in);

有效。