com.alogic.together3.catalog.FromClasspath Maven / Gradle / Ivy

Go to download
package com.alogic.together3.catalog;

import com.alogic.together3.service.TogetherServant;
import com.alogic.together3.TogetherServiceDescription;
import com.anysoft.util.*;
import com.logicbus.models.catalog.CatalogNode;
import com.logicbus.models.catalog.Path;
import com.logicbus.models.servant.ServiceDescription;
import com.logicbus.models.servant.impl.ServantCatalogNodeImpl;
import com.logicbus.models.servant.impl.XMLResourceServantCatalog;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.Enumeration;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

/**
 * 从Class路径装入服务目录
 * 
 * @author yyduan
 *
 * @since 1.6.17.2
 */
public class FromClasspath extends XMLResourceServantCatalog{
	
	/**
	 * 缺省的服务实现类
	 */
	private String servant = TogetherServant.class.getName();

	protected String suffix = ".xml";
	
	@Override
	public void configure(Properties p) {
		servant = PropertiesConstants.getString(p,"servant",servant,true);
		suffix = PropertiesConstants.getString(p,"suffix",suffix,true);
		loadDocument(p);
	}	

	@Override
	protected CatalogNode createCatalogNode(Path _path,Object _data){
		ServantCatalogNodeImpl catalogNode = new ServantCatalogNodeImpl(_path,_data);
		
		Element node = (Element)_data;
		
		String home = node.getAttribute("home");
		String clazz = node.getAttribute("bootstrap");
		if (StringUtils.isNotEmpty(home) && StringUtils.isNotEmpty(clazz)){
			ClassLoader cl = Settings.getClassLoader();			
			try {
				Class bootstrap = cl.loadClass(clazz);
				scanResource(new XmlElementProperties(node,Settings.get()),catalogNode,home,bootstrap);
			}catch (Exception ex){
				logger.error(String.format("Failed to scan resource:%s#%s",home,clazz));
			}
		}
		
		return catalogNode;
	}	
	
	protected void scanResource(Properties p,ServantCatalogNodeImpl node, String home, Class bootstrap) {
		URL url = bootstrap.getResource(home);
		
		if (url != null){
			if(url.toString().startsWith("file:")){
	            File file;
				try {
					file = new File(url.toURI());
				} catch (URISyntaxException e) {
					logger.error("Can not scan home:" + home,e);
					return;
				}
				scanFileSystem(p,node,home,file,bootstrap);
	        }
	        else{
	        	if (url.toString().startsWith("jar:")){
		        	JarFile jfile;
		        	try {
		        		String jarUrl = url.toString();
		        		jarUrl = jarUrl.substring(9, jarUrl.indexOf('!'));
						jfile = new JarFile(jarUrl);
					} catch (IOException e) {
						logger.error("Can not scan home:" + home,e);
						return;
					}
		        	scanJar(p,node,home,jfile,bootstrap);
	        	}
	        }
		}else{
			logger.warn(String.format("Can not find resource in %s#%s", home,bootstrap.getName()));
		}
	}

	protected void scanJar(Properties p,ServantCatalogNodeImpl node, String home, JarFile jfile,Class bootstrap) {
        Enumeration files = jfile.entries();
        while (files.hasMoreElements()) {
          JarEntry entry = files.nextElement();
          String name = entry.getName();
          if (name.endsWith(suffix)){
              int end = name.lastIndexOf('/');
              if (end > 0) {
				  String path = '/' + name.substring(0, end);
				  if (path.equals(home)) {
					  serviceFound(p, node, "/" + name, bootstrap);
				  }
			  }
          }
        } 
	}

	protected void scanFileSystem(Properties p,ServantCatalogNodeImpl node, String home,File file,Class bootstrap) {
        File[] files = file.listFiles();
        for (File item:files){
        	if (item.getName().endsWith(suffix)){
        		serviceFound(p,node,home + "/" + item.getName(),bootstrap);
        	}
        }	
	}

	protected void serviceFound(Properties p,ServantCatalogNodeImpl node, String src,Class bootstrap) {
		//从xml文件名中获取服务id
		int end = src.lastIndexOf('.');
		int start = src.lastIndexOf('/');
		String id = src.substring(start + 1, end);
		
		Path childPath = node.getPath().append(id);
		ServiceDescription sd = loadServiceDescription(id,childPath.getPath(),src,bootstrap,p);
		if (sd != null){
			node.addService(sd.getServiceID(), sd);
		}
	}
	
	protected ServiceDescription loadServiceDescription(String id,String path,String url,Class bootstrap,Properties p){
		TogetherServiceDescription sd = null;
		
		InputStream in = null;
		try {			
			in = bootstrap.getResourceAsStream(url);
			Document doc = XmlTools.loadFromInputStream(in);
			if (doc != null){
				Element root = doc.getDocumentElement();
				sd = new TogetherServiceDescription(id,path);
				sd.configure(XmlTools.setAttr(root,"module",servant,false), p);
			}
		}catch (Exception ex){
			logger.error("Can not load service from " + url);
			logger.error(ExceptionUtils.getStackTrace(ex));
		}finally{
			IOTools.close(in);
		}
		return sd;
	}
}