Question

我正在尝试在How to write a custom Protobuf CodeGenerator in Java上作为例子。当我尝试使用

编译文件时

import com.google.protobuf.compiler.PluginProtos;

import java.io.IOException;

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

由于CodeGenerator未知，我得到了编译错误。我在pom.xml Maven文件中的“ dependencies”标记中包含以下内容-

<dependency>
    <groupId>com.google.protobuf</groupId>
    <artifactId>protobuf-java</artifactId>
    <version>3.6.1</version>
</dependency>

我还需要向依赖项添加什么才能使我的插件正常工作？我计划使用https://developers.google.com/protocol-buffers/docs/reference/java/中的Java API。

Answer 1

是的，这有点愚蠢-CodeGenerator是一个自定义类，我们需要编写它，它的名称通过暗示它来自Google库而混淆了。

因此，编写后可能看起来像这样-https://www.expobrain.net/2015/09/13/create-a-plugin-for-google-protocol-buffer/的Python代码的近似类比，但没有打包成JSON和清理子字段：

import com.google.protobuf.DescriptorProtos;
import com.google.protobuf.compiler.PluginProtos;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class ProtocPlugin {
    private static List _traverse(String strPackage, List items) {
        List<List> res = new ArrayList<>();
        for(Object item : items) {
            res.add(Arrays.asList(item, strPackage));
            if(item instanceof DescriptorProtos.DescriptorProto) {
                DescriptorProtos.DescriptorProto dp = (DescriptorProtos.DescriptorProto) item;
                for(DescriptorProtos.EnumDescriptorProto e : dp.getEnumTypeList()) {
                    res.add(Arrays.asList(e, strPackage));
                }
                for(DescriptorProtos.DescriptorProto nested : dp.getNestedTypeList()) {
                    String strNestedPackage = strPackage + nested.getName();
                    for(Object nestedItem : _traverse(strNestedPackage, nested.getNestedTypeList())) {
                        res.add(Arrays.asList(((List)nestedItem).get(0), strNestedPackage));
                    }
                }
            }
        }
        return res;
    }

    public static void main(String[] args) throws IOException {
        StringBuilder data = new StringBuilder();
        PluginProtos.CodeGeneratorRequest codeGeneratorRequest = PluginProtos.CodeGeneratorRequest.parseFrom(System.in);
        codeGeneratorRequest.getProtoFileList().forEach((DescriptorProtos.FileDescriptorProto fileDescriptorProto) -> {
            String strPackage = fileDescriptorProto.getPackage();
            if(strPackage == null || strPackage.isEmpty()) {
                strPackage = "&lt;root&gt;";
            }
            data.append("package: ").append(strPackage).append("\n");
            data.append("filename: ").append(fileDescriptorProto.getName()).append("\n");

            List<DescriptorProtos.EnumDescriptorProto> enums = fileDescriptorProto.getEnumTypeList();
            for(Object pair : _traverse(strPackage, enums)) {
                data.append("type: enum").append("\n");
                data.append(((List)pair).get(0)).append(((List)pair).get(1)).append(" ");
            }

            List<DescriptorProtos.DescriptorProto> messageTypes = fileDescriptorProto.getMessageTypeList();
            for(Object pair : _traverse(strPackage, messageTypes)) {
                data.append("type: message").append("\n");
                data.append(((List)pair).get(0)).append(((List)pair).get(1)).append(" ");
            }
        });

        PluginProtos.CodeGeneratorResponse.Builder builder = PluginProtos.CodeGeneratorResponse.newBuilder();
        builder.addFileBuilder().setContent(data.toString()).setName("mytest.txt");
        PluginProtos.CodeGeneratorResponse response = builder.build();
        response.writeTo(System.out);
    }
}

protoc的发布可能与

protoc --plugin=protoc-gen-custom=my-plugin.bat --custom_out=. hello.proto

其中my-plugin.bat包含类似内容

@echo off
java -cp target/classes;c:/users/bover/.m2/repository/com/google/protobuf/protobuf-java/3.6.1/protobuf-java-3.6.1.jar ProtocPlugin

这里，我们假设我们的Java插件将ProtocPlugin.class编译到target / classes目录中。输出将在mytest.txt文件中。 hello.proto是来自上述Python示例的简单proto文件。

Protoc Java插件的依赖关系？

1 个答案: