Spring Boot 定制与优化内置的Tomcat容器实例详解

server:
 port: 8081
 # tomcat设置
 tomcat:
 accesslog:
 # 开启日志访问
  enabled: true
 # 日志保存路径
  directory: e:/tmp/logs

@Component
public class MyEmbeddedServletContainerCustomizer implements EmbeddedServletContainerCustomizer {
  @Override
  public void customize(ConfigurableEmbeddedServletContainer container) {
    //org.springframework.boot.context.embedded.tomcat.TomcatEmbeddedServletContainerFactory 
    //说明默认是的Tomcat容器
    System.out.println(container.getClass());
    TomcatEmbeddedServletContainerFactory factory = (TomcatEmbeddedServletContainerFactory) container;
    //设置端口
    factory.setPort(8088);
    //设置Tomcat的根目录
    factory.setBaseDirectory(new File("d:/tmp/tomcat"));
    //设置访问日志存放目录
    factory.addContextValves(getLogAccessLogValue());
    //设置Tomcat线程数和连接数
    factory.addConnectorCustomizers(new MyTomcatConnectorCustomizer());
    //初始化servletContext对象
    factory.addInitializers((servletContext) -> {
      System.out.println(" = = = = 获取服务器信息 = = " + servletContext.getServerInfo());
    });
  }
  private AccessLogValve getLogAccessLogValue() {
    AccessLogValve accessLogValve = new AccessLogValve();
    accessLogValve.setDirectory("d:/tmp/tomcat/logs");
    accessLogValve.setEnabled(true);
    accessLogValve.setPattern(Constants.AccessLog.COMMON_PATTERN);
    accessLogValve.setPrefix("springboot-access-log");
    accessLogValve.setSuffix(".txt");
    return accessLogValve;
  }
}
/**
 * 定制tomcat的连接数与线程数
 */
class MyTomcatConnectorCustomizer implements TomcatConnectorCustomizer {
  @Override
  public void customize(Connector connector) {
    //连接协议 HTTP/1.1
    System.out.println(connector.getProtocol());
    //连接协议处理器 org.apache.coyote.http11.Http11NioProtocol
    System.out.println(connector.getProtocolHandler().getClass());
    //Http11NioProtocol
    Http11NioProtocol protocolHandler = (Http11NioProtocol) connector.getProtocolHandler();
    // 设置最大连接数
    protocolHandler.setMaxConnections(2000);
    // 设置最大线程数
    protocolHandler.setMaxThreads(500);
  }
}

@SpringBootConfiguration
public class WebServerConfiguration {
  @Bean
  public EmbeddedServletContainerFactory embeddedServletContainerFactory() {
    TomcatEmbeddedServletContainerFactory factory = new TomcatEmbeddedServletContainerFactory();
    //设置端口
    factory.setPort(8089);
    //设置404错误界面
    factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/404.html"));
    //设置在容器初始化的时候触发
    factory.addInitializers((servletContext) -> {
      System.out.println(" = = = = 获取服务器信息 = = " + servletContext.getServerInfo());
    });
    //设置最大连接数和最大线程数
    factory.addConnectorCustomizers((connector) -> {
      Http11NioProtocol protocolHandler = (Http11NioProtocol) connector.getProtocolHandler();
      protocolHandler.setMaxConnections(2000);
      protocolHandler.setMaxThreads(500);
    });
    //设置访问日志记录文件的目录
    factory.addContextValves(getLogAccessLogValue());
    return factory;
  }
  private AccessLogValve getLogAccessLogValue() {
    AccessLogValve accessLogValve = new AccessLogValve();
    accessLogValve.setDirectory("d:/tmp/logs");
    accessLogValve.setEnabled(true);
    accessLogValve.setPattern(Constants.AccessLog.COMMON_PATTERN);
    accessLogValve.setPrefix("SpringBoot-Access-Log");
    accessLogValve.setSuffix(".txt");
    return accessLogValve;
  }
}