Java MethodDescriptorProto类代码示例

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private MethodContext buildMethodContext(MethodDescriptorProto methodProto, ProtoTypeMap typeMap) {
    MethodContext methodContext = new MethodContext();
    methodContext.methodName = lowerCaseFirst(methodProto.getName());
    methodContext.inputType = typeMap.toJavaTypeName(methodProto.getInputType());
    methodContext.outputType = typeMap.toJavaTypeName(methodProto.getOutputType());
    methodContext.deprecated = methodProto.getOptions() != null && methodProto.getOptions().getDeprecated();
    methodContext.isManyInput = methodProto.getClientStreaming();
    methodContext.isManyOutput = methodProto.getServerStreaming();
    if (!methodProto.getClientStreaming() && !methodProto.getServerStreaming()) {
        methodContext.reactiveCallsMethodName = "oneToOne";
        methodContext.grpcCallsMethodName = "asyncUnaryCall";
    }
    if (!methodProto.getClientStreaming() && methodProto.getServerStreaming()) {
        methodContext.reactiveCallsMethodName = "oneToMany";
        methodContext.grpcCallsMethodName = "asyncServerStreamingCall";
    }
    if (methodProto.getClientStreaming() && !methodProto.getServerStreaming()) {
        methodContext.reactiveCallsMethodName = "manyToOne";
        methodContext.grpcCallsMethodName = "asyncClientStreamingCall";
    }
    if (methodProto.getClientStreaming() && methodProto.getServerStreaming()) {
        methodContext.reactiveCallsMethodName = "manyToMany";
        methodContext.grpcCallsMethodName = "asyncBidiStreamingCall";
    }
    return methodContext;
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private String generateMethod(MethodDescriptorProto method, String inPutType, String outPutType,
    String methodName, String inputValue) {
  String methodStr =
      "public " + outPutType + " " + methodName + "(" + inPutType + " " + inputValue + ");";
  boolean isClientStream = !method.getServerStreaming() && method.getClientStreaming();
  boolean isBidiStream = method.getServerStreaming() && method.getClientStreaming();
  boolean isServerStream = method.getServerStreaming() && !method.getClientStreaming();
  if (isClientStream || isBidiStream) {
    methodStr =
        "public " + inPutType + " " + methodName + "(" + outPutType + " responseObserver);";
  } else if (isServerStream) {
    methodStr = "public void " + methodName + "(" + inPutType + " " + inputValue + ","
        + outPutType + " responseObserver);";
  }
  return methodStr;
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private MethodDescriptorProto generateMethod(Method method) {
  MethodDescriptorProto.Builder builder = MethodDescriptorProto.newBuilder();
  builder.setName(method.getName());
  builder.setInputType(getTypeName(method.getRequestTypeUrl()));
  builder.setOutputType(getTypeName(method.getResponseTypeUrl()));
  builder.setOptions(generateMethodOptions(method));
  builder.setClientStreaming(method.getRequestStreaming());
  // protoc set serverStreaming field as false for legacy streaming options,
  // but google.protobuf.Method set the responseStreaming field to true for both new and legacy
  // streaming setup. So we need to distinguish streaming style while generating
  // MethodDescriptorProto.
  // But we cannot distinguish if the new and old styles are both set which should be rare case.
  if (method.getResponseStreaming() && isLegacyStreaming(method)) {
    builder.setServerStreaming(false);
  } else {
    builder.setServerStreaming(method.getResponseStreaming());
  }
  return builder.build();
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
protected void decompile(ServiceDescriptorProto serviceDescriptor) throws IOException {
    indentedFormat("service %s {", serviceDescriptor.getName());
    indent++;
    if (serviceDescriptor.hasOptions()) {
        decompileOptions(serviceDescriptor.getOptions());
    }
    for (MethodDescriptorProto methodDescriptor : serviceDescriptor.getMethodList()) {
        indentedFormat("rpc %s (%s) returns (%s)",
                       methodDescriptor.getName(), methodDescriptor.getInputType(), methodDescriptor.getOutputType());
        if (methodDescriptor.hasOptions()) {
            write("{ ");
            indent++;
            decompileOptions(methodDescriptor.getOptions());
            indent--;
            indentedFormat("}");
        }
        else {
            write(";");
        }
    }
    indent--;
    indentedFormat("}");
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
@Override
protected List<String> collectFileData() {
  String className = super.getClassName();
  String packageName = super.getSourcePackageName().toLowerCase();
  List<String> fileData = Lists.newArrayList();
  fileData.add("package " + packageName + ";");
  fileData.add("public interface " + className + "{");
  for (MethodDescriptorProto method : serviceMethods) {
    String outPutType = method.getOutputType();
    String inPutType = method.getInputType();
    String methodName = method.getName();
    inPutType = CommonUtils.findPojoTypeFromCache(inPutType, pojoTypeCache);
    outPutType = CommonUtils.findPojoTypeFromCache(outPutType, pojoTypeCache);
    String stream = generateGrpcStream(method, inPutType, outPutType);
    if (method.getServerStreaming() || method.getClientStreaming()) {
      outPutType = "io.grpc.stub.StreamObserver<" + outPutType + ">";
    }
    String inputValue = CommonUtils.findNotIncludePackageType(inPutType).toLowerCase();
    if (method.getClientStreaming()) {
      inPutType = "io.grpc.stub.StreamObserver<" + inPutType + ">";
    }
    if (stream != null)
      fileData.add(stream);
    String methodStr = generateMethod(method, inPutType, outPutType, methodName, inputValue);
    fileData.add(methodStr);
  }
  fileData.add("}");
  return fileData;
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private Interface(ProtoFile parent, ServiceDescriptorProto proto, String path) {
  super(parent, proto.getName(), path);
  this.proto = proto;

  // Build methods.
  ImmutableList.Builder<Method> methodsBuilder = ImmutableList.builder();
  List<MethodDescriptorProto> methodProtos = proto.getMethodList();
  for (int i = 0; i < methodProtos.size(); i++) {
    String childPath = buildPath(path, ServiceDescriptorProto.METHOD_FIELD_NUMBER, i);
    methodsBuilder.add(Method.create(this, methodProtos.get(i), childPath));
  }

  methods = methodsBuilder.build();
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private Method(Interface parent, MethodDescriptorProto proto, String path) {
  super(parent, proto.getName(), path);
  this.isDeprecated = proto.getOptions().getDeprecated();
  this.descriptor = new MethodDescriptor(proto);
  this.requestStreaming = proto.getClientStreaming();
  this.responseStreaming = proto.getServerStreaming();
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private MethodDescriptor(final MethodDescriptorProto proto,
                         final FileDescriptor file,
                         final ServiceDescriptor parent,
                         final int index)
                  throws DescriptorValidationException {
  this.index = index;
  this.proto = proto;
  this.file = file;
  service = parent;

  fullName = parent.getFullName() + '.' + proto.getName();

  file.pool.addSymbol(this);
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private MethodDescriptor (final MethodDescriptorProto proto,
        final FileDescriptor file,
        final ServiceDescriptor parent,
        final int index)
        throws DescriptorValidationException {
    this.index = index;
    this.proto = proto;
    this.file = file;
    service = parent;

    fullName = parent.getFullName () + '.' + proto.getName ();

    file.pool.addSymbol (this);
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private void makeCanonicalService(final ServiceDescriptorProto.Builder service,
    final ServiceDescriptor serviceDescriptor) {
  for (final MethodDescriptorProto.Builder method : service.getMethodBuilderList()) {
    final MethodDescriptor methodDescriptor =
        serviceDescriptor.findMethodByName(method.getName());
    method.setInputType(ensureLeadingDot(methodDescriptor.getInputType().getFullName()));
    method.setOutputType(ensureLeadingDot(methodDescriptor.getOutputType().getFullName()));
  }
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
@Override
public void exitMethodStatement(final MethodStatementContext ctx) {
  final MethodDescriptorProto.Builder methodBuilder =
      MethodDescriptorProto.Builder.class.cast(scopes.getProtoBuilder());
  methodBuilder.setName(ctx.identifier().getText()).setInputType(ctx.extendedId(0).getText())
      .setOutputType(ctx.extendedId(1).getText());
  scopes.popScope();
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private void buildAllOptions(final ServiceDescriptorProto.Builder proto) {
  if (!buildOptions(proto.getOptionsBuilder())) {
    proto.clearOptions();
  }

  for (final MethodDescriptorProto.Builder methodProto : proto.getMethodBuilderList()) {
    if (!buildOptions(methodProto.getOptionsBuilder())) {
      methodProto.clearOptions();
    }
  }
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
private boolean isCanonical(final ServiceDescriptorProto serviceProto) {
  if (serviceProto.hasOptions() && serviceProto.getOptions().getUninterpretedOptionCount() > 0) {
    return false;
  }

  for (final MethodDescriptorProto methodProto : serviceProto.getMethodList()) {
    if (methodProto.hasOptions() && methodProto.getOptions().getUninterpretedOptionCount() > 0) {
      return false;
    }
  }

  return true;
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
public void handle(ServiceDescriptorProto service) throws IOException {
  ImmutableList.Builder<ServiceHandlerData.Method> methods = ImmutableList.builder();
  for (MethodDescriptorProto method : service.getMethodList()) {
    ServiceHandlerData.Method methodData = new ServiceHandlerData.Method(
        method.getName(),
        CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, method.getName()),
        types.lookup(method.getInputType()).toString(),
        types.lookup(method.getOutputType()).toString());
    methods.add(methodData);
  }

  String fullName = Joiner.on('.').skipNulls().join(protoPackage, service.getName());

  ServiceHandlerData.Service serviceData = new ServiceHandlerData.Service(
      service.getName(), fullName, methods.build());
  ServiceHandlerData data = new ServiceHandlerData(javaPackage, multipleFiles, serviceData);

  String template = Resources.toString(Resources.getResource(this.getClass(),
      "service_class.mvel"), Charsets.UTF_8);
  String serviceFile = (String) TemplateRuntime.eval(template,
      ImmutableMap.<String, Object>of("handler", data));

  CodeGeneratorResponse.Builder response = CodeGeneratorResponse.newBuilder();
  CodeGeneratorResponse.File.Builder file = CodeGeneratorResponse.File.newBuilder();
  file.setContent(serviceFile);
  file.setName(javaPackage.replace('.', '/') + '/' + service.getName() + ".java");
  if (!multipleFiles) {
    file.setName(javaPackage.replace('.', '/') + '/' + outerClassName + ".java");
    file.setInsertionPoint("outer_class_scope");
  }
  response.addFile(file);
  response.build().writeTo(output);
}
 

import com.google.protobuf.DescriptorProtos.MethodDescriptorProto; //导入依赖的package包/类
public void setServiceMethods(List<MethodDescriptorProto> serviceMethods) {
  this.serviceMethods = serviceMethods;
}