Examples with ProtoTypeMap com.salesforce.jprotoc.ProtoTypeMap used on opensource projects

Example 1 with ProtoTypeMap

use of com.salesforce.jprotoc.ProtoTypeMap in project dubbo by alibaba.

the class AbstractGenerator method generateFiles.

@Override
public List<PluginProtos.CodeGeneratorResponse.File> generateFiles(PluginProtos.CodeGeneratorRequest request) throws GeneratorException {
    final ProtoTypeMap typeMap = ProtoTypeMap.of(request.getProtoFileList());
    List<FileDescriptorProto> protosToGenerate = request.getProtoFileList().stream().filter(protoFile -> request.getFileToGenerateList().contains(protoFile.getName())).collect(Collectors.toList());
    List<ServiceContext> services = findServices(protosToGenerate, typeMap);
    return generateFiles(services);
}

Example 2 with ProtoTypeMap

use of com.salesforce.jprotoc.ProtoTypeMap in project dubbo by alibaba.

the class AbstractGenerator method buildServiceContext.

private ServiceContext buildServiceContext(ServiceDescriptorProto serviceProto, ProtoTypeMap typeMap, List<Location> locations, int serviceNumber) {
    ServiceContext serviceContext = new ServiceContext();
    serviceContext.fileName = getClassPrefix() + serviceProto.getName() + getClassSuffix() + ".java";
    serviceContext.className = getClassPrefix() + serviceProto.getName() + getClassSuffix();
    serviceContext.serviceName = serviceProto.getName();
    serviceContext.deprecated = serviceProto.getOptions() != null && serviceProto.getOptions().getDeprecated();
    List<Location> allLocationsForService = locations.stream().filter(location -> location.getPathCount() >= 2 && location.getPath(0) == FileDescriptorProto.SERVICE_FIELD_NUMBER && location.getPath(1) == serviceNumber).collect(Collectors.toList());
    Location serviceLocation = allLocationsForService.stream().filter(location -> location.getPathCount() == SERVICE_NUMBER_OF_PATHS).findFirst().orElseGet(Location::getDefaultInstance);
    serviceContext.javaDoc = getJavaDoc(getComments(serviceLocation), getServiceJavaDocPrefix());
    for (int methodNumber = 0; methodNumber < serviceProto.getMethodCount(); methodNumber++) {
        MethodContext methodContext = buildMethodContext(serviceProto.getMethod(methodNumber), typeMap, locations, methodNumber);
        serviceContext.methods.add(methodContext);
        serviceContext.methodTypes.add(methodContext.inputType);
        serviceContext.methodTypes.add(methodContext.outputType);
    }
    return serviceContext;
}