使用Groovy的CSV到XML转换器

时间:2014-01-29 11:37:29

标签: xml csv groovy

我的能力已经达到了我想要实现的目标,我需要使用groovy创建一个转换器,将CSV转换为XML。 我的能力非常有限,因此以下代码可能是一堆垃圾,但它是使用我发现的其他代码组合在一起。

package com.ns.extension.tee;

import com.ns.argo.business.api.ArgoUtils
import com.ns.argo.business.api.GroovyApi
import org.apache.log4j.Logger
import org.jdom.Document
import org.jdom.Element

import javax.xml.parsers.DocumentBuilder
import javax.xml.parsers.DocumentBuilderFactory
import javax.xml.transform.Result
import javax.xml.transform.Source
import javax.xml.transform.Transformer
import javax.xml.transform.TransformerFactory
import javax.xml.transform.dom.DOMSource
import javax.xml.transform.stream.StreamResult

/**
* Converting CSV to XML.
*
* * Date: 24/01/14 : 11:50
* Called from: Groovy Job.
*/
public class CSVtoXML extends GroovyApi {

/**
* Triggered from a Groovy Job.
* @param param
*/
public void execute(Map param) {
LOGGER.warn(String.format("At start of %s at %s", this.getClass().getName(), ArgoUtils.timeNow()));
convertCSVToXML();
LOGGER.warn(String.format("At end of %s at %s", this.getClass().getName(), ArgoUtils.timeNow()));
}

/**
* Loads and converts a file from CSV to XML.
*/
private void convertCSVToXML() {
List<String> headers = new ArrayList<String>(5);

String path = "C:\\convert\\";
path = path.replaceAll("\\", "/");
File file = new File(path, "CRR.csv");
BufferedReader reader = null;

try {

DocumentBuilderFactory domFactory = DocumentBuilderFactory.newInstance();
DocumentBuilder domBuilder = domFactory.newDocumentBuilder();

Document newDoc = domBuilder.newDocument() as Document;
// Root element
Element rootElement = newDoc.createElement("XMLCreators");
newDoc.appendChild(rootElement);

reader = new BufferedReader(new FileReader(file));
int line = 0;

String text = null;
while ((text = reader.readLine()) != null) {

StringTokenizer st = new StringTokenizer(text, " ", false);
String[] rowValues = new String[st.countTokens()];
int index = 0;
while (st.hasMoreTokens()) {

  String next = st.nextToken();
  rowValues[index++] = next;

}

//String[] rowValues = text.split(",");

if (line == 0) { // Header row
  for (String col : rowValues) {
    headers.add(col);
  }
} else { // Data row
  Element rowElement = newDoc.createElement("row");
  rootElement.appendChild(rowElement);
  for (int col = 0; col < headers.size(); col++) {
    String header = headers.get(col);
    String value = null;

    if (col < rowValues.length) {
      value = rowValues[col];
    } else {
      // ?? Default value
      value = "";
    }

    Element curElement = newDoc.createElement(header);
    curElement.appendChild(newDoc.createTextNode(value));
    rowElement.appendChild(curElement);
  }
}
line++;
}

ByteArrayOutputStream baos = null;
OutputStreamWriter osw = null;

try {
baos = new ByteArrayOutputStream();
osw = new OutputStreamWriter(baos);

TransformerFactory tranFactory = TransformerFactory.newInstance();
Transformer aTransformer = tranFactory.newTransformer();
aTransformer.setOutputProperty(OutputKeys.INDENT, "yes");
aTransformer.setOutputProperty(OutputKeys.METHOD, "xml");
aTransformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");

Source src = new DOMSource(newDoc);
Result result = new StreamResult(osw);
aTransformer.transform(src, result);

osw.flush();
System.out.println(new String(baos.toByteArray()));
} catch (Exception exp) {
exp.printStackTrace();
} finally {
try {
  osw.close();
} catch (Exception e) {
}
try {
  baos.close();
} catch (Exception e) {
}
}
} catch (Exception e) {
e.printStackTrace();
}
}

private final Logger LOGGER = Logger.getLogger(CSVtoXML.class);
}

我希望它以下面的格式输出,所有这些都是不同的值,这只是一个例子。

<?xml version="1.0" encoding="UTF-8"?>
<XMLCreators>
<row>
<Name>chi</Name>
<Age>23</Age>
<sex></sex>
</row>
<row>
<Name>kay</Name>
<Age>19</Age>
<sex>male</sex>
</row>
<row>
<Name>john</Name>
<Age></Age>
<sex>male</sex>
 </row>
 </XMLCreators>

示例CSV:

