4

我正在使用 protobuf，并且正在从以下 proto 文件生成 JAVA 类。

syntax = "proto3";
enum Greeting {
    NONE = 0;
    MR = 1;
    MRS = 2;
    MISS = 3;
}

message Hello {
    Greeting greeting = 1;
    string name = 2;
}

message Bye {
    string name = 1;
}

option java_multiple_files = true;

现在我需要向生成的文件添加一些代码，我发现可以使用自定义插件（https://developers.google.com/protocol-buffers/docs/reference/java-generated#plugins）。我正在尝试用 Java 生成那个插件，就像这样。

public class Test {
   PluginProtos.CodeGeneratorResponse.getDefaultInstance();
   /* Code to get generated files from java_out and use the insertion points */
   codeGeneratorResponse.writeTo(System.out);
}

然后我跑

protoc --java_out=./classes --plugin=protoc-gen-demo=my-plugin --demo_out=. example.proto

问题是，在我的Test.java主要方法中，我不知道如何访问由该选项创建的文件，--java_out以便我可以使用它们的插入点。当前CodeGeneratorResponse，默认实例的 为空（无文件）。

有谁知道我如何CodeGeneratorResponse从 --java_out 获取，以便我可以向生成的类添加更多代码？

提前致谢。

4

2 回答 2

3

我最近也在为此苦苦挣扎，无法找到一个好的答案。在盯着CodeGeneratorResponse消息中的评论一段时间后，我终于弄明白了。

起初让我失望的是，我将插件视为管道，其中一个的输出馈送到下一个。但是，每个插件都获得完全相同的输入（通过CodeGeneratorRequest消息表示的已解析 .proto 文件），并且插件生成的所有代码（包括内置代码）都合并到输出文件中。但是，插件可能会修改以前插件的输出，这就是插入点的设计目的。

具体到您的问题，您将file在响应中添加一个name字段设置为生成的 Java 文件的名称，该insertion_point字段设置为您要添加代码的插入点的名称，并且该content字段设置为到您要在该点插入的代码。

我发现这篇文章有助于创建一个简单的插件（在本例中为 python）。作为一个简单的测试，我修改了generate_code那篇文章中的函数，如下所示：

def generate_code(request, response):
    for proto_file in request.proto_file:
        f = response.file.add()
        f.name = "Test.java"
        f.insertion_point = "outer_class_scope"
        f.content = "// Inserting a comment as a test"

然后我用插件运行 protoc：

$ cat test.proto
syntax = "proto3";
message MyMsg {
    int32 num = 1;
}
$ protoc --plugin=protoc-gen-sample=sample_proto_gen.py --java_out=. --sample_out=. test.proto
$ tail -n3 Test.java
  // Inserting a comment as a test
  // @@protoc_insertion_point(outer_class_scope)
}

您的插件只需要是一些可执行文件，它CodeGeneratorRequest从标准输入读取消息并将消息写入CodeGeneratorResponse标准输出，因此当然可以用 Java 编写。我只是选择了 python，因为我通常更喜欢它，并找到了这个简单的例子。

作为参考，这是我为基于自定义 protobuf 选项生成代码而编写的插件。

于 2019-09-21T23:23:02.070 回答
0

我制作了一个自定义 python 插件。要运行我的插件，我使用以下命令：

protoc --plugin=protoc-gen-custom=my_plugin_executable_file --custom_out=./build test.proto

所以我认为，你必须从你的 .java 文件生成一个可执行文件并在你的命令中使用它。

于 2019-12-12T10:02:45.473 回答