org.tio.http.common.HttpConfig Java Examples

/**
 * 
 * @param httpConfig
 * @param requestHandler
 * @param tioExecutor
 * @param groupExecutor
 * @author tanyaowu
 */
public HttpServerStarter(HttpConfig httpConfig, HttpRequestHandler requestHandler, SynThreadPoolExecutor tioExecutor, ThreadPoolExecutor groupExecutor) {
	//		preAccessFileType.add("css");
	//		preAccessFileType.add("js");
	//		preAccessFileType.add("jsp");
	preAccessFileType.add("html");
	preAccessFileType.add("ftl");
	//		preAccessFileType.add("xml");
	//		preAccessFileType.add("htm");

	if (tioExecutor == null) {
		tioExecutor = Threads.getTioExecutor();
	}

	if (groupExecutor == null) {
		groupExecutor = Threads.getGroupExecutor();
	}

	init(httpConfig, requestHandler, tioExecutor, groupExecutor);
}
 

/**
 * @param args
 * @author tanyaowu
 * @throws IOException
 */
public static void main(String[] args) throws Exception {
	httpConfig = new HttpConfig(8080, null, null, null);
	httpConfig.setUseSession(false);
	httpConfig.setWelcomeFile(null);
	httpConfig.setCheckHost(false);
	httpConfig.setCompatible1_0(false);

	Routes routes = new Routes(TestController.class);

	requestHandler = new DefaultHttpRequestHandler(httpConfig, routes);
	requestHandler.setCompatibilityAssignment(false);
	httpServerStarter = new HttpServerStarter(httpConfig, requestHandler);
	serverTioConfig = httpServerStarter.getServerTioConfig();
	serverTioConfig.statOn = false;
	httpServerStarter.start();
}
 

private void init(HttpConfig httpConfig, Routes routes) throws Exception {
	if (httpConfig == null) {
		throw new RuntimeException("httpConfig can not be null");
	}
	this.contextPath = httpConfig.getContextPath();
	this.suffix = httpConfig.getSuffix();

	if (StrUtil.isNotBlank(contextPath)) {
		contextPathLength = contextPath.length();
	}
	if (StrUtil.isNotBlank(suffix)) {
		suffixLength = suffix.length();
	}

	this.httpConfig = httpConfig;

	if (httpConfig.getMaxLiveTimeOfStaticRes() > 0) {
		staticResCache = CaffeineCache.register(STATIC_RES_CONTENT_CACHENAME, (long) httpConfig.getMaxLiveTimeOfStaticRes(), null);
	}

	sessionRateLimiterCache = CaffeineCache.register(SESSIONRATELIMITER_CACHENAME, 60 * 1L, null);

	this.routes = routes;

	this.monitorFileChanged();
}
 

/**
 * 获取真实ip
 * @param channelContext
 * @param httpConfig
 * @param httpHeaders
 * @return
 * @author tanyaowu
 */
public static String getRealIp(ChannelContext channelContext, HttpConfig httpConfig, Map<String, String> httpHeaders) {
	if (httpConfig == null) {
		return channelContext.getClientNode().getIp();
	}

	if (httpConfig.isProxied()) {
		String headerName = null;
		String ip = null;
		for (String name : HEADER_NAMES_FOR_REALIP) {
			headerName = name;
			ip = httpHeaders.get(headerName);

			if (StrUtil.isNotBlank(ip) && !"unknown".equalsIgnoreCase(ip)) {
				break;
			}
		}

		if (StrUtil.isBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
			headerName = null;
			ip = channelContext.getClientNode().getIp();
		}

		if (ip.contains(",")) {
			if (log.isInfoEnabled()) {
				log.info("ip[{}], header name:{}", ip, headerName);

			}
			ip = ip.split(",")[0].trim();
		}
		return ip;
	} else {
		return channelContext.getClientNode().getIp();
	}
}
 

@SuppressWarnings("unchecked")
public <T extends Serializable> T getAttribute(String key, Class<T> clazz, T defaultObj, HttpConfig httpConfig) {
	T t = (T) data.get(key);
	if (t == null) {
		log.warn("key【{}】'value in session is null", key);
		if (defaultObj != null) {
			setAttribute(key, defaultObj, httpConfig);
		}
		return defaultObj;
	}
	return t;
}
 

/**
	 * 
	 * @param httpConfig
	 * @param root
	 * @return
	 * @throws IOException
	 */
	private Configuration createConfiguration(HttpConfig httpConfig, String root) throws IOException {
		if (httpConfig.getPageRoot() == null) {
			return null;
		}

		if (configurationCreater != null) {
			return configurationCreater.createConfiguration(httpConfig, root);
		}

		Configuration cfg = new Configuration(Configuration.getVersion());

		if (httpConfig.isPageInClasspath()) {
			cfg.setClassForTemplateLoading(this.getClass(), "/" + root/**.substring("classpath:".length())*/
			);
			//cfg.setClassForTemplateLoading(FreemarkerUtil.class, "/template");
		} else {
			cfg.setDirectoryForTemplateLoading(new File(root));
		}
		cfg.setDefaultEncoding(httpConfig.getCharset());
		//		cfg.setTemplateExceptionHandler(TemplateExceptionHandler.RETHROW_HANDLER);
		cfg.setLogTemplateExceptions(false);
		cfg.setWrapUncheckedExceptions(true);
		cfg.setTemplateExceptionHandler(ShortMessageTemplateExceptionHandler.me);
		cfg.setLocale(Locale.SIMPLIFIED_CHINESE);
		cfg.setNumberFormat("#");
//		cfg.setClassicCompatible(true);
		return cfg;
	}
 

public FreemarkerConfig(HttpConfig httpConfig, ModelGenerator modelGenerator, String[] suffixes, ConfigurationCreater configurationCreater) throws IOException {
	super();
	this.configurationCreater = configurationCreater;

	String pageRoot = httpConfig.getPageRoot();
	if (pageRoot == null) {
		throw new IOException("没有配置pageRoot");
	}

	httpConfig.setFreemarkerConfig(this);

	this.httpConfig = httpConfig;
	this.modelGenerator = modelGenerator;
	this.setSuffixes(suffixes);

	this.configuration = createConfiguration(httpConfig, pageRoot);

	Map<String, String> domainPageMap = httpConfig.getDomainPageMap();
	if (domainPageMap != null && domainPageMap.size() > 0) {
		Set<Entry<String, String>> set = domainPageMap.entrySet();
		for (Entry<String, String> entry : set) {
			String domain = entry.getKey();
			String file = entry.getValue();
			//				Configuration cfg = createConfiguration(httpConfig, file);
			addDomainConfiguration(domain, file);
		}
	}
}
 

private void init(HttpConfig httpConfig, HttpRequestHandler requestHandler, SynThreadPoolExecutor tioExecutor, ThreadPoolExecutor groupExecutor) {
	String system_timer_period = System.getProperty("tio.system.timer.period");
	if (StrUtil.isBlank(system_timer_period)) {
		System.setProperty("tio.system.timer.period", "50");
	}

	this.httpConfig = httpConfig;
	this.httpRequestHandler = requestHandler;
	httpConfig.setHttpRequestHandler(this.httpRequestHandler);
	this.httpServerAioHandler = new HttpServerAioHandler(httpConfig, requestHandler);
	httpServerAioListener = new HttpServerAioListener();
	String name = httpConfig.getName();
	if (StrUtil.isBlank(name)) {
		name = "Tio Http Server";
	}
	serverTioConfig = new ServerTioConfig(name, httpServerAioHandler, httpServerAioListener, tioExecutor, groupExecutor);
	serverTioConfig.setHeartbeatTimeout(1000 * 20);
	serverTioConfig.setShortConnection(true);
	serverTioConfig.setReadBufferSize(TcpConst.MAX_DATA_LENGTH);
	//		serverTioConfig.setAttribute(TioConfigKey.HTTP_SERVER_CONFIG, httpConfig);
	serverTioConfig.setAttribute(TioConfigKey.HTTP_REQ_HANDLER, this.httpRequestHandler);

	tioServer = new TioServer(serverTioConfig);

	HttpUuid imTioUuid = new HttpUuid();
	serverTioConfig.setTioUuid(imTioUuid);
}
 

public static String getSessionId(HttpRequest request) {
	String sessionId = request.getString(org.tio.http.common.HttpConfig.TIO_HTTP_SESSIONID);
	if (StrUtil.isNotBlank(sessionId)) {
		return sessionId;
	}

	Cookie cookie = getSessionCookie(request, request.httpConfig);
	if (cookie != null) {
		return cookie.getValue();
	}

	return null;
}
 

/**
 * @return the httpConfig
 */
public HttpConfig getHttpConfig() {
	return httpConfig;
}
 