Job         Type        Cntr Number     Date            Booking Ref     Commodity   Weight      Special Instructions
JOB0001     Circle      12              31/09/2013      Book0001        1           100.00      Carry
JOB0002     Square      13              31/11/2013      Book0001        2           200.00      None
JOB0003     Cube        15              31-Dec-13       Book0001        3           300.00      Hide

-------------------------------------------- ------------------------------------------------

03/02/2014当前代码

package com.ns.extension.tee;

import com.ns.argo.business.api.ArgoUtils
import com.ns.argo.business.api.GroovyApi
import static com.xlson.groovycsv.CsvParser.parseCsv
import groovy.xml.*
import org.apache.xalan.*

/**
 * Converting CSV to XML.
 * Date: 24/01/14 : 11:50
 * Called from: Groovy Job.
 */
public class CSVtoXML extends GroovyApi {

    public void execute(Map param) {
        new File( 'C:/convert/CRR_output.xml' ).withWriter { w ->
            new File( 'C:/convert/CRR.csv' ).withReader { r ->
                def csvParser = parseCsv( r , speerator: ',')
                w.println new StreamingMarkupBuilder().bind {
                    XMLCreators {
                        csvParser.each { line ->
                            println line
                            if( line ['Job'] ){
                                row {
                                    job( line[ 'Job' ] )
                                    type( line[ 'Type' ] )
                                    cntr( line[ 'Cntr Number' ] )
                                    date( line[ 'Date' ] )
                                    bref( line[ 'Booking Ref' ] )
                                    comm( line[ 'Commodity' ] )
                                    weig( line[ 'Weight' ] )
                                    spci( line[ 'Special Instructions' ] )
                                }
                            }}
                    }
                }
            }
        }

    }
    static main( args ) {
        new CSVtoXML().execute( [:] )
    }
}

1 个答案:

答案 0 :(得分:9)

因此,假设您在问题中放入的示例CSV位于名为"mycsv.csv"的文件中,您可以这样做:

@Grab('com.xlson.groovycsv:groovycsv:1.0')
import static com.xlson.groovycsv.CsvParser.parseCsv
import groovy.xml.*

println XmlUtil.serialize( new File( 'mycsv.csv' ).withReader { r ->
    def csvParser = parseCsv( r, separator:'\t' )
    new StreamingMarkupBuilder().bind {
        XMLCreators {
            csvParser.each { line ->
                row {
                    job( line[ 'Job' ] )
                    type( line[ 'Type' ] )
                    cntr( line[ 'Cntr Number' ] )
                }
            }
        }
    }
} )

打印出来:

<?xml version="1.0" encoding="UTF-8"?><XMLCreators>
  <row>
    <job>JOB0001</job>
    <type>Circle</type>
    <cntr>12</cntr>
  </row>
  <row>
    <job>JOB0002</job>
    <type>Square</type>
    <cntr>13</cntr>
  </row>
  <row>
    <job>JOB0003</job>
    <type>Cube</type>
    <cntr>15</cntr>
  </row>
</XMLCreators>

如果您不能执行@Grab并且您只需要将jar添加到类路径中,那么主要的groovycsv jar(及其依赖项)are listed here

在上面详述的类结构中编辑,格式化:

package com.navis.extension.tee;

import com.navis.argo.business.api.ArgoUtils
import com.navis.argo.business.api.GroovyApi
import static com.xlson.groovycsv.CsvParser.parseCsv
import groovy.xml.*

/**
 * Converting CSV to XML.
 *
 * * Date: 24/01/14 : 11:50
 * Called from: Groovy Job.
 */
public class CSVtoXML extends GroovyApi {
    public void execute(Map param) {
       new File( 'C:/convert/CRR_output.xml' ).withWriter { w ->
            new File( 'C:/convert/CRR.csv' ).withReader { r ->
                def csvParser = parseCsv( r , separator: ',')
                String xml = new StreamingMarkupBuilder().bind {
                    XMLCreators {
                        csvParser.each { line ->
                            println line
                            if( line.size() ) {
                                row {
                                    job( line[ 'Job' ] )
                                    type( line[ 'Type' ] )
                                    cntr( line[ 'Cntr Number' ] )
                                    date( line[ 'Date' ] )
                                    bref( line[ 'Booking Ref' ] )
                                    comm( line[ 'Commodity' ] )
                                    weig( line[ 'Weight' ] )
                                    spci( line[ 'Special Instructions' ] )
                                }
                            }
                        }
                    }
                }
                w.writeLine( xml )
            }
        }
    }
    static main( args ) {
        new CSVtoXML().execute( [:] )
    }
}