简体   繁体   中英

How to get value of protobuf custom option in Java?

Background: I'm writing a protoc plugin.

The custom protobuf option is implemented with the following:

syntax = "proto3";

package com.example.proto.options;

import "google/protobuf/descriptor.proto";

option java_multiple_files = true;
option java_outer_classname = "ServerOptionsProto";
option java_package = "com.example.proto.options";

extend google.protobuf.FileOptions {
    ServerOptions server = 50621;
}

message ServerOptions {
    // Java classname
    string name = 1;
}

The following is an example usage:

syntax = "proto3";

package com.example.testdata;

import "com/example/proto/options/server.proto";

option java_multiple_files = true;
option java_package = "com.example.testdata.protogen";
option java_outer_classname = "TestDataProto";

option (com.example.proto.options.server).name = "TestData";

Trying to follow https://developers.google.com/protocol-buffers/docs/proto#options , the following (in Groovy) doesn't work:

request.getProtoFileList().stream().filter { proto ->
  proto.serviceCount > 0
}.flatMap { proto ->
  serverName = proto.getDescriptor().getOptions()?.getExtension(com.example.proto.options.ServerOptionsProto.server)?.name
}

What's the right way in Java to access the value of the custom option?

The java_package and java_outer_classname options need to be used:

serverName = proto.getOptions()?.getExtension(com.example.proto.options.ServerOptionsProto.server)?.name

Also, since this is processed by a protoc plugin, the extension needs to be registered as per Extensions :

final registry = ExtensionRegistry.newInstance();
registry.add(ServerOptionsProto.server)
final request = PluginProtos.CodeGeneratorRequest.parseFrom(input, registry)

I believe the problem is that your option is a file-level one, but you are trying to access it as if it were a message-level option. Instead of proto.getDescriptor().getOptions() , try proto.getDescriptor().getFile().getOptions() .

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM