xmldeptdao.java

用java开发的QQ管理系统

package com.zlf.dao.xml;

/***********************************************************************
 * Module:  XmlDeptDao.java
 * Author:  zlf
 * Purpose: Defines the Class XmlDeptDao
 ***********************************************************************/

import java.io.FileWriter;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import java.util.Vector;

import javax.swing.JOptionPane;

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

import com.zlf.dao.DeptDao;
import com.zlf.dao.vo.Dept;
import com.zlf.qqserver.utils.QQUtils;


public class XmlDeptDao implements DeptDao {

	
	
	
	
	
	
	
	
	
	/**
	 * 
	 * 
	 * @return Document
	 * 
	 * 读取XML返回Document
	 */
	public Document getDocument(){
		SAXReader saxReader = new SAXReader();
		Document doc = null;
		try {
			doc = saxReader.read("xml/dept.xml");
			//System.out.println("ok");
		} catch (DocumentException e) {
			//System.out.println("oh no");
			e.printStackTrace();
			JOptionPane.showMessageDialog(null,"XML读取出错");
			QQUtils.properties.setProperty("noshow","no");
			QQUtils.saveProperties();
		}
		return doc;
		
	}
	
	/**
	 * 
	 * @param doc
	 * @param id
	 * @return
	 * 根据部门ID查找
	 * 
	 */
	public Element findElement(Document doc,String id){
		Element root = doc.getRootElement();
		Node node = root.selectSingleNode("/depts/dept[@deptId='"+id+"']");
		
		if (node != null && node.getNodeType() == Node.ELEMENT_NODE){
			return (Element)node;
		}else
			return null;
				
	
	}

	/**
	 * 
	 * @param doc
	 * @param name
	 * @return
	 * 根据部门名称查找
	 */
	public Element findElementByName(Document doc,String name){
		
		Element root = doc.getRootElement();
		Node node = root.selectSingleNode("/deptrs/dept/deptName[text()='"+name+"']/parent::*");
		
		if (node != null && node.getNodeType() == Node.ELEMENT_NODE){
			return (Element)node;
		}else
			return null;
	}
	
	/**
	 * 
	 * @param doc
	 * 
	 * 将Doc输出至XML
	 */
	public void outputXML(Document doc){

		try {
			OutputFormat format = OutputFormat.createPrettyPrint();
			format.setEncoding("GBK");
			XMLWriter writer = new XMLWriter(new FileWriter("xml/dept.xml"),format);
			writer.write(doc);
			writer.close();
		} catch (IOException e) {
			e.printStackTrace();
			
		}
		
	}
	
	
	
	

	/**
	 * @param dept
	 *
	 *更新部门
	 */
	
	
	public int updateDept(Dept dept) {
		
		try {
			Document doc = this.getDocument();
			List deptList = doc.selectNodes("/depts/dept");

			for (int i = 0; i < deptList.size(); i++) {
				Element e = (Element) deptList.get(i);
			
	
				if (e.attribute("deptId").getText().equals(dept.getDeptId())) {
					e.selectSingleNode("deptName").setText(dept.getDeptName());
					e.selectSingleNode("deptRemark").setText(dept.getDeptRemark());
				}
			}

			
			this.outputXML(doc);
			
			return 1;

		} catch (Exception e) {
			e.printStackTrace();
			return 0;
		}
	}

	/**
	 * @param dept
	 * 
	 * 增加部门
	 * 
	 */
	public int addDept(Dept dept) {
		Document doc = getDocument();
		if (doc != null){
			if (findElement(doc,dept.getDeptId()) == null){
				Element root = doc.getRootElement();
				
				Element edept = root.addElement("dept");
				
				edept.addAttribute("deptId", dept.getDeptId());
				
				
				Element edeptname = edept.addElement("deptName");
				edeptname.setText(dept.getDeptName());
				
				Element edeptremark = edept.addElement("deptRemark");
				edeptremark.setText(dept.getDeptRemark());
				
				outputXML(doc);
				
				return 1;
			}
		}else
			return 0;
		return 0;
	}



