package cn.datasource.xml;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

public class AddDrug {
	public static void addUser(Map<String,String> drug){
		try{
		  //读取文件
		  SAXReader saxReader = new SAXReader();
		  Document doc = saxReader.read(new File(Constant.DRUGFILEPATH));
		  //得到跟结点root
		  Element root = doc.getRootElement();
		  //往根节点添加drug元素
		  Element Drug = root.addElement("drug");
		  //设置drug的drugID
		  Drug.addAttribute("drugID", drug.get("code").toString());
		  //往user添加各种子元素和设置值
		  Element code = Drug.addElement("code");
		  code.setText(drug.get("code").toString());
		  Element name = Drug.addElement("name");
		  name.setText(drug.get("name").toString());
		  Element pill = Drug.addElement("pill");
		  pill.setText(drug.get("pill").toString());
		  Element type = Drug.addElement("type");
		  type.setText(drug.get("type").toString());
		  Element grade = Drug.addElement("grade");
		  grade.setText(drug.get("grade").toString());
		  //添加元素结束后往新的文件重新写入doc数据
		  XMLWriter output = new XMLWriter(new FileWriter( new File(Constant.DRUGFILEPATH) ));
		  output.write(doc);
		  //关闭输出流
		  output.close();
		}catch(DocumentException e){
			e.printStackTrace();
		}catch(IOException e){
			e.printStackTrace();
		}
	}
	
	public static void main(String[] args) {
		Map<String,String> drugMap = new HashMap<String,String>();
		drugMap.put("code", "X-J05AF-A014-A006");
		drugMap.put("name", "阿德福韦酯");
		drugMap.put("pill", "分散片");
		drugMap.put("type", "西药");
		drugMap.put("grade", "乙");
		
		AddDrug.addUser(drugMap);
	}
}