net.gdface.service.sdk.spring.RestfulService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of faceapi-thriftservice Show documentation
The newest version!
package net.gdface.service.sdk.spring;

import org.apache.catalina.connector.Connector;
import org.apache.coyote.http11.Http11NioProtocol;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.context.embedded.EmbeddedServletContainerFactory;
import org.springframework.boot.context.embedded.tomcat.TomcatConnectorCustomizer;
import org.springframework.boot.context.embedded.tomcat.TomcatEmbeddedServletContainerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;

import net.gdface.sdk.FaceApi;
import net.gdface.sdk.FaceApiSpringController;
import net.gdface.sdk.FaceApiSpringController.InstanceSupplier;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.Contact;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

import static com.google.common.base.Preconditions.*;

import java.lang.reflect.Proxy;

/**
 * 将faceapi接口封装为RESTful接口spring web应用
 * @author guyadong
 */
@SpringBootApplication
@ComponentScan({"net.gdface.sdk"})
@EnableSwagger2
public class RestfulService {
	public static final int DEFAULT_HTTP_PORT = 8080;
	/** web服务端口 */
	private static int httpPort = DEFAULT_HTTP_PORT;
	/** tomcat连接参数 */
	private static TomcatConnectorCustomizer customizer = new ConnectorCustomizer();
	/** 是否显示在线swagger文档 */
	private static boolean swaggerEnable = true;
	private static FaceApi apiInstance;
	/** test only  */
	public static void main(String[] args) throws Exception {
		SpringApplication.run(RestfulService.class, args);
	}
	/**
	 * 启动spring boot应用
	 */
	public static void run(){
		SpringApplication.run(RestfulService.class, new String[]{});
	}
    @Bean
    public Docket faceapiApi() { 
        return new Docket(DocumentationType.SWAGGER_2)
        	.enable(swaggerEnable)
        	.apiInfo(apiInfo())
        	.select()
        	.apis(RequestHandlerSelectors.basePackage("net.gdface.sdk"))
        	.paths(PathSelectors.any())
        	.build();    
    }
    private static String getImplClass(FaceApi instance){
    	checkNotNull(instance,"instance is null");
    	if(Proxy.isProxyClass(instance.getClass())){
    		return "UNKNOW";
    	}
    	return instance.getClass().getSimpleName();
    }
    private ApiInfo apiInfo() {    	
    	FaceApi instance = checkNotNull(apiInstance,"apiInstance is uninitialized");
        return new ApiInfoBuilder()
                .title("FaceApi Document")
                .description(FaceApiSpringController.DESCRIPTION                 		
                		+ "
\nImplementation Class(算法实现): " + getImplClass(instance)
                		+ "
\nSDKVERSION(SDK版本号):" + instance.sdkVersion())
                .contact(new Contact("10km", "https://gitee.com/l0km/faceapi", "[email protected]"))
                .build();
    }
	@Bean
	public EmbeddedServletContainerFactory getTomcatEmbeddedServletContainerFactory(){
		TomcatEmbeddedServletContainerFactory tomcatFactory = new TomcatEmbeddedServletContainerFactory();  
		tomcatFactory.addConnectorCustomizers(customizer);  
		return tomcatFactory; 

	}
	/**
	 * 默认的tomcat连接参数实例
	 * @author guyadong
	 *
	 */
	private static class ConnectorCustomizer implements TomcatConnectorCustomizer  
	{  
		public void customize(Connector connector)  
		{  
			Http11NioProtocol protocol = (Http11NioProtocol) connector.getProtocolHandler();  
			protocol.setPort(httpPort);
			//设置最大连接数  
			protocol.setMaxConnections(2000);  
			//设置最大线程数  
			protocol.setMaxThreads(2000);  
			protocol.setConnectionTimeout(30000);  
		}  
	}
	/**
	 * 返回HTTP端口
	 * @return httpPort
	 */
	public static int getHttpPort() {
		return httpPort;
	}
	/**
	 * 设置HTTP端口,默认{@value #DEFAULT_HTTP_PORT}
	 * @param httpPort 要设置的 httpPort
	 */
	public static void setHttpPort(int httpPort) {
		RestfulService.httpPort = httpPort;
	}
	/**
	 * 返回tomcat参数定义接口实例
	 * @return customizer
	 */
	public static TomcatConnectorCustomizer getCustomizer() {
		return customizer;
	}
	/**
	 * 设置tomcat参数定义接口实例
	 * @param customizer 要设置的 customizer，不可为{@code null}
	 */
	public static void setCustomizer(TomcatConnectorCustomizer customizer) {
		RestfulService.customizer = checkNotNull(customizer,"customizer is null");
	}

	/**
	 * 设置{@link FaceApi}实例
	 * @param apiInstance {@link FaceApi}实例，不可为{@code null}
	 */
	public static void setInterfaceInstance(FaceApi apiInstance){
		RestfulService.apiInstance = checkNotNull(apiInstance,"apiInstance is null");
		FaceApiSpringController.setInstanceSupplier(new InstanceSupplier(){
			@Override
			public FaceApi instanceOfFaceApi() {
				return RestfulService.apiInstance;
			}});
	}
	/**
	 * @return swaggerEnable
	 */
	public static boolean isSwaggerEnable() {
		return swaggerEnable;
	}
	/**
	 * 设置是否显示在线swagger文档
	 * @param swaggerEnable 
	 */
	public static void setSwaggerEnable(boolean swaggerEnable) {
		RestfulService.swaggerEnable = swaggerEnable;
	}
}