All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.yx.http.HttpPlugin Maven / Gradle / Ivy

There is a newer version: 4.0.2
Show newest version
/**
 * Copyright (C) 2016 - 2030 youtongluan.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * 		http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.yx.http;

import java.lang.reflect.Constructor;
import java.util.ArrayList;
import java.util.List;

import org.slf4j.Logger;
import org.yx.annotation.Bean;
import org.yx.bean.IOC;
import org.yx.bean.Loader;
import org.yx.bean.Plugin;
import org.yx.common.KeyValuePair;
import org.yx.common.Lifecycle;
import org.yx.conf.AppInfo;
import org.yx.conf.Const;
import org.yx.exception.SumkException;
import org.yx.http.act.HttpActionNode;
import org.yx.http.act.HttpActions;
import org.yx.http.handler.HttpHandler;
import org.yx.http.handler.HttpHandlerChain;
import org.yx.http.handler.RestType;
import org.yx.http.invoke.WebHandler;
import org.yx.http.kit.HttpSettings;
import org.yx.http.start.WebAnnotationResolver;
import org.yx.http.user.WebSessions;
import org.yx.log.Logs;
import org.yx.main.StartConstants;
import org.yx.main.StartContext;
import org.yx.main.SumkServer;
import org.yx.util.ExceptionUtil;
import org.yx.util.StringUtil;

@Bean
public class HttpPlugin implements Plugin {

	protected Lifecycle server;

	@Override
	public void stop() {
		if (this.server != null) {
			server.stop();
			this.server = null;
		}
	}

	@Override
	public int order() {
		return 10100;
	}

	protected void resolveWebAnnotation(List beans) {
		WebAnnotationResolver factory = new WebAnnotationResolver();
		try {
			List> infos = new ArrayList<>(100);
			for (Object bean : beans) {
				List> acts = factory.resolve(bean);
				if (acts != null && acts.size() > 0) {
					infos.addAll(acts);
				}
			}
			HttpActions.init(infos);
		} catch (Exception e) {
			throw SumkException.wrap(e);
		}
	}

	@Override
	public void startAsync() {
		int port = StartContext.httpPort();
		if (!this.isHttpEnable() || port < 1) {
			return;
		}
		try {
			HttpHeaderName.init();
			HttpSettings.init();
			List beans = StartContext.inst().getBeans();
			resolveWebAnnotation(beans);
			WebHandler.init();
			this.buildHttpHandlers();
			this.initServer(port);
		} catch (Exception e) {
			throw ExceptionUtil.toRuntimeException(e);
		}
	}

	protected void initServer(int port) throws Exception {
		String nojetty = StartConstants.NOJETTY;
		if (StartContext.inst().get(nojetty) != null || AppInfo.getBoolean(nojetty, false)) {
			return;
		}

		String httpServerClass = StringUtil.isEmpty(AppInfo.get(Const.KEY_STORE_PATH)) ? "JettyServer"
				: "JettyHttpsServer";
		String hs = AppInfo.get("sumk.http.starter.class", "org.yx.http.start." + httpServerClass);
		Class httpClz = Loader.loadClassExactly(hs);
		Constructor c = httpClz.getConstructor(int.class);
		this.server = (Lifecycle) c.newInstance(port);
	}

	protected void buildHttpHandlers() {
		List handlers = IOC.getBeans(HttpHandler.class);
		List restHandlers = new ArrayList<>(handlers.size());
		List uploadHandlers = new ArrayList<>(handlers.size());
		for (HttpHandler h : handlers) {
			if (h.supportRestType(RestType.PLAIN)) {
				restHandlers.add(h);
			}
			if (h.supportRestType(RestType.MULTI_PART)) {
				uploadHandlers.add(h);
			}
		}
		HttpHandlerChain.rest.setHandlers(restHandlers);
		Logger logger = Logs.http();
		if (logger.isDebugEnabled()) {
			logger.debug("rest  handlers:{}", this.buildString(restHandlers));
		}
		if (HttpSettings.isUploadEnable()) {
			HttpHandlerChain.multipart.setHandlers(uploadHandlers);
			if (logger.isDebugEnabled()) {
				logger.debug("upload handlers:{}", this.buildString(uploadHandlers));
			}
		}
	}

	protected boolean isHttpEnable() {
		if (!SumkServer.isHttpEnable()) {
			return false;
		}
		try {
			Loader.loadClassExactly("javax.servlet.http.HttpServlet");
		} catch (Exception e) {
			Logs.http().warn("javax-servlet-api-**.jar is not imported");
			return false;
		}
		return true;
	}

	@Override
	public void afterStarted() {
		Lifecycle server = this.server;
		if (!SumkServer.isHttpEnable() || server == null) {
			return;
		}
		WebSessions.initSession();
		server.start();
	}

	protected String buildString(List hs) {
		StringBuilder sb = new StringBuilder();
		for (HttpHandler h : hs) {
			sb.append(" ").append(h.getClass().getSimpleName()).append("(").append(h.order()).append(")");
		}
		return sb.toString();
	}

	public Lifecycle getServer() {
		return server;
	}
}