XmlGenerator - Java에서 XML 생성

프로젝트 도중에 XML 파싱 및 생성을 해야 할 일이 생겼다.
하드 코딩하기에는 별로 마음에 안 들고 쓰기 편하게 생성해주는 클래스를 하나 만들었다.

Dom방식으로 하며, String으로 값을 받아서 해당하는 엘리먼트에 엘리먼트나 속성을 추가하는 형태이다.
최종 산출을 창에 뿌리거나 파일로 뽑아내게 해놓았다.

package com.galimit.gwf.plugin; import java.io.File; import java.io.FileOutputStream; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.parsers.ParserConfigurationException; import javax.xml.transform.OutputKeys; import javax.xml.transform.Transformer; import javax.xml.transform.TransformerConfigurationException; import javax.xml.transform.TransformerException; import javax.xml.transform.TransformerFactory; import javax.xml.transform.dom.DOMSource; import javax.xml.transform.stream.StreamResult; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.NodeList; /** * XmlGenerator * @author Chang-Hwan Han * @version 1.0 * @since 2012.03.08 */ public class XmlGenerator { private Document doc; private Transformer transformer = null; /** * Constructor */ public XmlGenerator() { DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance(); DocumentBuilder docBuilder = null; try { docBuilder = docFactory.newDocumentBuilder(); } catch (ParserConfigurationException e) { e.printStackTrace(); } // root element doc = docBuilder.newDocument(); // TransformerFactory transformerFactory = TransformerFactory.newInstance(); try { transformer = transformerFactory.newTransformer(); } catch (TransformerConfigurationException e) { e.printStackTrace(); } transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8"); transformer.setOutputProperty(OutputKeys.INDENT, "yes"); } public void addRootElement(String elementName) { Element rootElement = doc.createElement(elementName); doc.appendChild(rootElement); } public boolean addElement(String parentElementName, int index, String elementName, String elementValue) { NodeList nodeList = doc.getElementsByTagName(parentElementName); if(nodeList == null || nodeList.getLength() == 0) return false; Element element = doc.createElement(elementName); element.appendChild(doc.createTextNode(elementValue)); nodeList.item(index).appendChild(element); return true; } public boolean addAttribute(String elementName, int index, String attributeName, String attributeValue) { NodeList nodeList = doc.getElementsByTagName(elementName); if(nodeList == null) return false; Element element = (Element) nodeList.item(index); element.setAttribute(attributeName, attributeValue); return true; } private void generator(StreamResult result) throws TransformerException { DOMSource source = new DOMSource(doc); transformer.transform(source, result); } public boolean generator(String fileName) { try { StreamResult result = new StreamResult(new FileOutputStream(new File(fileName))); generator(result); } catch (Exception e) { e.printStackTrace(); return false; } return true; } public boolean print() { try { StreamResult result = new StreamResult(System.out); generator(result); } catch (Exception e) { e.printStackTrace(); return false; } return true; } public static void main(String args[]) { XmlGenerator generator = new XmlGenerator(); generator.addRootElement("Staff"); generator.addElement("Staff", 0, "b", "c"); generator.print(); } }