自动生成代码工具 配置文件工具类

                                                     自动生成代码工具 配置文件工具类

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.ijwt.element.Constants;
import com.ijwt.element.DataSource;
import com.ijwt.element.Global;
import com.ijwt.element.Table;
import com.ijwt.element.XMLElement;
import com.ijwt.exception.XMLParserException;

/**
 * @author InJavaWeTrust
 */
public enum UtilXML {
	
	INSTANCE;
	
	private static Logger log = LoggerFactory.getLogger(UtilXML.class);
	private Global global         = null;
	private DataSource dataSource = null;
	private List<Table> tableList = null;
	
	private XMLElement<List<Element>, Global> globalElement = elementList -> {
		Global globalParam = new Global();
		if (!elementList.isEmpty()) {
			elementList.forEach(element -> {
				if("osdir".equals(element.getName())) {
					globalParam.setOsdir(element.attributeValue("value"));
				} else if ("javapackage".equals(element.getName())) {
					globalParam.setJavapackage(element.attributeValue("value"));
				} else if ("author".equals(element.getName())) {
					globalParam.setAuthor(element.attributeValue("value"));
				} else if ("project".equals(element.getName())) {
					globalParam.setProject(element.attributeValue("value"));
				}
			});
		}
		return globalParam;
	};
	
	private XMLElement<List<Element>, DataSource> dataSourceElement = elementList -> {
		DataSource dataSourceParam = new DataSource();
		if (!elementList.isEmpty()) {
			elementList.forEach(element -> {
				if("driver".equals(element.getName())) {
					dataSourceParam.setDriver(element.attributeValue("value"));
				} else if ("url".equals(element.getName())) {
					dataSourceParam.setUrl(element.attributeValue("value"));
				} else if ("username".equals(element.getName())) {
					dataSourceParam.setUsername(element.attributeValue("value"));
				} else if ("password".equals(element.getName())) {
					dataSourceParam.setPassword(element.attributeValue("value"));
				}
			});
		}
		return dataSourceParam;
	};
	
	private XMLElement<List<Element>, List<Table>> tableElement = elementList -> {
		List<Table> tableListParam = new ArrayList<>();
		if (!elementList.isEmpty()) {
			elementList.forEach(element -> {
				if ("table".equals(element.getName())) {
					Table table = new Table();
					table.setTableName(element.attributeValue("value"));
					tableListParam.add(table);
				}
			});
		}
		return tableListParam;
	};

	
	static {
		try (InputStream in = UtilXML.class.getClassLoader()
				.getResourceAsStream("config.xml")) {
			UtilXML.INSTANCE.parseConfigXml(in);
		} catch (XMLParserException | IOException e) {
			log.error("##解析配置文件出现问题！！！[" + e.getMessage() + "]");
		}
	}
	
	@SuppressWarnings("unchecked")
	private void parseConfigXml(InputStream in) throws XMLParserException {
		try {
			SAXReader reader = new SAXReader();
			Document document = reader.read(in);
			Element root = document.getRootElement();
			List<Element> elements = root.elements();
			elements.forEach(element -> {
				switch (element.getName()) {
				case Constants.GLOBAL : 
					global = globalElement.getElement(element.elements());
					break;
				case Constants.DATASOURCE : 
					dataSource = dataSourceElement.getElement(element.elements());
					break;
				case Constants.TABLELIST : 
					tableList = tableElement.getElement(element.elements());
					break;
				default : log.warn("## 未知元素 [{}]", element.getName());
				}
			});
		} catch (DocumentException e) {
			throw new XMLParserException("解析 config.xml 时发生异常 [" + e.getMessage() + "]");
		}
	}
	
	public Global getGlobal() {
		return global;
	}
	
	public DataSource getDataSource() {
		return dataSource;
	}

	public List<Table> getTableList() {
		return tableList;
	}
	
	public static void main(String[] args) {
		List<Table> tabLst = UtilXML.INSTANCE.getTableList();
		tabLst.forEach(table -> System.out.println(table.getTableName()));
		Global global = new Global();
		global = UtilXML.INSTANCE.getGlobal();
		System.out.println(global.getAuthor());
		System.out.println(global.getJavapackage());
		System.out.println(global.getOsdir());
		System.out.println(global.getProject());
		DataSource dataSource = new DataSource();
		dataSource = UtilXML.INSTANCE.getDataSource();
		System.out.println(dataSource.getDriver());
		System.out.println(dataSource.getPassword());
		System.out.println(dataSource.getUrl());
		System.out.println(dataSource.getUsername());
		
	}

}