org.docshare.boot.ServerMain Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yangmvc-boot Show documentation
YangMVC is an efficient, fast, convenient, easy to learn, easy to use MVC framework and ORM framework. Use yangmvc-boot is something like spring-boot . Call ServerMain.main(args); in your main method can start a server.
There is a newer version: 2023.06.19
Show newest version
package org.docshare.boot;




import java.io.IOException;
import java.util.Scanner;

import org.docshare.log.Log;
import org.docshare.mvc.Config;
import org.docshare.util.FileTool;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.ContextHandler;
import org.eclipse.jetty.server.handler.ContextHandlerCollection;
import org.eclipse.jetty.server.handler.DefaultHandler;
import org.eclipse.jetty.server.handler.HandlerCollection;
import org.eclipse.jetty.server.handler.ResourceHandler;
import org.eclipse.jetty.server.session.*;


public class ServerMain {
	
	private static int port = 1985;
	static void parseParam(String[] args){
		Log.i("Usage: java -jar xxx.jar [port] [nowebsocket] ");
		Log.i("   eg: java -jar aa_boot.jar ");
		Log.i("   eg: java -jar aa_boot.jar 80 nowebsocket ");
		Log.i("   That means listen at port 80 and use no websocket");
		if(args.length>=1 ){
			port = Integer.parseInt(args[0]);
		}
		if(args.length >= 2  && "nowebsocket".equals(args[1])){
			supportWebSocket = false;
		}
	}
	public static void main(String[] args) {
		Log.i("Server start .........");
		parseParam(args);
		if(true){
			try {
				start();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}
    public static ContextHandler contextHandler = null;
    public static boolean supportWebSocket = true;
    public static boolean useSession = true;
	public static boolean useRedis = true;
	private static SessionHandler sessionHandler=null;
	public static SessionHandler getSessionHandler(){
		return sessionHandler;
	}

	private static void applyRedis(){
		Log.i("use redis sessionCache");
		NullSessionCache dsc = new NullSessionCache(sessionHandler);
		String redis_host = Config.getProperty("redis_host", "localhost");
		String redis_port = Config.getProperty("redis_port","6379");
		SessionDataMap sessionDataMap = null;
		try {
			Class redisClass = Class.forName("org.docshare.boot.RedisSessionDataMap");
			sessionDataMap = (SessionDataMap) redisClass.newInstance();

			dsc.setSessionDataStore(new CachingSessionDataStore(sessionDataMap,
					new NullSessionDataStore()));
			sessionHandler.setSessionCache(dsc);

		} catch (Exception e) {
			Log.e("Unable to load redis class" +e.getMessage());
		}

	}
	public static void start() throws Exception{
        Server server = new Server(port);
        String url = "http://127.0.0.1";
        if(port != 80){
            url = String.format("http://127.0.0.1:%d",port);
        }
        Config.loadProperties("/web.properties");
		try{
			org.eclipse.jetty.util.log.Log.setLog(new YangLogger());
	        
	        HandlerCollection collection =new HandlerCollection(); 

	        contextHandler = new ContextHandler();
	        
	        contextHandler.setContextPath("/");
	        contextHandler.setResourceBase("./WebRoot");
	        contextHandler.setClassLoader(Thread.currentThread().getContextClassLoader());
	        //contextHandler.setHandler(YangHandle.getIns(server));
	        if(useSession){
	        	sessionHandler=new SessionHandler();
				//----------------------------
				if(useRedis){
					applyRedis();
				}

				//----------------------------
	        	collection.addHandler(sessionHandler);
	        }
	        collection.addHandler(YangHandle.getIns(server));
	        collection.addHandler(new ResourceHandler());
	        collection.addHandler(new DefaultHandler());
	        
	        
	        
	        contextHandler.setHandler(collection);
	        if(supportWebSocket){
		        //support websocket 
		        ContextHandler wsHandler = new ContextHandler();
		        wsHandler.setContextPath("/ws");
		        wsHandler.setHandler(new MyWebSocketHandler());
				
				
				ContextHandlerCollection final_handler = new ContextHandlerCollection();
		        
				final_handler.addHandler(wsHandler);
		        final_handler.addHandler(contextHandler);
		        
		        server.setHandler(final_handler);
	        }else{
	        	server.setHandler(contextHandler);
	        }
	        server.start();
	        
	        Log.i("服务器已经开启 Server is Started");
	        Log.i("please visit "+url);
	        IpHelper.showIP();
	        readConsole(url);
	        
	        //server.join(); 
		}catch (IOException e) {
			Log.e("绑定端口号错误，一般是是因为端口号被其他应用占用了。 binding port error :"+port);
			String pid = IpHelper.showPortUsed(port);
			//重试！！
			if(System.getProperty("os.name").contains("Win")){
				try {
					IpHelper.killPID(pid);
					Thread.sleep(3000);
					server.start();
			        readConsole(url);
				} catch (Exception e1) {
					Log.e(e1);
				}
			}
		}
	}
	
	private static void readConsole(String url) {
		Scanner scanner  = null;
		try {
			scanner= new Scanner(System.in);
			while (true) {
				Log.i("每按一次回车键打开一次浏览器,Press Enter key to open browser");
				String ret = scanner.nextLine();
				if("exit".equals(ret))break;
				Runtime.getRuntime().exec("rundll32 url.dll,FileProtocolHandler " + url);
			}
		} catch (Exception e) {
			//Log.e(e);
		}finally {
			FileTool.safelyClose(scanner);
		}
	}

}