xml 小练习

以XML为一个小的数据库，写一个小练习，命令式菜单，练习java解析xml--dom方式的增删改查。

下面是项目分包：注意xml文件跟src是并列的。

一开始Contact.xml文档如下图：

写一个工具类生产uuid：

package cn.hncu.contact.util;

import java.util.UUID;

public class IDGenerator {
   private IDGenerator(){
     
   }
   
     public static String  getId(){
       String uuid=UUID.randomUUID().toString();
        uuid=uuid.replaceAll("-", "");
       return uuid;
     }
}

重点：因为是多用户，虽然是单机版，但这里要用到单例！！！

生成单例的代码如下，写一个工厂类：

package cn.hncu.contact.common;

import java.io.File;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.TransformerFactoryConfigurationError;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.w3c.dom.Document;


public class DocumentFactory {
    private static Document dom;
    private static File file;
    static{
        try {
            file = new File("./xml/Contact.xml");
            if(!file.exists()){
                throw new Exception("数据库不存在!");
            }
            DocumentBuilder db=DocumentBuilderFactory.newInstance().newDocumentBuilder();
            dom = db.parse(file);
        } catch (ParserConfigurationException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    public static Document getDocument(){
        return dom;
    }
    
    public static void save(){
        try {
            Transformer trans=TransformerFactory.newInstance().newTransformer();
            trans.transform(new DOMSource(dom), new StreamResult(file));
        } catch (TransformerConfigurationException e) {
            e.printStackTrace();
        } catch (TransformerFactoryConfigurationError e) {
            e.printStackTrace();
        } catch (TransformerException e) {
            e.printStackTrace();
        }
        
    }
}

然后写一个测试类 ：测试一下工具类：

package cn.hncu.testContact;

import org.junit.Test;
import org.w3c.dom.Document;

import cn.hncu.contact.common.DocumentFactory;
import cn.hncu.contact.util.IDGenerator;

public class ContactTest {
  @Test
     public void testGetId(){
      String id = IDGenerator.getId();
      System.out.println(id);
      System.out.println(IDGenerator.getId());
      System.out.println(IDGenerator.getId());
     }
     
     
     @Test
     public void testGetDom(){
       Document dom1 = DocumentFactory.getDocument();
       Document dom2 = DocumentFactory.getDocument();
       System.out.println(dom1.hashCode());
       System.out.println(dom2.hashCode());
     }
}

结果如下: 

第一个是生成的单例的hashCode，一样说明，是单例！ 第二个是随机生成的UUID

接下来写Dao层，接口，工厂，实现类：

package cn.hncu.contact.dao;

import java.util.List;
import java.util.Map;

public interface ContactDAO {
   public boolean login(String name,String pwd);
   public void reg(String name,String pwd);
   public boolean add(String name, String tel, String sex);
   public List<Map<String, String>> queryAllContacts();
   public String del(String id);
   public boolean upd(String id,String name,String tel,String sex,String target);
}

package cn.hncu.contact.dao.factory;

import cn.hncu.contact.dao.ContactDAO;
import cn.hncu.contact.dao.impl.ContactDaoImpl;

public class ContactDAOFactory {
   private ContactDAOFactory(){
      
   }
   public static ContactDAO getContactDao(){
      return new ContactDaoImpl();
   }
}

package cn.hncu.contact.dao.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

import cn.hncu.contact.common.DocumentFactory;
import cn.hncu.contact.dao.ContactDAO;
import cn.hncu.contact.util.IDGenerator;

public class ContactDaoImpl implements ContactDAO {

  private Element eCurrentUser = null; // 当前用户

  @Override
  public boolean login(String name, String pwd) {
    Document dom = DocumentFactory.getDocument();
    Element root = (Element) dom.getFirstChild();
    NodeList eUserList = root.getElementsByTagName("user");
    for (int i = 0; i < eUserList.getLength(); i++) {
      Element eUser = (Element) eUserList.item(i);
      if (eUser.getAttribute("name").equalsIgnoreCase(name)
          && eUser.getAttribute("pwd").equals(pwd)) {
        eCurrentUser = eUser; // 把当前登录成功的用户保存到类成员变量eCurrentUser，供其它业务方法使用
        return true;
      }
    }

    return false;
  }

  @Override
  public void reg(String name, String pwd) {
    Document dom = DocumentFactory.getDocument();
    // 把新用户节点添加到dom中
    Element root = (Element) dom.getFirstChild();

    // 创建一个新的用户节点 eNewUser
    Element eNewUser = dom.createElement("user");
    eNewUser.setAttribute("name", name);
    eNewUser.setAttribute("pwd", pwd);
    // 把eNewUser添加为root的孩子
    root.appendChild(eNewUser);

    // 保存
    DocumentFactory.save();
  }

  @Override
  public boolean add(String name, String tel, String sex) {
    Document dom = DocumentFactory.getDocument();
    // 创建一个新的联系人节点<contact>, eNewContact对象
    Element eNewContact = dom.createElement("contact");
    eNewContact.setAttribute("id", IDGenerator.getId());

    // 创建一个<name>元素对应的 eName,并把它添加成eNewContact的孩子
    Element eName = dom.createElement("name");
    eName.setTextContent(name);
    eNewContact.appendChild(eName);

    // 创建一个<tel>元素对应的 eTel,并把它添加成eNewContact的孩子
    Element eTel = dom.createElement("tel");
    eTel.setTextContent(tel);
    eNewContact.appendChild(eTel);

    // 创建一个<sex>元素对应的 eSex,并把它添加成eNewContact的孩子
    Element eSex = dom.createElement("sex");
    eSex.setTextContent(sex);
    eNewContact.appendChild(eSex);

    // 把上面组装好的eNewContact添加成eCurrentUser的孩子
    eCurrentUser.appendChild(eNewContact);

    // 序列化
    DocumentFactory.save();
    return false;
  }

  @Override
  public List<Map<String, String>> queryAllContacts() {
    List<Map<String, String>> resultList = new ArrayList<Map<String, String>>();
    if (eCurrentUser == null) {
      return resultList;
    }
    NodeList eContactList = eCurrentUser.getElementsByTagName("contact");
    for (int i = 0; i < eContactList.getLength(); i++) {
      Element eContact = (Element) eContactList.item(i);
      // 把每一个eContact中的信息封装成一个map
      Map<String, String> map = new HashMap<String, String>();
      map.put("id", eContact.getAttribute("id"));// id
      map.put("name", eContact.getElementsByTagName("name").item(0)
          .getTextContent());// name
      map.put("sex", eContact.getElementsByTagName("sex").item(0)
          .getTextContent());// sex
      map.put("tel", eContact.getElementsByTagName("tel").item(0)
          .getTextContent());// tel

      // 把map添加到resultList中
      resultList.add(map);
    }
    return resultList;
  }

  @Override
  public String del(String id) {
    NodeList eContactList = eCurrentUser.getElementsByTagName("contact");
    for (int i = 0; i < eContactList.getLength(); i++) {
      Element eContact = (Element) eContactList.item(i);
      String iid = eContact.getAttribute("id");
      if (id.equals(iid)) {
        // 先保存要返回的数据
        String name = eContact.getElementsByTagName("name").item(0)
            .getTextContent();
        eContact.getParentNode().removeChild(eContact);
        DocumentFactory.save();

        return name;
      }
    }
    return null;

  }

  @Override
  public boolean upd(String id, String name, String tel, String sex,String target) {
    NodeList eContactList = eCurrentUser.getElementsByTagName("contact");
    boolean boo = false;
    for (int i = 0; i < eContactList.getLength(); i++) {
      Element eContact = (Element) eContactList.item(i);
      String iid = eContact.getAttribute("id");
      if (id.equals(iid)) {
        if ("0".equals(target)) {
          tel = eContact.getElementsByTagName("tel").item(0)
              .getTextContent();
          sex = eContact.getElementsByTagName("sex").item(0)
              .getTextContent();
          Element eName = (Element) eContact.getElementsByTagName(
              "name").item(0);
          eName.setTextContent(name);
          boo = true;
        } else if ("1".equals(target)) {
          sex = eContact.getElementsByTagName("sex").item(0)
              .getTextContent();
          name = eContact.getElementsByTagName("name").item(0)
              .getTextContent();
          Element eTel = (Element) eContact.getElementsByTagName(
              "tel").item(0);
          eTel.setTextContent(tel);
          boo = true;
        } else if ("2".equals(target)) {
          tel = eContact.getElementsByTagName("tel").item(0)
              .getTextContent();
          name = eContact.getElementsByTagName("name").item(0)
              .getTextContent();
          Element eSex = (Element) eContact.getElementsByTagName(
              "sex").item(0);
          eSex.setTextContent(sex);
          boo = true;
        } else {
          boo = false;
        }
      
      } 
      DocumentFactory.save();
      }
    return boo;
  }
   
}

最后是表现层：

package cn.hncu.contact.cmd;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Scanner;

import cn.hncu.contact.dao.ContactDAO;
import cn.hncu.contact.dao.factory.ContactDAOFactory;

public class ContactAction {
  private Scanner sc = null;
  // 声明一个联系人id集合---存放序号与id对应关系的hashMap
  // 在每次显示所有联系人时重新存入，在删除与修改时读取
  Map<Integer, String> idMap = null;
  // 注入dao
  private ContactDAO dao = ContactDAOFactory.getContactDao();

  public ContactAction() {
    sc = new Scanner(System.in);
    System.out.println("※※警告用户:非GBK编码时,MyEclipse控制台中不支持中文输入!※※");
    System.out
        .println("---------------------------------------------------");

    while (true) {
      // 软件的主界面显示
      System.out.println("1:登录");
      System.out.println("2:注册");
      System.out.println("0:退出");

      String op = sc.nextLine();
      if ("1".equals(op)) {
        login();
      } else if ("2".equals(op)) {
        reg();
      } else if ("0".equals(op)) {
        System.out.println("程序马上退出....");
        break;
      }

    }
  }

  private void reg() {
    System.out.println("*********************");
    System.out.println("*      用户注册               *");
    System.out.println("*********************");
    System.out.println("请输入用户名:");
    String name = sc.nextLine();
    System.out.println("请输入密码:");
    String pwd = sc.nextLine();
    System.out.println("请再次输入密码:");
    String pwd2 = sc.nextLine();
    if (name.trim().length() > 0) {
      if (pwd.equals(pwd2)) {
        // 调用dao层进行注册
        dao.reg(name, pwd);
        System.out.println("恭喜,注册成功!");
      } else {
        System.out.println("两次密码不一致，请重新注册!");
        reg();
      }
    } else {
      System.out.println("姓名不能为空字符");
      reg();
    }

  }

  private void login() {
    System.out.println("*********************");
    System.out.println("*      用户登录               *");
    System.out.println("*********************");
    System.out.println("请输入用户名:");
    String name = sc.nextLine();
    System.out.println("请输入密码:");
    String pwd = sc.nextLine();
    boolean boo = dao.login(name, pwd);
    if (boo) {
      System.out.println("恭喜，登录成功，请选择下一步操作!");
      operation();
    } else {
      System.out.println("登录失败!!!!");
    }
  }

  private void operation() {
    System.out.println("1:显示所有联系人");
    System.out.println("2:添加联系人");
    System.out.println("3:删除联系人");
    System.out.println("4:修改联系人");
    System.out.println("0:返回到软件主界面");

    String sel = sc.nextLine();
    if ("1".equals(sel)) {// 显示所有联系人
      queryAll();
    } else if ("2".equals(sel)) {// 添加联系人
      addContact();
    } else if ("3".equals(sel)) {// 删除联系人
      delContact();
    } else if ("4".equals(sel)) {// 修改联系人
           updContact();
    } else if ("0".equals(sel)) {// 退回到上一级菜单
      return;
    }
    operation();
  }

  private void queryAll() {
    List<Map<String, String>> list = dao.queryAllContacts();
    System.out.println("序号\t姓名\t电话\t性别");
    System.out.println("-------------------------------");
    idMap = new HashMap<Integer, String>();
    int i = 1;
    for (Map<String, String> map : list) {
      String id = map.get("id");
      idMap.put(i, id); // 为以后的删除与修改做存储

      String name = map.get("name");
      String tel = map.get("tel");
      String sex = map.get("sex");
      if ("0".equals(sex)) {
        sex = "男";
      } else {
        sex = "女";
      }
      System.out.println((i++) + "\t" + name + "\t" + tel + "\t"
          + sex);
    }
    
  }

  private void updContact() {
    queryAll();
    String name=null;
    String sex=null;
    String tel=null;
    boolean boo=false;
    System.out.println("请输入要修改联系人的序号");
    int num = sc.nextInt();
    sc.nextLine();//吸掉多余的换行符
    String id = idMap.get(num);
    System.out.println("请输入要修改的属性：0-姓名 1-电话 2-性别");
    String target=sc.nextLine();
    if ("0".equals(target)) {
      System.out.println("请输入要修改姓名的内容：");
      name=sc.nextLine();
        boo=dao.upd(id, name, tel, sex,target);
    }else if("1".equals(target)){
      System.out.println("请输入要修改电话的内容：");
      tel=sc.nextLine();
        boo=dao.upd(id, name, tel, sex,target);
    }else if("2".equals(target)){
      System.out.println("请输入要修改性别的内容：");
      sex=sc.nextLine();
        boo=dao.upd(id, name, tel, sex,target);
    }
    if(boo){
      System.out.println("修改成功~~");
      queryAll();
    }else{
      System.out.println("修改失败！！");
    }
  }

  private void delContact() {
    queryAll();
    System.out.println("请输入要删除联系人的序号");
    int num = sc.nextInt();
    sc.nextLine();//吸掉多余的换行符
    String id = idMap.get(num);
    String name = dao.del(id);
    if(name!=null){
      System.out.println("已经删除联系人:"+name);
      queryAll();
    }else{
      System.out.println("不存在该联系人");
    }
  }

  private void addContact() {
    System.out.println("请输入联系人的信息");
    System.out.println("姓名:");
    String name = sc.nextLine();
    System.out.println("电话:");
    String tel = sc.nextLine();
    System.out.println("性别:");
    String sex = sc.nextLine();
    
    dao.add(name,tel,sex);
  }

  public static void main(String[] args) {
    new ContactAction();
  }

}

最后的XML文件是：

<?xml version="1.0" encoding="UTF-8" standalone="no"?><contacts>
  <user name="1234" pwd="1234">
    <contact id="2eaa2dbb32064cf0afd6a5e1013a6cbf">
      <name>simons</name>
      <tel>88888</tel>
      <sex>0</sex>
    </contact>
    <contact id="7fe2cf73804645f0abbca1303b78ec9a">
      <name>james</name>
      <tel>99999</tel>
      <sex>0</sex>
    </contact>
  <contact id="f49e5457dc144ca483fbbf82667cf7d3"><name>111</name><tel>56666</tel><sex>0</sex></contact></user>
  <user name="tom" pwd="1111">
    <contact id="688f0e262e5841ba8d6ceceffa9f9f4d">
      <name>lx</name>
      <tel>12333</tel>
      <sex>1</sex>
    </contact>
  </user>
</contacts>