manager/zyplayer-doc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

grpc优化

Browse Source
This commit is contained in:
暮光：城中城
2019-04-05 21:55:18 +08:00
parent 0e0f512571
commit a6c1360593
9 changed files with 201 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/controller/GrpcDocController.java
View File

@@ -11,13 +11,23 @@ import com.zyplayer.doc.grpc.controller.po.ColumnInfo;
import com.zyplayer.doc.grpc.controller.po.GrpcDocInfo;
import com.zyplayer.doc.grpc.controller.po.GrpcServiceAndColumn;
import com.zyplayer.doc.grpc.controller.po.MethodParam;
import com.zyplayer.doc.grpc.framework.config.DocGrpcContext;
import com.zyplayer.doc.grpc.framework.config.DocGrpcService;
import com.zyplayer.doc.grpc.framework.config.SpringContextUtil;
import com.zyplayer.doc.grpc.framework.consts.Const;
import io.grpc.Channel;
import io.grpc.ManagedChannel;
import io.grpc.Metadata;
import io.grpc.netty.NettyChannelBuilder;
import io.grpc.stub.AbstractStub;
import io.grpc.stub.MetadataUtils;
import io.netty.handler.ssl.SslContext;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.annotation.Resource;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.util.HashMap;
@@ -37,8 +47,9 @@ import java.util.regex.Pattern;
@RequestMapping("/zyplayer-doc-grpc")
public class GrpcDocController {
@Resource
GrpcChannelFactory grpcChannelFactory;
@Autowired(required = false)
DocGrpcContext docGrpcContext;
private static Map<String, ColumnInfo> allColumnsMap = new HashMap<>();
private static Map<String, Object> allBlockingStubMap = new HashMap<>();
@@ -50,7 +61,7 @@ public class GrpcDocController {
*/
@RequestMapping("/service")
public DocResponseJson service() {
List<Object> grpcServiceList = SpringContextUtil.getBeanWithAnnotation(GrpcService.class);
List<Object> grpcServiceList = SpringContextUtil.getBeanWithAnnotation(DocGrpcService.class);
if (grpcServiceList == null || grpcServiceList.size() <= 0) {
return DocResponseJson.ok();
}
@@ -80,12 +91,17 @@ public class GrpcDocController {
* @since 2019年3月31日
*/
@RequestMapping("/execute")
public DocResponseJson execute(String service, String params) throws Exception {
List<GrpcDocInfo> grpcDocInfoList = this.getServiceInfoByJar(Class.forName(service));
public DocResponseJson execute(String docService, String params) throws Exception {
List<GrpcDocInfo> grpcDocInfoList = this.getServiceInfoByJar(Class.forName(docService));
JSONObject executeResult = null;
if (grpcDocInfoList != null && grpcDocInfoList.size() > 0) {
JSONObject paramMap = JSON.parseObject(params);
executeResult = this.executeFunction(grpcDocInfoList.get(0), paramMap);
try {
executeResult = this.executeFunction(grpcDocInfoList.get(0), paramMap);
} catch (Exception e) {
e.printStackTrace();
return DocResponseJson.warn("执行方法失败");
}
}
return DocResponseJson.ok(executeResult);
}
@@ -119,6 +135,9 @@ public class GrpcDocController {
if (newBuilder == null) {
throw new ConfirmException("参数组装失败");
}
if (docGrpcContext == null) {
throw new ConfirmException("注入grpc服务失败");
}
// 创建参数对象
Method build = newBuilder.getClass().getMethod("build");
Object request = build.invoke(newBuilder);
@@ -127,13 +146,24 @@ public class GrpcDocController {
Object blockingStub = allBlockingStubMap.get(grpcDocInfo.getService());
if (blockingStub == null) {
// 找到父类
Class<?> serviceClass = Class.forName(grpcDocInfo.getService());
String serviceSuperName = serviceClass.getSuperclass().getName();
// Class<?> serviceClass = Class.forName(grpcDocInfo.getService());
// String serviceSuperName = serviceClass.getSuperclass().getName();
String serviceSuperName = grpcDocInfo.getService();
int index = serviceSuperName.lastIndexOf(".");
serviceSuperName = serviceSuperName.substring(0, index) + "$" + serviceSuperName.substring(index + 1);
String superClassName = serviceSuperName.substring(0, serviceSuperName.indexOf("$"));
// 注册
Class<?> superClass = Class.forName(superClassName);
Method newBlockingStubMethod = superClass.getMethod("newBlockingStub", Channel.class);
blockingStub = newBlockingStubMethod.invoke(null, grpcChannelFactory.createChannel());
SslContext sslContext = docGrpcContext.getSslContext();
ManagedChannel channel = NettyChannelBuilder.forAddress(docGrpcContext.getHost(), docGrpcContext.getPort())
.sslContext(sslContext)
.build();
blockingStub = newBlockingStubMethod.invoke(null, channel);
Metadata metadata = docGrpcContext.getMetadata();
if (metadata != null) {
blockingStub = MetadataUtils.attachHeaders((AbstractStub) blockingStub, metadata);
}
allBlockingStubMap.put(grpcDocInfo.getService(), blockingStub);
}
Method sayHello = blockingStub.getClass().getMethod(grpcDocInfo.getMethod(), Class.forName(grpcDocInfo.getParamType()));
@@ -250,7 +280,7 @@ public class GrpcDocController {
* @author 暮光：城中城
* @since 2019年3月31日
*/
private List<MethodParam> getSetterFunction(Class clazz) {
private List<MethodParam> getSetterFunction(Class<?> clazz) {
List<MethodParam> result = new LinkedList<>();
Method[] methods = clazz.getDeclaredMethods();
StringBuilder nameSb = new StringBuilder();
@@ -306,7 +336,7 @@ public class GrpcDocController {
* @author 暮光：城中城
* @since 2019年3月31日
*/
private List<ColumnInfo> findClassColumns(Class clazz) throws Exception {
private List<ColumnInfo> findClassColumns(Class<?> clazz) throws Exception {
Method getMoney = clazz.getMethod("newBuilder");
Object newBuilder = getMoney.invoke(clazz);
@@ -331,7 +361,7 @@ public class GrpcDocController {
* @author 暮光：城中城
* @since 2019年3月31日
*/
private List<GrpcDocInfo> getServiceInfoByJar(Class clazz) {
private List<GrpcDocInfo> getServiceInfoByJar(Class<?> clazz) {
List<GrpcDocInfo> providerList = new LinkedList<>();
try {
Method[] methods = clazz.getDeclaredMethods();
@@ -353,13 +383,20 @@ public class GrpcDocController {
}
}
GrpcDocInfo grpcDocInfo = new GrpcDocInfo();
DocGrpcService annotation = clazz.getAnnotation(DocGrpcService.class);
if (annotation != null) {
grpcDocInfo.setService(annotation.service());
} else {
grpcDocInfo.setService(clazz.getName());
}
grpcDocInfo.setDocService(clazz.getName());
grpcDocInfo.setMethod(methodName);
grpcDocInfo.setService(clazz.getName());
grpcDocInfo.setParamType(paramType);
grpcDocInfo.setResultType(resultType);
providerList.add(grpcDocInfo);
}
} catch (Exception e) {
e.printStackTrace();
return null;
}
return providerList;

9
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/controller/po/GrpcDocInfo.java
View File

@@ -9,6 +9,7 @@ package com.zyplayer.doc.grpc.controller.po;
public class GrpcDocInfo {
private String method;
private String service;
private String docService;
private String explain;
private String result;
private String paramType;
@@ -70,4 +71,12 @@ public class GrpcDocInfo {
public void setMethod(String method) {
this.method = method;
}
public String getDocService() {
return docService;
}
public void setDocService(String docService) {
this.docService = docService;
}
}

16
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/framework/config/DocGrpcContext.java Normal file
View File

@@ -0,0 +1,16 @@
package com.zyplayer.doc.grpc.framework.config;
import io.grpc.Metadata;
import io.netty.handler.ssl.SslContext;
public interface DocGrpcContext {
SslContext getSslContext();
String getHost();
Metadata getMetadata();
Integer getPort();
}

20
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/framework/config/DocGrpcService.java Normal file
View File

@@ -0,0 +1,20 @@
package com.zyplayer.doc.grpc.framework.config;
import org.springframework.stereotype.Service;
import java.lang.annotation.*;
/**
* grpc服务的注解
*
* @author 暮光：城中城
* @since 2019年3月31日
*/
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Service
public @interface DocGrpcService {
String service() default "";
}

91
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/framework/config/GrpcClientConfiguration.java
View File

@@ -1,46 +1,45 @@
package com.zyplayer.doc.grpc.framework.config;
import com.nxest.grpc.client.*;
import com.nxest.grpc.client.configure.GrpcClientProperties;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.annotation.Resource;
/**
* grpc配置
*
* @author 暮光：城中城
* @since 2019年3月31日
*/
@Configuration
@AutoConfigureAfter(GrpcServerConfiguration.class)
public class GrpcClientConfiguration {
@Resource
private GrpcProperties grpcProperties;
@Bean
public GrpcClientProperties clientProperties() {
return grpcProperties.getClient();
}
@Bean
public GrpcChannelFactory channelFactory() {
return new AddressChannelFactory(clientProperties());
}
@Bean
public GrpcClientInterceptorDiscoverer clientInterceptorDiscoverer() {
return new AnnotationClientInterceptorDiscoverer();
}
@Bean
@ConditionalOnClass(GrpcClient.class)
public GrpcClientBeanPostProcessor grpcClientBeanPostProcessor() {
return new GrpcClientBeanPostProcessor(channelFactory(), clientInterceptorDiscoverer());
}
}
//package com.zyplayer.doc.grpc.framework.config;
//
//import com.nxest.grpc.client.*;
//import com.nxest.grpc.client.configure.GrpcClientProperties;
//import org.springframework.boot.autoconfigure.AutoConfigureAfter;
//import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
//import org.springframework.context.annotation.Bean;
//import org.springframework.context.annotation.Configuration;
//
//import javax.annotation.Resource;
//
///**
// * grpc配置
// *
// * @author 暮光：城中城
// * @since 2019年3月31日
// */
//@Configuration
////@AutoConfigureAfter(GrpcServerConfiguration.class)
//public class GrpcClientConfiguration {
//
// @Resource
// private GrpcProperties grpcProperties;
//
// @Bean
// public GrpcClientProperties clientProperties() {
// return grpcProperties.getClient();
// }
//
// @Bean
// public GrpcChannelFactory channelFactory() {
// return new AddressChannelFactory(clientProperties());
// }
//
// @Bean
// public GrpcClientInterceptorDiscoverer clientInterceptorDiscoverer() {
// return new AnnotationClientInterceptorDiscoverer();
// }
//
// @Bean
// @ConditionalOnClass(GrpcClient.class)
// public GrpcClientBeanPostProcessor grpcClientBeanPostProcessor() {
// return new GrpcClientBeanPostProcessor(channelFactory(), clientInterceptorDiscoverer());
// }
//}

68
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/framework/config/GrpcProperties.java
View File

@@ -1,34 +1,34 @@
package com.zyplayer.doc.grpc.framework.config;
import com.nxest.grpc.client.configure.GrpcClientProperties;
import com.nxest.grpc.server.configure.GrpcServerProperties;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;
/**
* grpc配置文件
*
* @author 暮光：城中城
* @since 2019年3月31日
*/
@Configuration
@ConfigurationProperties(prefix = "grpc")
public class GrpcProperties {
/**
* Grpc server properties
*/
private final GrpcServerProperties server = new GrpcServerProperties();
/**
* Grpc client properties
*/
private final GrpcClientProperties client = new GrpcClientProperties();
public GrpcServerProperties getServer() {
return server;
}
public GrpcClientProperties getClient() {
return client;
}
}
//package com.zyplayer.doc.grpc.framework.config;
//
//import com.nxest.grpc.client.configure.GrpcClientProperties;
//import com.nxest.grpc.server.configure.GrpcServerProperties;
//import org.springframework.boot.context.properties.ConfigurationProperties;
//import org.springframework.context.annotation.Configuration;
//
///**
// * grpc配置文件
// *
// * @author 暮光：城中城
// * @since 2019年3月31日
// */
//@Configuration
//@ConfigurationProperties(prefix = "grpc")
//public class GrpcProperties {
//
// /**
// * Grpc server properties
// */
// private final GrpcServerProperties server = new GrpcServerProperties();
// /**
// * Grpc client properties
// */
// private final GrpcClientProperties client = new GrpcClientProperties();
//
// public GrpcServerProperties getServer() {
// return server;
// }
//
// public GrpcClientProperties getClient() {
// return client;
// }
//}

42
zyplayer-doc-grpc/src/main/java/com/zyplayer/doc/grpc/framework/config/GrpcServerConfiguration.java
View File

@@ -12,24 +12,24 @@ import javax.annotation.Resource;
* @author 暮光：城中城
* @since 2019年3月31日
*/
@Configuration
public class GrpcServerConfiguration {
@Resource
private GrpcProperties grpcProperties;
@Bean
public GrpcServiceDiscoverer serviceDiscoverer() {
return new AnnotationGrpcServiceDiscoverer();
}
@Bean
public GrpcServerFactory severFactory() {
return new NettyGrpcServerFactory(serviceDiscoverer(), grpcProperties.getServer());
}
@Bean(name = "grpcServer", initMethod = "start", destroyMethod = "destroy")
public GrpcServer serverRunner() {
return severFactory().createServer();
}
}
//@Configuration
//public class GrpcServerConfiguration {
//
// @Resource
// private GrpcProperties grpcProperties;
//
// @Bean
// public GrpcServiceDiscoverer serviceDiscoverer() {
// return new AnnotationGrpcServiceDiscoverer();
// }
//
// @Bean
// public GrpcServerFactory severFactory() {
// return new NettyGrpcServerFactory(serviceDiscoverer(), grpcProperties.getServer());
// }
//
// @Bean(name = "grpcServer", initMethod = "start", destroyMethod = "destroy")
// public GrpcServer serverRunner() {
// return severFactory().createServer();
// }
//}

3
zyplayer-doc-grpc/src/main/resources/doc-grpc.html
View File

@@ -312,11 +312,12 @@
},
requestExecute() {
var fuc = app.grpcInfo.function;
var docService = app.grpcInfo.docService;
var service = fuc.substring(0, fuc.lastIndexOf("."));
var method = fuc.substring(fuc.lastIndexOf(".") + 1, fuc.length);
var paramColumnRequest = JSON.stringify(JSON.parse(app.paramColumnRequest));
var param = {
service: service,
docService: docService,
method: method,
params: paramColumnRequest,//JSON.stringify(params),
};

8
zyplayer-doc-grpc/src/main/resources/webjars/doc-grpc/js/formatjson.js
View File

@@ -26,7 +26,7 @@ var Formatjson = {
if (obj.length == 0) {
html += this.getRow(indent, "<span class='array-brace'>[ ]</span>" + comma, isPropertyContent);
} else {
var clpsHtml = '<span><img class="option-img" src="webjars/doc-dubbo/img/expanded.png" onClick="Formatjson.expImgClicked(this);" /></span><span class="collapsible">';
var clpsHtml = '<span><img class="option-img" src="webjars/doc-grpc/img/expanded.png" onClick="Formatjson.expImgClicked(this);" /></span><span class="collapsible">';
var annotation = '';
if(showAnnotation && isNotEmpty(keyName) && isNotEmpty(this.annotationObject[keyName])) {
annotation = '<span class="annotation">// '+this.annotationObject[keyName]+'</span>';
@@ -48,7 +48,7 @@ var Formatjson = {
if (numProps == 0) {
html += this.getRow(indent, "<span class='object-brace'>{ }</span>" + comma, isPropertyContent);
} else {
var clpsHtml = '<span><img class="option-img" src="webjars/doc-dubbo/img/expanded.png" onClick="Formatjson.expImgClicked(this);" /></span><span class="collapsible">';
var clpsHtml = '<span><img class="option-img" src="webjars/doc-grpc/img/expanded.png" onClick="Formatjson.expImgClicked(this);" /></span><span class="collapsible">';
var annotation = '';
if(showAnnotation && isNotEmpty(keyName) && isNotEmpty(this.annotationObject[keyName])) {
annotation = '<span class="annotation">// '+this.annotationObject[keyName]+'</span>';
@@ -80,10 +80,10 @@ var Formatjson = {
var container = img.parentNode.nextSibling;
if(!container) return;
var disp = "none";
var src = "webjars/doc-dubbo/img/collapsed.png";
var src = "webjars/doc-grpc/img/collapsed.png";
if(container.style.display == "none"){
disp = "inline";
src = "webjars/doc-dubbo/img/expanded.png";
src = "webjars/doc-grpc/img/expanded.png";
}
container.style.display = disp;
img.src = src;