public void setHttpConfig(HttpConfig httpConfig) {
	this.httpConfig = httpConfig;
}
 

public void update(HttpConfig httpConfig) {
	httpConfig.getSessionStore().put(id, this);
}
 

/**
 * @return
 * @author tanyaowu
 */
@Override
public String sessionId(HttpConfig httpConfig, HttpRequest request) {
	return String.valueOf(snowflake.nextId());
}
 

/**
 * @return
 * @author tanyaowu
 */
@Override
public String sessionId(HttpConfig httpConfig, HttpRequest request) {
	return UUID.randomUUID().toString().replace("-", "");
}
 

/**
 * @param httpConfig the httpConfig to set
 */
public void setHttpConfig(HttpConfig httpConfig) {
	this.httpConfig = httpConfig;
}
 

/**
 * @return the httpConfig
 */
public HttpConfig getHttpConfig() {
	return httpConfig;
}
 

@Override
public HttpConfig getHttpConfig(HttpRequest request) {
	HttpRequestHandler httpRequestHandler = getHttpRequestHandler(request);
	return httpRequestHandler.getHttpConfig(request);
}
 

public static Cookie getSessionCookie(HttpRequest request, HttpConfig httpConfig) {
	Cookie sessionCookie = request.getCookie(httpConfig.getSessionCookieName());
	return sessionCookie;
}
 

/**
 * @param httpConfig the httpConfig to set
 */
public void setHttpConfig(HttpConfig httpConfig) {
	this.httpConfig = httpConfig;
}
 

/**
 * @return the httpConfig
 */
public HttpConfig getHttpConfig(HttpRequest request) {
	return httpConfig;
}
 

/**
 * 
 * @param httpConfig
 * @param scanRootClasse
 * @throws Exception 
 */
public DefaultHttpRequestHandler(HttpConfig httpConfig, Class<?> scanRootClasse) throws Exception {
	this(httpConfig, new Class<?>[] { scanRootClasse });
}
 

/**
 * 
 * @return
 * @author tanyaowu
 */
public HttpConfig getHttpConfig(HttpRequest request);
 

/**
 * 
 * @param httpConfig
 * @param scanRootClasse
 * @param controllerFactory
 * @throws Exception 
 */
public DefaultHttpRequestHandler(HttpConfig httpConfig, Class<?> scanRootClasse, ControllerFactory controllerFactory) throws Exception {
	this(httpConfig, new Class<?>[] { scanRootClasse }, controllerFactory);
}
 

/**
 * 
 * @param httpConfig
 * @param scanRootClasses
 * @throws Exception 
 */
public DefaultHttpRequestHandler(HttpConfig httpConfig, Class<?>[] scanRootClasses) throws Exception {
	this(httpConfig, scanRootClasses, null);
}
 

/**
 * 设置会话属性
 * @param key
 * @param value
 * @param httpConfig
 * @author tanyaowu
 */
public void setAttribute(String key, Serializable value, HttpConfig httpConfig) {
	data.put(key, value);
	update(httpConfig);
}
 

/**
 *
 * @param key
 * @param httpConfig
 * @author tanyaowu
 */
public void removeAttribute(String key, HttpConfig httpConfig) {
	data.remove(key);
	update(httpConfig);
}
 

/**
 * 
 * @param httpConfig
 * @param scanRootClasses
 * @param controllerFactory
 * @throws Exception 
 */
public DefaultHttpRequestHandler(HttpConfig httpConfig, Class<?>[] scanRootClasses, ControllerFactory controllerFactory) throws Exception {
	Routes routes = new Routes(scanRootClasses, controllerFactory);
	init(httpConfig, routes);
}
 

/**
 * 清空所有属性
 * @param httpConfig
 * @author tanyaowu
 */
public void clear(HttpConfig httpConfig) {
	data.clear();
	update(httpConfig);
}
 

/**
 * 
 * @param httpConfig
 * @param scanPackage
 * @throws Exception 
 */
public DefaultHttpRequestHandler(HttpConfig httpConfig, String scanPackage) throws Exception {
	this(httpConfig, scanPackage, null);
}
 

/**
 * 
 * @param httpConfig
 * @param scanPackage
 * @param controllerFactory
 * @throws Exception 
 */
public DefaultHttpRequestHandler(HttpConfig httpConfig, String scanPackage, ControllerFactory controllerFactory) throws Exception {
	this(httpConfig, new String[] { scanPackage }, controllerFactory);
}