Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.anysoft.batch.Main Maven / Gradle / Ivy
package com.anysoft.batch;
import java.io.File;
import java.io.FilenameFilter;
import java.io.InputStream;
import java.io.PrintStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.List;
import org.apache.commons.lang3.BooleanUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import com.anysoft.util.CommandLine;
import com.anysoft.util.Copyright;
import com.anysoft.util.DefaultProperties;
import com.anysoft.util.Factory;
import com.anysoft.util.IOTools;
import com.anysoft.util.Properties;
import com.anysoft.util.PropertiesConstants;
import com.anysoft.util.SystemProperties;
import com.anysoft.util.XmlTools;
import com.anysoft.util.resource.ResourceFactory;
/**
* 处理入口
*
* @author duanyy
*
* @version 1.6.1.3 [20141119 duanyy] commands = new Hashtable();// NOSONAR
/**
* 扩展CLASSPATH列表
*/
protected List libs = new ArrayList();
/**
* 当前的资源工厂
*/
protected ResourceFactory resourceFactory = null;
@Override
public void printHelp(PrintStream ps) {
if (!"all".equals(helpTopic)){
Command cmd = commands.get(helpTopic);
if (cmd != null){
ps.println("Syntax:");
ps.println("Main cmd= [=]");
cmd.printHelp(ps);
}else{
printRootHelp(ps);
}
}else{
printRootHelp(ps);
}
}
private void printRootHelp(PrintStream ps) {
ps.println("Syntax:");
ps.println("Main cmd= [=]");
ps.println("Commands supported are listed below:");
Enumeration cmds = commands.elements();
while (cmds.hasMoreElements()){
Command cmd = cmds.nextElement();
ps.println("\t-" + cmd.getId() + "\t:" + cmd.getNote());
}
ps.println("\t-help\t:Print this help");
}
/**
* 从本地文件装入配置
* @param p 变量集
* @param filename 本地文件名
*/
private void loadConfigFromLocalFile(DefaultProperties p,String filename){
File file = new File(filename);
if (file.exists() && file.isFile()) {
try {
Document doc = XmlTools.loadFromFile(file);
if (doc != null){
loadConfigFromElement(p,doc.getDocumentElement());
}
} catch (Exception e) {
LOG.error("Can not load xml file,url = " + filename, e);
}
} else {
LOG.error("The config file is not a valid file,url = "
+ filename);
}
}
private ResourceFactory getResourceFactory(DefaultProperties p){
if (resourceFactory == null) {
// 设置全局的ResourceFactory
String rf = p.GetValue("resource.factory",
"com.anysoft.util.resource.ResourceFactory");
try {
LOG.info("Use resource factory:" + rf);
resourceFactory = (ResourceFactory) Class.forName(rf).newInstance();
} catch (Exception ex) {
LOG.error("Can not create instance of :" + rf,ex);
}
if (resourceFactory == null) {
resourceFactory = new ResourceFactory();
LOG.info("Use default:" + ResourceFactory.class.getName());
}
}
return resourceFactory;
}
private void loadConfigFromResource(DefaultProperties p,String fileUrl){
ResourceFactory rf = getResourceFactory(p);
InputStream in = null;
try {
in = rf.load(fileUrl, null);
Document doc = XmlTools.loadFromInputStream(in);
if (doc != null){
loadConfigFromElement(p,doc.getDocumentElement());
}
}catch (Exception ex){
LOG.error("The config file is not a valid file,url = "
+ fileUrl);
}finally{
IOTools.close(in);
}
}
/**
* 装入配置信息
* @param p 环境变量集
*
*/
protected void loadConfig(DefaultProperties p){
//装入配置文件,从参数conf中读入，缺省为config.xml
String filename = PropertiesConstants.getString(p,"conf","");
if (filename != null && filename.length() > 0){
loadConfigFromLocalFile(p,filename);
}else{
filename = PropertiesConstants.getString(p,"conf.url","java:///config.xml");
loadConfigFromResource(p,filename);
}
}
protected void loadConfig(DefaultProperties p, String link) {
loadConfigFromResource(p,link);
}
private void loadParameterConfigFromElement(DefaultProperties p,Element e){ // NOSONAR
//首先处理环境变量：settings/parameter
NodeList nodeList = XmlTools.getNodeListByPath(e, "settings/parameter");
if (nodeList != null && nodeList.getLength() > 0){
for (int i = 0 ;i < nodeList.getLength() ; i ++){
Node n = nodeList.item(i);
if (Node.ELEMENT_NODE != n.getNodeType()){
continue;
}
Element parameter = (Element)n;
String id = parameter.getAttribute("id");
String value = parameter.getAttribute("value");
boolean system = "true".equals(e.getAttribute("system"))?true:false;
if (system){
if (id != null && value != null){ // NOSONAR
System.setProperty(id, value);
}
}else{
// 支持final标示,如果final为true,则不覆盖原有的取值
boolean isFinal = "true".equals(e.getAttribute("final")) ? true : false;
if (isFinal) { // NOSONAR
String oldValue = p.GetValue(id, "", false, true);
if (oldValue == null || oldValue.length() <= 0) {
p.SetValue(id, value);
}
} else {
p.SetValue(id, value);
}
}
}
}
}
private void loadCommandConfigFromElement(DefaultProperties p,Element e){
//处理commands/command
NodeList nodeList = XmlTools.getNodeListByPath(e, "commands/command");
if (nodeList != null && nodeList.getLength() > 0){
for (int i = 0 ;i < nodeList.getLength() ; i ++){
Node node = nodeList.item(i);
if (Node.ELEMENT_NODE != node.getNodeType()){
continue;
}
Element element = (Element) node;
Command cmd = new Command();
cmd.configure(element, p);
if (cmd.isOk()){
boolean overwrite = BooleanUtils.toBoolean(element.getAttribute("overwrite"));
if (!overwrite){
overwrite = !commands.containsKey(cmd.getId());
}
if (overwrite){
commands.put(cmd.getId(), cmd);
}
}
}
}
}
private void loadIncludeConfigFromElement(DefaultProperties p,Element e){
//处理includes/include
NodeList nodeList = XmlTools.getNodeListByPath(e, "includes/include");
if (nodeList != null && nodeList.getLength() > 0){
for (int i = 0 ;i < nodeList.getLength() ; i++){
Node node = nodeList.item(i);
if (Node.ELEMENT_NODE != node.getNodeType()){
continue;
}
Element element = (Element)node;
String link = element.getAttribute("link");
if (StringUtils.isNotEmpty(link)){
String loadable = element.getAttribute("loadable");
if (StringUtils.isNotEmpty(loadable)){ // NOSONAR
String load = p.transform(loadable);
if (StringUtils.isNotEmpty(load)){
loadConfig(p,p.transform(link));
}else{
LOG.info("Find xml link file,but the loadable is null,ignore...");
}
}else{
loadConfig(p,p.transform(link));
}
}
}
}
}
protected void loadConfigFromElement(DefaultProperties p,Element e) {
loadParameterConfigFromElement(p,e);
loadCommandConfigFromElement(p,e);
loadIncludeConfigFromElement(p,e);
loadLibaryFromElement(p,e);
}
private void loadLibaryFromElement(DefaultProperties p, Element e) {
//处理libs/path
NodeList nodeList = XmlTools.getNodeListByPath(e, "libs/lib");
if (nodeList != null && nodeList.getLength() > 0){
for (int i = 0 ;i < nodeList.getLength() ; i++){
Node node = nodeList.item(i);
if (Node.ELEMENT_NODE != node.getNodeType()){
continue;
}
Element element = (Element)node;
String path = element.getAttribute("path");
if (StringUtils.isNotEmpty(path)){
String loadable = element.getAttribute("loadable");
if (StringUtils.isNotEmpty(loadable)){ // NOSONAR
String load = p.transform(loadable);
if (StringUtils.isNotEmpty(load)){
addLibary(libs,p.transform(path));
}else{
LOG.info("Find xml lib file,but the loadable is null,ignore..");
}
}else{
addLibary(libs,p.transform(path));
}
}
}
}
}
protected static void addLibary(List urls,String path){
LOG.info("Searching jar libary in :" + path);
File dir = new File(path);
if (!dir.isDirectory()){
return ;
}
File[] jars = dir.listFiles(new FilenameFilter() {
public boolean accept(File file, String name) {
if (name.endsWith(".jar") || name.endsWith(".zip"))
return true;
return false;
}
});
if (jars != null){
for (File file:jars){
try {
urls.add(file.toURI().toURL());
LOG.info("Add Jar:" + file.toString());
} catch (MalformedURLException e) {
}
}
}
}
@Override
public int init(DefaultProperties p) {
commandLine = new DefaultProperties("default",p);
commandLine.addSettings(p);
//从配置文件中装入
loadConfig(commandLine);
command = PropertiesConstants.getString(p, "cmd", CMD_HELP);
if (command.equals(CMD_HELP)) {
helpTopic = PropertiesConstants.getString(p, "topic", "all");
} else {
boolean helpMode = PropertiesConstants.getBoolean(p, CMD_HELP,
false);
if (helpMode) {
helpTopic = command;
command = CMD_HELP;
}
}
return 0;
}
protected int checkAndComputeArguments(Command cmd,Properties p){
// 检查参数
List arguments = cmd.getArguments();
for (Argument argu : arguments) {
String id = argu.getId();
String value = argu.getValue(p);
if (value == null || value.length() <= 0) {
if (!argu.isNullable()) {
helpTopic = command;
helpPS.println("Can not find argument named " + id);
printHelp(helpPS);
return -1;
}
} else {
p.SetValue(id, value);
}
}
return 0;
}
@Override
public int run() {
boolean bless = PropertiesConstants.getBoolean(commandLine, "bless", true);
if (bless){
Copyright.bless(LOG, "\t\t");
}
if (command.equals(CMD_HELP)){
printHelp(helpPS);
return -1;
}
Command cmd = commands.get(command);
if (cmd == null){
helpTopic = "all";
helpPS.println("Please give me a valid command...");
printHelp(helpPS);
return -1;
}
//装入扩展CLASSPATH
if (libs.size() > 0){
ClassLoader parent = Thread.currentThread().getContextClassLoader();
if (parent == null){
parent = Main.class.getClassLoader();
}
if (parent == null){
parent = ClassLoader.getSystemClassLoader();
}
URLClassLoader cl = new URLClassLoader(libs.toArray(new URL[libs.size()]),
parent);
Thread.currentThread().setContextClassLoader(cl);
}
//创建Process对象
String module = cmd.getModule();
Process process = null;
try {
Factory factory = new Factory();// NOSONAR
process = factory.newInstance(module);
}catch (Exception ex){
LOG.error("Can not create process instance,module=" +module,ex);
return -1;
}
DefaultProperties p = commandLine;
if (checkAndComputeArguments(cmd,p) < 0){
return -1;
}
int result = process.init(p);
if (result == 0) {
result = process.run();
}
return result;
}
public static void main(String[] args) {
int result = 0;
try {
CommandLine cmdLine = new CommandLine(args,new SystemProperties());
Main main = new Main();
result = main.init(cmdLine);
if (result == 0){
result = main.run();
}
}catch (Exception ex){
LOG.error(ex.getMessage(), ex);
result = -1;
}
System.exit(result);
}
}
