Dom4j 应用

这段时间把XML的技术重新复习了一遍，用Dom4j 做了一个小练习

准备jar包：dom4j-1.6.1.jar

 

一、用来解析的XML文件user.xml

<?xml version="1.0" encoding="UTF-8"?>
<users>
 <user id="13211324" username="aaa" password="123456" email="aa@yahoo.com" birthday="1980-01-02" nickname="小强" />
</users>

二、用来封装XML数据的JavaBean User.java

package com.joeho.domain;

import java.util.Date;

public class User {

 private String id;
 private String username;
 private String nickname;
 private String password;
 private String email;
 public String getId() {
  return id;
 }
 public void setId(String id) {
  this.id = id;
 }

 public String getNickname() {
  return nickname;
 }
 public void setNickname(String nickname) {
  this.nickname = nickname;
 }
 public String getPassword() {
  return password;
 }
 public void setPassword(String password) {
  this.password = password;
 }
 public String getEmail() {
  return email;
 }
 public void setEmail(String email) {
  this.email = email;
 }
 public Date getBirthday() {
  return birthday;
 }
 public void setBirthday(Date birthday) {
  this.birthday = birthday;
 }
 public void setUsername(String username) {
  this.username = username;
 }
 public String getUsername() {
  return username;
 }
 private Date birthday;
}

 

三、用来读写XML的工具类

package com.joeho.utils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

import org.dom4j.Document;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

public class XmlUtils {
 
 private static String filepath;
 static{
  filepath=XmlUtils.class.getClassLoader().getResource("users.xml").getPath();
 }
 
 public static Document getDocument() throws Exception{
  SAXReader reader = new SAXReader();
  Document document =reader.read(new File(filepath));
  return document;
 }

 public static void write2Xml(Document document) throws IOException{
  OutputFormat format = OutputFormat.createPrettyPrint();
  format.setEncoding("utf-8");
  XMLWriter writer = new XMLWriter(new FileOutputStream(filepath),format);
  writer.write(document);
  writer.close();
 }
}

 

四、对XML文件进行保存，查找的DAO

package com.joeho.dao.impl;

import java.text.SimpleDateFormat;
import java.util.Iterator;

import org.dom4j.Document;
import org.dom4j.Element;

import com.joeho.dao.UserDao;
import com.joeho.domain.User;
import com.joeho.utils.XmlUtils;

public class UserDaoImpl implements UserDao {

 public void addUser(User user) {
  try {
   Document document = XmlUtils.getDocument();
   Element root = document.getRootElement();
   Element xmlUser = root.addElement("user");
   xmlUser.setAttributeValue("id", user.getId());
   xmlUser.setAttributeValue("username", user.getUsername());
   xmlUser.setAttributeValue("nickname", user.getNickname());
   xmlUser.setAttributeValue("password", user.getPassword());
   xmlUser.setAttributeValue("email", user.getEmail());
   xmlUser.setAttributeValue("birthday", user.getBirthday() == null ? ""
     : user.getBirthday().toLocaleString());
   XmlUtils.write2Xml(document);
  } catch (Exception e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
  }

 }

 public User findUser(String name, String password) {
  try {
   Document document = XmlUtils.getDocument(); 
   Element root = document.getRootElement();
    //遍历当前元素(在此是根元素)的子元素
         for (Iterator iter = root.elementIterator(); iter.hasNext();) {
             Element xmlUser = (Element) iter.next();             
             if(xmlUser.attributeValue("username").equals(name)&&xmlUser.attributeValue("password").equals(password)){ 
              
    User newUser = new User();
    String birthday = xmlUser.attributeValue("birthday");
    if (birthday == null || birthday.equals("")) {
     newUser.setBirthday(null);
    } else {
     SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
     newUser.setBirthday(sdf.parse(birthday));
    }
    newUser.setEmail(xmlUser.attributeValue("email"));
    newUser.setId(xmlUser.attributeValue("id"));
    newUser.setNickname(xmlUser.attributeValue("nickname"));
    newUser.setPassword(password);
    newUser.setUsername(name);
    return newUser;
             }            
         } 

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

 public boolean findUser(String name) {
  boolean isfind = false;
  try {
   Document document = XmlUtils.getDocument();
   Element root = document.getRootElement();
    //遍历当前元素(在此是根元素)的子元素
         for (Iterator iter = root.elementIterator(); iter.hasNext();) {
             Element user = (Element) iter.next();             
             if(user.attributeValue("username").equals(name)){
              isfind= true;
             }            
         } 

  } catch (Exception e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
  }
  return isfind;
 }
}

 

五、Junit 测试类

package junit.test;

import java.util.Date;

import org.junit.Test;

import com.joeho.dao.impl.UserDaoImpl;
import com.joeho.domain.User;

public class UserDaoTest {
 
 @Test
 public void testAdd(){
  User user = new User();
  user.setBirthday(new Date());
  user.setEmail("li@yahoo.com");
  user.setId("213414238");
  user.setNickname("小毛");
  user.setPassword("98948");
  user.setUsername("lio");
  
  UserDaoImpl udao= new UserDaoImpl();
  udao.findUser("adsfaf");
 }
 @Test
 public void testFindUser(){
  UserDaoImpl udao= new UserDaoImpl();
  udao.findUser("lio");
 }
}