com.vmware.vip.core.conf.LiteTomcatConnectionCustomizer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of singleton-manager-i18n-lite Show documentation

A service that provides support for Software Internationalization and Localization

There is a newer version: 0.7.12

Show newest version

/*
 * Copyright 2019-2022 VMware, Inc.
 * SPDX-License-Identifier: EPL-2.0
 */
package com.vmware.vip.core.conf;

import org.apache.catalina.connector.Connector;
import org.apache.coyote.http11.Http11NioProtocol;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.web.embedded.tomcat.TomcatConnectorCustomizer;

import com.vmware.vip.common.constants.ConstantsTomcat;

/**
 * 
 * this is custom tomcat Connector
 *
 */
public class LiteTomcatConnectionCustomizer implements TomcatConnectorCustomizer {
	private static Logger logger = LoggerFactory.getLogger(LiteTomcatConnectionCustomizer.class);
	private LiteServerProperties serverProperties;

	private String compression;

	private int compressionMinSize;

	public LiteTomcatConnectionCustomizer(LiteServerProperties prop, String compression, int compressionMinSize) {
		this.serverProperties = prop;
		this.compression = compression;
		this.compressionMinSize = compressionMinSize;
	}

	@Override
	public void customize(Connector connector) {
		
		if (this.serverProperties.getServerScheme().equalsIgnoreCase(ConstantsTomcat.HTTP_HTTPS) ||
				this.serverProperties.getServerScheme().equalsIgnoreCase(ConstantsTomcat.HTTPS_HTTP)) {
			logger.info("the tomcat support http and https protocol");
			connector.setPort(serverProperties.getHttpPort());
			connector.setAttribute("protocol", ConstantsTomcat.HTTP);
			connector.setAttribute("redirectPort", ConstantsTomcat.REDIRECT_PORT);
			connector.setAllowTrace(serverProperties.isAllowTrace());
			Http11NioProtocol protocol = (Http11NioProtocol) connector.getProtocolHandler();
			protocol.setMaxHttpHeaderSize(serverProperties.getMaxHttpHeaderSize());
			protocol.setCompression(compression);
			protocol.setCompressionMinSize(compressionMinSize);
		} else if(serverProperties.getServerScheme().equalsIgnoreCase(ConstantsTomcat.HTTP)){
			logger.info("the tomcat only support http protocol");
			connector.setPort(serverProperties.getHttpPort());
			connector.setAttribute("protocol", ConstantsTomcat.HTTP);
			connector.setAttribute("redirectPort", ConstantsTomcat.REDIRECT_PORT);
			connector.setAllowTrace(serverProperties.isAllowTrace());
			Http11NioProtocol protocol = (Http11NioProtocol) connector.getProtocolHandler();
			protocol.setMaxHttpHeaderSize(serverProperties.getMaxHttpHeaderSize());
			protocol.setCompression(compression);
			protocol.setCompressionMinSize(compressionMinSize);
		}else{
			logger.info("the tomcat only support https protocol");
			connector.setScheme(ConstantsTomcat.HTTPS);
			connector.setPort(serverProperties.getServerPort());
			connector.setSecure(true);
			Http11NioProtocol protocol = (Http11NioProtocol) connector.getProtocolHandler();
			protocol.setSSLEnabled(true);
			protocol.setKeystoreFile(serverProperties.getHttpsKeyStore());
			protocol.setKeystorePass(serverProperties.getHttpsKeyStorePassword());
			protocol.setKeystoreType(serverProperties.getHttpsKeyStoreType());
			protocol.setKeyPass(serverProperties.getHttpsKeyPassword());
			protocol.setKeyAlias(serverProperties.getHttpsKeyAlias());
			protocol.setMaxHttpHeaderSize(serverProperties.getMaxHttpHeaderSize());
			connector.setRedirectPort(ConstantsTomcat.REDIRECT_PORT);
			connector.setAllowTrace(serverProperties.isAllowTrace());
			protocol.setCompression(compression);
			protocol.setCompressionMinSize(compressionMinSize);

		}

	}
}