package com.xjts.cipher.util;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.XMLWriter;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.annotation.JSONType;
import com.xjts.model.XmlEngine;
public class XmlGen {
/*****************************
* 组装请求xml串
* list 数据库中配置的报文头，报文体
* map 请求参数
*****************************/
public static String assembleRequestXml(List<XmlEngine> list, Map<String, String> map) {
Document doc = DocumentHelper.createDocument();
Element root = doc.addElement("message");
Element head = root.addElement("head");
Set<Map.Entry<String, String>> itermap = map.entrySet();
for(Map.Entry<String, String> entry: itermap){
String key = entry.getKey();
String value = entry.getValue();
for(int i = 0; i < list.size(); i++ ){
if(key.equals(list.get(i).getName().substring(10))){
list.get(i).setDefaultvalue(value);
}
}
}
for(int i = 0; i < list.size(); i++){
if("head".equals(list.get(i).getName().substring(0, 4))){
Element field = head.addElement("field");
field.addAttribute("name",
list.get(i).getName().substring(10));
String str = list.get(i).getDefaultvalue();
if (null != str && !"".equals(str)){
field.addAttribute("name",
list.get(i).getName().substring(10))
.setText(str);
}
}
}
Element body = root.addElement("body");
for(int i = 0; i < list.size(); i++){
if("body".equals(list.get(i).getName().substring(0, 4))){
Element field = body.addElement("field");
field.addAttribute("name",
list.get(i).getName().substring(10));
String str = list.get(i).getDefaultvalue();
if (null != str && !"".equals(str)){
field.addAttribute("name",
list.get(i).getName().substring(10))
.setText(str);
}
}
}
OutputFormat format = OutputFormat.createPrettyPrint();
//.createCompactFormat(); //createPrettyPrint() 输出格式化
format.setEncoding("gb2312");
StringWriter writer = new StringWriter();
XMLWriter output = new XMLWriter(writer, format);
try {
output.write(doc);
writer.close();
output.close();
// System.out.println(writer.toString());
} catch (IOException e) {
e.printStackTrace();
return null;
}
String content=writer.toString();
StringBuilder sb = new StringBuilder();
sb.append(content);
//加入 standalone="yes"
// sb.insert(37, " standalone="yes"");
return sb.toString();
}
/***********************************************
* 组装响应xml串
* list xml中head和body内的标签
* map xml中head和body内的标签的key-value
* listDetail 应答报文的明细数据，记录
***********************************************/
public static String assembleResponseXml(List<XmlEngine> list, Map<String, String> map
, List<Map<String, String>> listDetail){
Document doc = DocumentHelper.createDocument();
Element root = doc.addElement("message");
Element head = root.addElement("head");
Set<Map.Entry<String, String>> itermap = map.entrySet();
for(Map.Entry<String, String> entry: itermap){
String key = entry.getKey();
String value = entry.getValue();
for(int i = 0; i < list.size(); i++ ){
if(key.equals(list.get(i).getName().substring(10))){
list.get(i).setDefaultvalue(value);
}
}
}
for(int i = 0; i < list.size(); i++){
if("head".equals(list.get(i).getName().substring(0, 4))){
Element field = head.addElement("field");
field.addAttribute("pin",
"false");
field.addAttribute("name",
list.get(i).getName().substring(10));
String str = list.get(i).getDefaultvalue();
if (null != str && !"".equals(str)){
field.addAttribute("name",
list.get(i).getName().substring(10))
.addCDATA(str);
}else{
field.addAttribute("name",
list.get(i).getName().substring(10))
.addCDATA("");
}
}
}
Element body = root.addElement("body");
for(int i = 0; i < list.size(); i++){
if("body".equals(list.get(i).getName().substring(0, 4))){
Element field = body.addElement("field");
field.addAttribute("pin",
"false");
field.addAttribute("name",
list.get(i).getName().substring(10));
String str = list.get(i).getDefaultvalue();
if (null != str && !"".equals(str)){
field.addAttribute("name",
list.get(i).getName().substring(10))
.addCDATA(str);
}
}
}
if(listDetail != null){
if(listDetail.size() != 0){
Element field1 = body.addElement("field-list");
field1.addAttribute("name",
"ARRAY_" + list.get(0).getName().substring(4, 10));
for(int i = 0; i < listDetail.size(); i++){
Element field2 = field1.addElement("field-list");
field2.addAttribute("name",
"" + i);
Set<Map.Entry<String, String>> itermap1 = listDetail.get(i).entrySet();
for(Map.Entry<String, String> entry: itermap1){
String key = entry.getKey();
String value = entry.getValue();
Element field3 = field2.addElement("field");
field3.addAttribute("pin",
"false");
field3.addAttribute("name",
key).addCDATA(value);;
}
}
}
}
OutputFormat format = OutputFormat.createPrettyPrint();
//createCompactFormat(); //createPrettyPrint() 格式化
format.setEncoding("gb2312");
StringWriter writer = new StringWriter();
XMLWriter output = new XMLWriter(writer, format);
try {
output.write(doc);
writer.close();
output.close();
// System.out.println(writer.toString());
} catch (IOException e) {
e.printStackTrace();
return null;
}
return writer.toString();
}
/********************************************
* 针对List<Map<String, String>>类型的对象
* 转为Json串,此方法和 parseXmlStr(String xml)
* 有关联
********************************************/
public static String listMap2Json(List<Map<String, String>> listMap){
if(listMap == null) return "null";
String jsonXmlStr = JSON.toJSONString(listMap);
String[] splitStr = jsonXmlStr.split("\{"\d{1,2}":""\}");
String retStr = splitStr[0];
for(int i = 1; i < splitStr.length; i++){
retStr += splitStr[i].replaceAll("\},\{", ",").substring(1);
}
return retStr;
}
/*****************************
* 解析xml串，返回Map对象
*****************************/
public static List<Map<String, String>> parseXmlStr(String xml) throws Exception {
//final SAXReader sax = new SAXReader();// 创建一个SAXReader对象
//final File xmlFile = new File("E:\req_message.xml");// 根据指定的路径创建file对象
//final Document document = sax.read(xmlFile);// 获取document对象,如果文档无节点，则会抛出Exception提前结束
Document document = DocumentHelper.parseText(xml);
final Element root = document.getRootElement();// 获取根节点
List<Map<String, String>> listMap = new ArrayList<Map<String, String>>();
getNodes(root, listMap);// 从根节点开始遍历所有节点
return listMap;
}
/**
* 从指定节点Element node开始,递归遍历其所有子节点
*/
public static void getNodes(final Element node, List<Map<String,String>> listMap) {
//System.out.println("-------开始新节点-------------");
String str = node.getTextTrim();
// 当前节点的名称、文本内容和属性
//System.out.println("当前节点名称：" + node.getName());// 当前节点名称
//System.out.println("当前节点的内容：" + node.getTextTrim());// 当前节点内容
final List<Attribute> listAttr = node.attributes();// 当前节点的所有属性
for (final Attribute attr : listAttr) {// 遍历当前节点的所有属性
final String name = attr.getName();// 属性名称
final String value = attr.getValue();// 属性的值
//System.out.println("属性名称：" + name + "---->属性值：" + value);
if ("name".equals(name)) {
Map map = new LinkedHashMap();
map.put(value, str);
listMap.add( map);
}
}
// 递归遍历当前节点所有的子节点
final List<Element> listElement = node.elements();// 所有一级子节点的list
for (final Element e : listElement) {// 遍历所有一级子节点
getNodes(e, listMap);// 递归
}
}
/**
* 从指定节点Element node开始,递归遍历其所有子节点
*/
public static void getNodes(final Element node, Map map) {
String str = node.getTextTrim();
final List<Attribute> listAttr = node.attributes();
for (final Attribute attr : listAttr) {
final String name = attr.getName();
final String value = attr.getValue();
if (null != name && !"".equals(str)) {
if ("name".equals(name)) {
map.put(value, str);
}
}
}
final List<Element> listElement = node.elements();
for (final Element e : listElement) {
getNodes(e, map);//递归调用
}
}
public void saveDocument(Document document, File outputXml){
try {
OutputFormat format = OutputFormat.createPrettyPrint();
XMLWriter output = new XMLWriter(new FileWriter(outputXml),
format);
output.write(document);
output.close();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
public static void main(String [] argv){
XmlGen dom4j = new XmlGen();
Document document = null;
dom4j.saveDocument(document, new File("output.xml"));
//System.out.println(assembleResponseXml(null, null));
}
public class StandaloneWriter extends XMLWriter {
protected void writeDeclaration() throws IOException {
OutputFormat format = getOutputFormat();
String encoding = format.getEncoding();
if (!format.isSuppressDeclaration()) {
writer.write("<?xml version="1.0"");
if (!format.isOmitEncoding()) {
if (encoding.equals("UTF8"))
writer.write(" encoding="UTF-8"");
else
writer.write(" encoding="" + encoding + """);
}
writer.write(" standalone="true"");
writer.write("?>");
if (format.isNewLineAfterDeclaration()) { println(); }
}
}
}
}