Java tutorial
/** * Copyright 2016-2017 Sixt GmbH & Co. Autovermietung KG * Licensed under the Apache License, Version 2.0 (the "License"); you may * not use this file except in compliance with the License. You may obtain a * copy of the License at http://www.apache.org/licenses/LICENSE-2.0 * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the * License for the specific language governing permissions and limitations * under the License. */ package com.squareup.wire.schema.internal.parser; import com.google.protobuf.GeneratedMessage; import com.google.protobuf.GeneratedMessageV3; import com.google.protobuf.Message; import com.sixt.service.framework.rpc.RpcClient; import com.sixt.service.framework.rpc.RpcClientFactory; import com.sixt.service.framework.servicetest.service.ServiceMethod; import io.github.lukehutch.fastclasspathscanner.FastClasspathScanner; import io.github.lukehutch.fastclasspathscanner.scanner.ScanResult; import org.apache.commons.io.FileUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.nio.file.Files; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.stream.Collectors; import java.util.stream.Stream; import java.util.zip.ZipEntry; import java.util.zip.ZipFile; import java.util.zip.ZipInputStream; public class RpcMethodScanner { private static final Logger logger = LoggerFactory.getLogger(RpcMethodScanner.class); private RpcClientFactory rpcClientFactory; public RpcMethodScanner(RpcClientFactory rpcClientFactory) { this.rpcClientFactory = rpcClientFactory; } @SuppressWarnings("unchecked") public Map<String, ServiceMethod<Message>> getMethodHandlers(String serviceName) { String serviceNameUnderscores = serviceName.replace("-", "_"); List<RpcMethodDefinition> rpcMethodDefinitions = getRpcMethodDefinitions(serviceNameUnderscores); // get the generated proto classes List<String> protoClasses = getGeneratedProtoClasses(serviceNameUnderscores); Map<String, ServiceMethod<Message>> serviceMethods = new HashMap<>(); for (RpcMethodDefinition def : rpcMethodDefinitions) { try { logger.info("Adding endpoint {} for {}", def.getMethodName(), serviceName); Class<? extends Message> protobufClass = findProtobufClass(protoClasses, def.getResponseType()); RpcClient rpcClient = rpcClientFactory.newClient(serviceName, def.getMethodName(), protobufClass) .build(); serviceMethods.put(def.getMethodName(), new ServiceMethod(rpcClient, protobufClass)); } catch (ClassNotFoundException ex) { logger.error("Error while adding RPC endpoint for service " + serviceName, ex); } } return serviceMethods; } public List<RpcMethodDefinition> getRpcMethodDefinitions(String serviceName) { // first try: search local directory for proto file List<RpcMethodDefinition> rpcMethodDefinitions = searchDirectory(System.getProperty("user.dir"), serviceName); // if rpcMethodDefinitions still empty, search in jars of classpath if (rpcMethodDefinitions.isEmpty()) { rpcMethodDefinitions = searchClasspath(serviceName); } if (rpcMethodDefinitions.isEmpty()) { logger.warn("No RPC endpoints found for {}", serviceName); } return rpcMethodDefinitions; } public List<String> getGeneratedProtoClasses(String serviceName) { FastClasspathScanner cpScanner = new FastClasspathScanner(serviceName); ScanResult scanResult = cpScanner.scan(); List<String> oldProtobuf = scanResult.getNamesOfSubclassesOf(GeneratedMessage.class); List<String> newProtobuf = scanResult.getNamesOfSubclassesOf(GeneratedMessageV3.class); return Stream.concat(oldProtobuf.stream(), newProtobuf.stream()).collect(Collectors.toList()); } @SuppressWarnings("unchecked") public Class<? extends Message> findProtobufClass(List<String> protoClasses, String requestType) throws ClassNotFoundException { for (String pbClass : protoClasses) { if (pbClass.contains(requestType)) { Class<? extends Message> retval = (Class<? extends Message>) Class.forName(pbClass); return retval; } } return null; } private List<RpcMethodDefinition> searchClasspath(String serviceName) { logger.info("Scanning classpath for proto files of {}", serviceName); List<RpcMethodDefinition> rpcMethodDefinitions = new ArrayList<>(); String classpath = System.getProperty("java.class.path"); String jars[] = classpath.split(":"); for (String jar : jars) { try { rpcMethodDefinitions = searchJar(jar, serviceName); } catch (IOException e) { e.printStackTrace(); } if (!rpcMethodDefinitions.isEmpty()) { break; } } return rpcMethodDefinitions; } private List<RpcMethodDefinition> searchDirectory(String directory, String serviceName) { logger.info("Scanning {} for proto files of {}", directory, serviceName); List<RpcMethodDefinition> rpcMethodDefinitions = new ArrayList<>(); File scanDir = new File(directory); List<File> protoFiles = (List<File>) FileUtils.listFiles(scanDir, new String[] { "proto" }, true); // if no proto files have been found also scan the parent directory. we need this for the go services if (protoFiles.isEmpty()) { protoFiles = (List<File>) FileUtils.listFiles(scanDir.getParentFile(), new String[] { "proto" }, true); } for (File f : protoFiles) { try { rpcMethodDefinitions.addAll(inspectProtoFile(new FileInputStream(f), serviceName)); } catch (IOException e) { e.printStackTrace(); } if (!rpcMethodDefinitions.isEmpty()) { return rpcMethodDefinitions; } } return rpcMethodDefinitions; } private List<RpcMethodDefinition> searchJar(String jarpath, String serviceName) throws IOException { List<RpcMethodDefinition> defs = new ArrayList<>(); File jarFile = new File(jarpath); if (!jarFile.exists() || jarFile.isDirectory()) { return defs; } ZipInputStream zip = new ZipInputStream(new FileInputStream(jarFile)); ZipEntry ze; while ((ze = zip.getNextEntry()) != null) { String entryName = ze.getName(); if (entryName.endsWith(".proto")) { ZipFile zipFile = new ZipFile(jarFile); try (InputStream in = zipFile.getInputStream(ze)) { defs.addAll(inspectProtoFile(in, serviceName)); if (!defs.isEmpty()) { zipFile.close(); break; } } zipFile.close(); } } zip.close(); return defs; } private List<RpcMethodDefinition> inspectProtoFile(InputStream instream, String serviceName) throws IOException { File tempFile = File.createTempFile("proto", null); tempFile.delete(); Files.copy(instream, tempFile.toPath()); SixtProtoParser parser = new SixtProtoParser(tempFile, serviceName); List<RpcMethodDefinition> methodDefinitions = parser.getRpcMethods(); tempFile.delete(); instream.close(); return methodDefinitions; } }