Compass/compass/public/assets/highcharts/exporting-server/java/highcharts-export/highcharts-export-convert/src/main/java/com/highcharts/export/pool/ServerObjectFactory.java

package com.highcharts.export.pool;

import java.nio.file.*;
import java.util.HashMap;
import java.util.Map;

import javax.annotation.PostConstruct;

import org.apache.log4j.Logger;

import com.highcharts.export.server.Server;
import com.highcharts.export.server.ServerState;
import com.highcharts.export.util.TempDir;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.TreeMap;

import org.apache.commons.io.IOUtils;
import org.springframework.core.io.ClassPathResource;

public class ServerObjectFactory implements ObjectFactory<Server> {

	public String exec;
	public String script;
	private String host;
	private int basePort;
	private int readTimeout;
	private int poolSize;
	private int connectTimeout;
	private int maxTimeout;
	private static TreeMap<Integer, PortStatus> portUsage = new TreeMap<>();
	protected static Logger logger = Logger.getLogger("pool");

	private enum PortStatus {
        BUSY,
        FREE;
	}

	@Override
	public Server create() {
		logger.debug("in makeObject, " + exec + ", " +  script + ", " +  host);
		Integer port = this.getAvailablePort();
		String separator = FileSystems.getDefault().getSeparator();

        if (script.isEmpty()) {
            // use the bundled highcharts-convert.js script
            script = TempDir.getPhantomJsDir().toAbsolutePath().toString() + separator + "highcharts-convert.js";
        }
        Server server = new Server(exec, script, host, port, connectTimeout, readTimeout, maxTimeout);
		portUsage.put(port, PortStatus.BUSY);
		return server;
	}

	@Override
	public boolean validate(Server server) {
		boolean isValid = false;
		try {
			if(server.getState() != ServerState.IDLE) {
				logger.debug("server didn\'t pass validation");
				return false;
			}
			String result = server.request("{\"status\":\"isok\"}");
			if(result.indexOf("OK") > -1) {
				isValid = true;
				logger.debug("server passed validation");
			} else {
				logger.debug("server didn\'t pass validation");
			}
		} catch (Exception e) {
			logger.error("Error while validating object in Pool: " + e.getMessage());
		}
		return isValid;
	}

	@Override
	public void destroy(Server server) {
		ServerObjectFactory.releasePort(server.getPort());
		server.cleanup();
	}

	@Override
	public void activate(Server server) {
		server.setState(ServerState.ACTIVE);
	}

	@Override
	public void passivate(Server server) {
		server.setState(ServerState.IDLE);
	}

	public static void releasePort(Integer port) {
		logger.debug("Releasing port " + port);
		portUsage.put(port, PortStatus.FREE);
	}

	public Integer getAvailablePort() {

		/* first we check within the defined port range from baseport
		* up to baseport + poolsize
		*/
		int port = basePort;
		for (; port < basePort + poolSize; port++) {

			if (portUsage.containsKey(port)) {
				if (portUsage.get(port) == PortStatus.FREE) {
					return port;
				}
			} else {
				// doesn't exist any longer, but is within the valid port range
				return port;
			}

		}

		// at this point there is no free port, we have to look outside of the valid port range
		logger.debug("Nothing free in Portusage " + portUsage.toString());
		return portUsage.lastKey() + 1;
	}

	/*Getters and Setters*/

	public String getExec() {
		return exec;
	}

	public void setExec(String exec) {
		this.exec = exec;
	}

	public String getScript() {
		return script;
	}

	public void setScript(String script) {
		this.script = script;
	}

	public String getHost() {
		return host;
	}

	public void setHost(String host) {
		this.host = host;
	}

	public int getBasePort() {
		return basePort;
	}

	public void setBasePort(int basePort) {
		this.basePort = basePort;
	}

	public int getReadTimeout() {
		return readTimeout;
	}

	public void setReadTimeout(int readTimeout) {
		this.readTimeout = readTimeout;
	}

	public int getConnectTimeout() {
		return connectTimeout;
	}

	public void setConnectTimeout(int connectTimeout) {
		this.connectTimeout = connectTimeout;
	}

	public int getMaxTimeout() {
		return maxTimeout;
	}

	public void setMaxTimeout(int maxTimeout) {
		this.maxTimeout = maxTimeout;
	}

	public int getPoolSize() {
		return poolSize;
	}

	public void setPoolSize(int poolSize) {
		this.poolSize = poolSize;
	}

	@PostConstruct
	public void afterBeanInit() {
		
		URL u = getClass().getProtectionDomain().getCodeSource().getLocation();
		URLClassLoader jarLoader = new URLClassLoader(new URL[]{u}, Thread.currentThread().getContextClassLoader());
		String filenames[] = new String[] {"highcharts-convert.js",
				"jquery.1.9.1.min.js",
				"highcharts.js",
				"highstock.js",
				"highcharts-more.js",
				"data.js",
				"drilldown.js",
				"funnel.js",
				"heatmap.js",
				"highcharts-3d.js",
				"no-data-to-display.js",
				"solid-gauge.js",
				"map.js",
				"broken-axis.js",
				"treemap.js"};
		
		for (String filename : filenames) {
		
			ClassPathResource resource = new ClassPathResource("phantomjs/" + filename, jarLoader);
			if (resource.exists()) {
				Path path = Paths.get(TempDir.getPhantomJsDir().toString(), filename);
				File file;
				try {
					file = Files.createFile(path).toFile();
					file.deleteOnExit();
					try (InputStream in = resource.getInputStream();
					     OutputStream out=new FileOutputStream(file))
			        {
					    IOUtils.copy(in, out);
			        }
				} catch (IOException ioex) {
					logger.error("Error while setting up phantomjs environment: " + ioex.getMessage());
				}
			} else {
				logger.debug("Copy javascript file to temp folder, resource doesn't exist: " + filename);
			}
		}
		
	}



}