	/**
	 * 
	 * 删除部门
	 */
	public int deleteDept(String deptId) {
		
		try {
			Document doc = this.getDocument();
			// dept
			List deptList = doc.selectNodes("/depts/dept");

			for (int i = 0; i < deptList.size(); i++) {
				Element e = (Element) deptList.get(i);
				if (e.attribute("deptId").getText().equals(deptId)) {
					deptList.remove(i);
				}
			}
			this.outputXML(doc);
			
			return 1;

		} catch (Exception e) {
			e.printStackTrace();
			return 0;
		}
	}


	/**
	 * 
	 * 查询部门
	 * 
	 */
	public Vector selectDept(HashMap hashMap) {

		Vector data = new Vector();

		
		try {
			Document doc = this.getDocument();
			// dept
			//List deptList = doc.selectNodes("//depts/@dept");
			List deptList = doc.selectNodes("/depts/dept");
			Set set = hashMap.keySet();
			Iterator iter = set.iterator();
			while (iter.hasNext()) {
				String st = (String) iter.next();
				if (st.equals("1")) {
					// 1.表示查找出所有的字段
					for (int i = 0; i < deptList.size(); i++) {
						Element e = (Element) deptList.get(i);
						// System.out.println(e.selectSingleNode("deptId").getText());
						Vector row = new Vector();
						row.add(e.attribute("deptId").getText());
						row.add(e.selectSingleNode("deptName").getText());
						row.add(e.selectSingleNode("deptRemark").getText());
						data.add(row);
					}
				} else if (st.equals("2")) {
					// 2.自动生成编号
					int max = 0;
					if (deptList.size() == 0) {
						data.add("000001");
					} else {
						for (int i = 0; i < deptList.size(); i++) {
							Element e = (Element) deptList.get(i);

							if (Integer
									.parseInt(e.attribute("deptId").getText()) > max) {
								max = Integer.parseInt(e.attribute("deptId")
										.getText());
							}
						}
						max++;
						String maxStr = "000000".substring(0, 6 - String
								.valueOf(max).length())
								+ String.valueOf(max);
						data.add(maxStr);
					}

				}else if(st.equals("3")){
//					3. 返回部门所有名称
					String[] deptName=new String[deptList.size()+1];
					deptName[0]="";
					for (int i = 0; i < deptList.size(); i++) {
						Element e = (Element) deptList.get(i);
						deptName[i+1]=e.selectSingleNode("deptName").getText();
					}
					data.add(deptName);
				}else if(st.equals("4")){
					//4. 根据部门的编号，到数据库中查找出部门信息
					for (int i = 0; i < deptList.size(); i++) {
						Element e = (Element) deptList.get(i);
						if (e.attribute("deptId").getText().equals((String)hashMap.get("4"))){
							Vector row = new Vector();
							row.add(e.attribute("deptId").getText());
							row.add(e.selectSingleNode("deptName").getText());
							row.add(e.selectSingleNode("deptRemark").getText());
							data.add(row);
						}
					}
					
				}else if(st.equals("5")){
//					 5.根据编号查到部门的名称
					for (int i = 0; i < deptList.size(); i++) {
						Element e = (Element) deptList.get(i);
						if (e.attribute("deptId").getText().equals(hashMap.get("5"))){
							data.add(e.selectSingleNode("deptName").getText());
						}
					}
				}else if (st.equals("6")){
					//6. 根据名称查到编号
					for (int i = 0; i < deptList.size(); i++) {
						Element e = (Element) deptList.get(i);
						if (e.selectSingleNode("deptName").getText().equals((String)hashMap.get("6"))){
							data.add(e.attribute("deptId").getText());
						}
					}
				}
			}


		} catch (Exception e) {
			e.printStackTrace();
		}
		return data;
	}




}