import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.Map.Entry;
import org.jdom2.Document;
import org.jdom2.Element;
import org.jdom2.JDOMException;
import org.jdom2.input.SAXBuilder;
/**
* 文件處理類,解析文件方法,解析的文件類型:properties文件,xml文件。
* @author lukmao
* @since 2013-08-19
* @category 工具類
* @version 1.1.0
*/
public class FileUtil {
private static int number = 0;
/**
* 去除文件BOM頭
* @param poFile 文件
* @return 去除BOM頭的文件輸入流
*/
public static final FileInputStream getNoBomInputStream(File poFile){
final byte[] bom = new byte[] { (byte)0xEF, (byte)0xBB, (byte)0xBF };
byte[] head = new byte[3];
try {
FileInputStream is = new FileInputStream(poFile);
is.read(head);
if (bom[0] == head[0] && bom[1] == head[1] && bom[2] == head[2]){
return is;
}else{
is.close();
return new FileInputStream(poFile);
}
} catch (Exception e) {
e.printStackTrace();
return null;
}
}
/**
* 通過給定的properties文件獲取其鍵值對列表
* @param file 給定的properties文件
* @return 鍵值對列表
* @throws IOException IO異常
*/
public static Map<String,String> readProperties(File file) throws IOException{
Map<String,String> result = new HashMap<String,String>();
FileInputStream fis = new FileInputStream(file);
Properties prop = new Properties();
prop.load(fis);
Set<Entry<Object, Object>> keyPires = prop.entrySet();
for(Entry<Object, Object> keyPire : keyPires ){
result.put((String)keyPire.getKey(),(String)keyPire.getValue());
}
return result;
}
/**
* 通過給定的properties文件名獲取其鍵值對列表
* @param file 給定的properties文件名
* @return 鍵值對列表
* @throws IOException IO異常
*/
public static Map<String,String> readProperties(String filename) throws IOException{
return readProperties(new File(filename));
}
/**
* 通過給定的節點和子節點名字,生成一個子節點列表
* @param element 給定節點
* @param str 子節點名字
* @return 子節點列表
*/
private static List<Element> getChindren(Element element, String str){
List<Element> result = element.getChildren(str);
return result;
}
/**
* 通過給定的節點列表和子節點名字,生成一個子節點列表
* @param elements 給定節點列表
* @param str 子節點名字
* @return 子節點列表
*/
private static List<Element> getChindren(List<Element> elements, String str){
List<Element> result = new ArrayList<Element>();
for(Element element : elements){
List<Element> lElement = element.getChildren(str);
for(Element e: lElement){
result.add(e);
}
}
return result;
}
/**
* 通過xml文件名和需要獲取的節點路徑獲取xml內容
* @param filename xml文件名
* @param path 需要獲取的節點路徑,例如test/subtest,不需要帶上Root節點
* @return 獲取的節點內容列表
* @throws IOException IO異常
* @throws JDOMException xml解析異常
*/
public static List<String> getFromXML(File file, String path) throws IOException, JDOMException{
List<String> result = null;
SAXBuilder builder = new SAXBuilder();
Document doc = builder.build(file);
Element root = doc.getRootElement();
List<Element> lElement = null;
List<String> lStr = StringUtil.splitToList(path, "/");
for(int i = 0; i < lStr.size(); i++){
if(lStr.size() == 1){
lElement = getChindren(root, lStr.get(i));
result = new ArrayList<String>();
for(Element e : lElement){
result.add(e.getTextTrim());
}
}else{
if(i == 0){
lElement = getChindren(root, lStr.get(i));
}
else if(i != lStr.size() - 1){
lElement = getChindren(lElement, lStr.get(i));
}else{
lElement = getChindren(lElement, lStr.get(i));
result = new ArrayList<String>();
for(Element e : lElement){
result.add(e.getTextTrim());
}
}
}
}
return result;
}
/**
* 通過xml文件和需要獲取的節點路徑獲取xml內容
* @param filename xml文件
* @param path 需要獲取的節點路徑,例如test/subtest,不需要帶上Root節點
* @return 獲取的節點內容列表
* @throws IOException IO異常
* @throws JDOMException xml解析異常
*/
public static List<String> getFromXML(String filename, String path) throws IOException, JDOMException{
return getFromXML(new File(filename), path);
}
@SuppressWarnings("unchecked")
public static void parseBeanToXML(String filename, Boolean rootFlag, Object params)
throws ClassNotFoundException, IOException, SecurityException, NoSuchFieldException, InstantiationException, IllegalAccessException, NoSuchMethodException, IllegalArgumentException, InvocationTargetException{
File file = new File(filename);
StringBuffer sb = new StringBuffer("");
sb.append("<?xml version=\"1.0\" encoding=\"utf-8\" ?> \n");
if(rootFlag){
sb.append("<ROOT>\n");
}
Class clazz = params.getClass();
Field[] fields = clazz.getDeclaredFields();
for(Field field : fields){
String fieldname = field.getName();
Method method = clazz.getMethod("get" + fieldname.substring(0,1).toUpperCase() + fieldname.substring(1), new Class[]{});
Object value = method.invoke(params, new Object[]{});
getSubElement(sb, field, value);
}
if(rootFlag){
sb.append("</ROOT>\n");
}
PrintWriter pw = new PrintWriter(new FileWriter(file));
pw.println(sb.toString());
pw.flush();
}
@SuppressWarnings("unchecked")
private static StringBuffer getSubElement(StringBuffer sb, Field field, Object value) throws SecurityException, NoSuchMethodException, IllegalArgumentException, IllegalAccessException, InvocationTargetException, InstantiationException{
String fieldname = field.getName();
//System.out.println(fieldname);
if(value == null){
return sb;
}
if(isSimpleType(value)){
sb.append("\t<").append(fieldname).append(">");
sb.append(value);
sb.append("</").append(fieldname).append(">\n");
}else{
if(value instanceof List){
List list = (List)value;
for(Object disvalue : list){
for(int i = 0; i < number; i++){
sb.append("\t");
}
sb.append("\t<").append(fieldname).append(">");
sb.append(disvalue);
sb.append("</").append(fieldname).append(">\n");
}
}
else if(value instanceof Date){
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-mm-dd hh:MM:ss");
Date date = (Date)value;
sb.append("\t<").append(fieldname).append(">");
sb.append(sdf.format(date));
sb.append("</").append(fieldname).append(">\n");
}else{
number++;
sb.append("\t<").append(fieldname).append(">\n");
Class subclazz = field.getType();
Field[] subfields = subclazz.getDeclaredFields();
for(Field subfield : subfields){
String subname = subfield.getName();
Method method = subclazz.getMethod("get" + subname.substring(0,1).toUpperCase() + subname.substring(1), new Class[]{});
Object subvalue = method.invoke(value, new Object[]{});
sb = getSubElement(sb,subfield, subvalue);
}
sb.append("\t</").append(fieldname).append(">\n");
number = 0;
}
}
return sb;
}
private static Boolean isSimpleType(Object obj){
Boolean result = false;
if(obj instanceof String){
result = true;
}
if(obj instanceof Number){
result = true;
}
if(obj instanceof Boolean){
result = true;
}
return result;
}
}