Search code examples
javaprotocol-buffersprotobuf-java

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?


Solution

  • 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)