Javafx做的密码管理程序

package dataDao; import java.io.File; import java.io.PrintWriter; import java.util.ArrayList; import java.util.Iterator; import java.util.List; import org.dom4j.Document; import org.dom4j.Element; import org.dom4j.io.OutputFormat; import org.dom4j.io.SAXReader; import org.dom4j.io.XMLWriter; /** * @author GaoShuhang */ public class DataDaoImpl implements DataDao { private String getPath() { String path = this.getClass().getResource("").getPath(); path = path.substring(0, path.length() - 8); path += "data.xml"; return path; } @Override public void addPassword(Data data) throws Exception { String path = this.getPath(); SAXReader sax = new SAXReader(); String item = data.getItem(); String password = data.getPassword(); Document doc = sax.read(path); Element root = doc.getRootElement(); Element userNode = root.addElement("node"); userNode.addAttribute("item", item); userNode.addAttribute("password", password); OutputFormat format = OutputFormat.createPrettyPrint(); XMLWriter writer = new XMLWriter(new PrintWriter(path, "utf-8"), format); writer.write(doc); writer.close(); } @Override public void removePassword(String item) throws Exception { SAXReader sax = new SAXReader(); String path = this.getPath(); Document doc = sax.read(new File(path)); Element root = doc.getRootElement(); for ( @SuppressWarnings("rawtypes") Iterator i = root.elementIterator("node"); i.hasNext(); ) { Element node = (Element) i.next(); if(node.attributeValue("item").equals(item)) { root.remove(node); } } OutputFormat format = OutputFormat.createPrettyPrint(); XMLWriter writer = new XMLWriter(new PrintWriter(path, "UTF-8"), format); writer.write(doc); writer.close(); } @Override public boolean haveItem(String item) throws Exception { SAXReader sax = new SAXReader(); String path = this.getPath(); Document doc = sax.read(path); Element root = doc.getRootElement(); for ( @SuppressWarnings("rawtypes") Iterator i = root.elementIterator("node"); i.hasNext(); ) { Element node = (Element) i.next(); if(node.attributeValue("item").equals(item)) { return true; } } return false; } @Override public List<Data> listData() throws Exception { String path = this.getPath(); List<Data> arr = new ArrayList<>(); SAXReader sax = new SAXReader(); Document doc = sax.read(path); Element root = doc.getRootElement(); for ( @SuppressWarnings("rawtypes") Iterator i = root.elementIterator("node"); i.hasNext(); ) { Element node = (Element) i.next(); arr.add(new Data(node.attributeValue("item"), node.attributeValue("password"))); } return arr; } }