filehandle.java

最新的数据交换

package com.gemt.dataswap.io;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.dom4j.Document;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.XMLWriter;
import com.gemt.dataswap.resources.ReadConfig;

public class FileHandle {
	public static final Log log = LogFactory.getLog(FileHandle.class);
	/**
	 * 把生成的XML文件保存到指定的目录
	 * @param document
	 * @param tableName
	 * @throws IOException
	 */
	public static void write(Document document, String tableName) throws IOException {       
		if(log.isDebugEnabled()) {
			log.debug("write file begin ......");
		}
		boolean flag = false;
		flag = new File(ReadConfig.outfolder).mkdirs();
		if(log.isDebugEnabled()) {
			log.debug("create folder flag is " + flag);
		}
		// 指定文件
		OutputFormat format = OutputFormat.createPrettyPrint();
		format.setEncoding("GBK");		
		XMLWriter writer = new XMLWriter(new FileWriter(new File(ReadConfig.outfolder, tableName + ".xml")), format);
		writer.write( document );
		writer.close();
		/*
		// 美化格式
		OutputFormat format = OutputFormat.createPrettyPrint();
		format.setEncoding("UTF-8");
		writer = new XMLWriter( System.out, format );
		writer.write( document );
		// 缩减格式
		format = OutputFormat.createCompactFormat();
		writer = new XMLWriter( System.out, format );
		writer.write( document );
		*/
		if(log.isDebugEnabled()) {
			log.debug("write file end ......");
		}		
	}
	
	/**
	 * 扫描XML文件的存放目录，如果目录不存在，则根据system-config.xml
	 * 中的配置建立改目录，并提示系统出错信息，然后应用结束
	 * @return
	 */
	public static File[] scanFolder() {
		if(log.isDebugEnabled()) {
			log.debug("Collection<String> scanFolder.begin()");
		}
		File file = new File(ReadConfig.infolder);
		if(!file.exists() || !file.isDirectory()) {
			try {
				if(log.isErrorEnabled()) {
					log.error("存放导入数据库的XML的文件的目录不存在，导入失败！ 按任意键结束。。。。");
				}
				file.mkdirs();
				System.in.read();
				System.exit(1);
			}
			catch (IOException e)
			{
				if(log.isErrorEnabled()) {
					log.error("创建存放XML文件的目录失败");
				}
			}
		}
		File fileErr = new File(ReadConfig.failfolder);
		if(!fileErr.exists() || !fileErr.isDirectory()) {
			fileErr.mkdirs();
		}
		
		File fileok = new File(ReadConfig.okfolder);
		if(!fileok.exists() || !fileok.isDirectory()) {
			fileok.mkdirs();
		}
		
		File[] fileArr = file.listFiles(new ExtensionFileFilter("xml"));
		if(fileArr == null || fileArr.length == 0) {
			if(log.isDebugEnabled()) {
				log.debug("no file at xml import folder, application exit!!!! ");
			}
			System.exit(1);
		}
		for(int i = 0; i < fileArr.length; i ++) {
			if(log.isDebugEnabled()) {
				log.debug("files in this folder are(is) " + fileArr[i].getName());
			}
		}
		
		
		if(log.isDebugEnabled()) {
			log.debug("Collection<String> scanFolder.end()");
		}		
		return fileArr;
	}
	
	public static boolean moveFile(File dir, File file, String name) {
		if(log.isDebugEnabled()) {
			log.debug("move file begin() ");
		}
	    // Move file to new directory
	    boolean success = file.renameTo(new File(dir, name));
	    if (!success) {
	        // File was not successfully moved
	    	if(log.isErrorEnabled()) {
	    		log.error("移动文件 " + name + " 到目录 " + dir.getPath() + " 失败");
	    	}
	    }
	    
		if(log.isDebugEnabled()) {
			log.debug("move file end() ");
		}	    
		return success;
	}
	
	public static boolean moveFile(File dir, File file) {
		return moveFile(dir, file, file.getName());
	}
	
	public static void main(String[] args) {
		scanFolder();
	}
}