com.logicbus.backend.message.XMLMessage Maven / Gradle / Ivy
package com.logicbus.backend.message;
import java.io.InputStream;
import java.io.OutputStream;
import javax.xml.parsers.ParserConfigurationException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import com.anysoft.util.IOTools;
import com.anysoft.util.Settings;
import com.anysoft.util.XmlTools;
import com.logicbus.backend.Context;
/**
* XML 消息
* @author duanyy
*
* @version 1.0.4 [20140410 duanyy]
* - 提升encoding为父类成员
*
* @version 1.0.5 [20140412 duanyy]
* - 修改消息传递模型。
*
* @version 1.4.0 [20141117 duanyy]
* - Message被改造为接口
*
* @version 1.6.1.1 [20141118 duanyy]
* - 修正没有读入的情况下,root为空的bug
* - MessageDoc暴露InputStream和OutputStream
*
* @version 1.6.1.2 [20141118 duanyy]
* - 支持MessageDoc的Raw数据功能
*
* @version 1.6.2.1 [20141223 duanyy]
* - 增加对Comet的支持
*
* @version 1.6.3.14 [20150409 duanyy]
* - 修正formContentType所取的参数名问题，笔误
*
* @version 1.6.5.6 [20160523 duanyy]
* - 淘汰MessageDoc，采用Context替代
*
* @version 1.6.7.9 [20170201 duanyy]
* - 采用SLF4j日志框架输出日志
*/
public class XMLMessage implements Message {
protected static final Logger logger = LoggerFactory.getLogger(XMLMessage.class);
protected Document xmlDoc = null;
protected Element root = null;
protected String contentType = "text/xml;charset=utf-8";
public Document getDocument(){return xmlDoc;}
public Element getRoot(){return root;}
public void init(Context ctx) {
String data = null;
{
byte [] inputData = ctx.getRequestRaw();
if (inputData != null){
try {
data = new String(inputData,ctx.getEncoding());
}catch (Exception ex){
}
}
}
if (data == null){
//当客户端通过form来post的时候，Message不去读取输入流。
String _contentType = ctx.getRequestContentType();
if (_contentType == null || !_contentType.startsWith(formContentType)){
InputStream in = null;
try {
in = ctx.getInputStream();
data = Context.readFromInputStream(in, ctx.getEncoding());
}catch(Exception ex){
logger.error("Error when reading data from inputstream",ex);
}finally{
IOTools.close(in);
}
}
}
if (data != null && data.length() > 0){
try {
xmlDoc = XmlTools.loadFromContent(data);
} catch (Exception ex) {
logger.error("Error when parsing data from xml",ex);
}
}
if (xmlDoc == null){
try {
xmlDoc = XmlTools.newDocument("root");
} catch (ParserConfigurationException e) {
}
}
root = xmlDoc.getDocumentElement();
contentType = "text/xml;charset=" + ctx.getEncoding();
}
public void finish(Context ctx,boolean closeStream) {
root.setAttribute("code",ctx.getReturnCode());
root.setAttribute("reason", ctx.getReason());
root.setAttribute("duration", String.valueOf(ctx.getDuration()));
root.setAttribute("host", ctx.getHost());
root.setAttribute("serial", ctx.getGlobalSerial());
OutputStream out = null;
try {
ctx.setResponseContentType(contentType);
out = ctx.getOutputStream();
XmlTools.saveToOutputStream(xmlDoc, out);
out.flush();
} catch (Exception ex) {
logger.error("Error when writing data to outputstream",ex);
}finally {
if (closeStream)
IOTools.closeStream(out);
}
}
public String toString(){
return XmlTools.node2String(xmlDoc);
}
protected static String formContentType = "application/x-www-form-urlencoded";
static {
formContentType = Settings.get().GetValue("http.formContentType",
"application/x-www-form-urlencoded");
}
@Override
public String getContentType() {
return contentType;
}
@Override
public long getContentLength() {
return 0;
}
}