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());
}
}