com.textrecruit.ustackserver.Main Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of textrecruit-ustack-server Show documentation
TextRecruit UStack Server
There is a newer version: 1.0.31
Show newest version
package com.textrecruit.ustackserver;

import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.net.InetSocketAddress;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicInteger;

import com.textrecruit.ustackserver.server.ServerFactory;
import com.textrecruit.ustackserverapi.util.APIPerms;
import org.apache.log4j.Logger;
import org.apache.log4j.xml.DOMConfigurator;
import org.bson.types.ObjectId;
import org.jboss.netty.bootstrap.ServerBootstrap;
import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory;
import org.jboss.netty.handler.execution.ExecutionHandler;
import org.jboss.netty.handler.execution.OrderedMemoryAwareThreadPoolExecutor;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonParseException;
import com.mongodb.DBObject;
import com.textrecruit.ustack.aaa.ResourceDefinition;
import com.textrecruit.ustack.aaa.ResourceLink;
import com.textrecruit.ustack.aaa.RoleDefinition;
import com.textrecruit.ustack.data.APIClient;

public class Main {

    static Logger           		logger               	= Logger.getLogger(Main.class);

	public static final Gson gson = new GsonBuilder()
			.registerTypeAdapter(ObjectId.class, ObjectIdDeserializer.INSTANCE)
			.registerTypeAdapter(JsonElement.class, JsonElementDeserializer.INSTANCE)
			.setDateFormat("yyyyMMddHHmmssZ").create();

	public enum ObjectIdDeserializer implements JsonDeserializer {
		INSTANCE;
	
		public ObjectId deserialize(JsonElement json, Type typeOfT,
			JsonDeserializationContext context) throws JsonParseException {
			return new ObjectId(json.getAsString());
		}
	}
	
	public enum JsonElementDeserializer implements
		JsonDeserializer {
		INSTANCE;
	
		public JsonElement deserialize(JsonElement json, Type typeOfT,
			JsonDeserializationContext context) throws JsonParseException {
			return json;
		}
	}

    public static void main(String[] args) throws Exception {

		DOMConfigurator.configure("log4j.xml");
    	int port = 8081;
    	
    	if (args.length >= 1)
    		port = Integer.valueOf(args[1]);
    	
    	Main main = new Main(args[0], port);
    	
		main.run();
    }

    private int port;
    private String apiVersion;
    
    public String getAPIVersion() {
    	return apiVersion;
    }
    
    public Main(String code, int p) {

    	port = p;
    	
    	try {
    		
			Class apiClass = Class.forName(code);
			Object apiObj = apiClass.newInstance();
				
    		Method m = apiClass.getMethod("setup");
    		m.invoke(apiObj, (Object[])null);

    		m = apiClass.getMethod("getAPIVersion");
    		
    		Object o = m.invoke(apiObj);
    		apiVersion = (String)o;

    	} catch (Exception e) {
    		logger.warn("Failed to load API subset", e);
    	}
    	
    }
    	
    public static ResourceDefinition setupAPIResource() throws Exception
    {
    	ResourceDefinition rs = ResourceDefinition.getByInternalName("API");
		
		if (rs == null)
		{
			rs = ResourceDefinition.createResource("API", ResourceDefinition.TYPE_APIACCESS);
			rs.setInternalName("API");
			
			RoleDefinition role = new RoleDefinition("API Manager");
			role.addPermission(APIPerms.APIClientManager.getPermission());
			role.addPermission(APIPerms.APIResourceManager.getPermission());
			role.addPermission(APIPerms.APIRoleManager.getPermission());
			rs.addRole(role);
			rs.save("Initial Run");
		}

		return rs;
    }
    
	public void run() throws Exception {
		
		// check if we have at least one API client
		long clientCnt = APIClient.getDBCollection().count();
		if (clientCnt == 0)
		{
			// create initial API client with rights to manage resources and other API clients
			APIClient client = APIClient.createAPI("Initial Run", "admin-api");
			
			// add rights
			ResourceDefinition rs = setupAPIResource();
			
			ResourceLink rl = new ResourceLink(rs, "API Manager");
			client.addResourceLink("Initial", rl);
			client.save("Initial Run");
			
			DBObject k = (DBObject)client.getAPIKeys().get(0);
			
			// output key to log for user to grab
 			logger.info(String.format("Initial API Client/Key: %s/%s", client.getClientId(), client.getKey((String)k.get("uid"))));
		}
		
		logger.info("Staring client server on port " + port);
		
		ExecutionHandler executionHandler = new ExecutionHandler(
							new OrderedMemoryAwareThreadPoolExecutor(
									Integer.valueOf(System.getProperty("UStackServer.Threads", "48")), 
									4048576, 4048576));
		
		ServerBootstrap bootstrap = new ServerBootstrap(
											new NioServerSocketChannelFactory(
													Executors.newCachedThreadPool(new NamedThreadFactory("netty-boss")),
													Executors.newCachedThreadPool(new NamedThreadFactory("netty-worker"))));
				
		bootstrap.setOption("backlog", 1000);
		bootstrap.setOption("reuseAddress", true);
//		bootstrap.setOption("keepAlive", true);
//		bootstrap.setOption("child.keepAlive", true);

		// Set up the event pipeline factory.
		bootstrap.setPipelineFactory(new ServerFactory(executionHandler));
		
		// Bind and start to accept incoming connections.
		bootstrap.bind(new InetSocketAddress(port));
	}
	
	public class NamedThreadFactory implements ThreadFactory {
		  private final String baseName;
		  private final AtomicInteger threadNum = new AtomicInteger(0); 

		  public NamedThreadFactory(String baseName) {
		    this.baseName = baseName;
		  }

		  @Override
		  public synchronized Thread newThread(Runnable r) {
		    Thread t = Executors.defaultThreadFactory().newThread(r);
		    
		    t.setName(baseName + "-" + threadNum.getAndIncrement());
		    
		    return t; 
		  }
		}
}