com.logicbus.backend.message.JsonMessage Maven / Gradle / Ivy
package com.logicbus.backend.message;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.anysoft.util.IOTools;
import com.anysoft.util.JsonTools;
import com.anysoft.util.Settings;
import com.jayway.jsonpath.spi.JsonProvider;
import com.jayway.jsonpath.spi.JsonProviderFactory;
import com.logicbus.backend.Context;
import com.logicbus.backend.message.Message;
/**
* 基于JSON的消息协议
* @author duanyy
*
* @since 1.2.1
*
* @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替代
* - 增加getContentType和getContentLength
*
* @version 1.6.7.9 [20170201 duanyy]
* - 采用SLF4j日志框架输出日志
*
* @version 1.6.7.15 [20170221 duanyy]
* - 输出时设置Content-Length以便支持keepalive
*/
public class JsonMessage implements Message {
protected static final Logger logger = LoggerFactory.getLogger(JsonMessage.class);
protected static JsonProvider provider = null;
static {
provider = JsonProviderFactory.createProvider();
}
protected static String formContentType = "application/x-www-form-urlencoded";
static {
formContentType = Settings.get().GetValue("http.formContentType",
"application/x-www-form-urlencoded");
}
private String contentType = "application/json;charset=utf-8";
/**
* Json结构的根节点
*/
protected Map root = null;
private long contentLength = 0;
@SuppressWarnings("unchecked")
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){
contentLength += data.getBytes().length;
JsonProvider provider = JsonProviderFactory.createProvider();
Object rootObj = provider.parse(data);
if (rootObj instanceof Map){
root = (Map)rootObj;
}
}
if (root == null){
root = new HashMap();
}
contentType = "application/json;charset=" + ctx.getEncoding();
}
public void finish(Context ctx,boolean closeStream) {
Map _root = getRoot();
JsonTools.setString(_root, "code", ctx.getReturnCode());
JsonTools.setString(_root, "reason", ctx.getReason());
JsonTools.setString(_root, "duration", String.valueOf(ctx.getDuration()));
JsonTools.setString(_root, "host", ctx.getHost());
JsonTools.setString(_root, "serial", ctx.getGlobalSerial());
OutputStream out = null;
try {
String data = provider.toJson(_root);
String jsonp = ctx.GetValue("jsonp", "");
if (jsonp != null && jsonp.length() > 0){
data = jsonp + "(" + data + ")";
}
out = ctx.getOutputStream();
ctx.setResponseContentType(contentType);
byte [] bytes = data.getBytes(ctx.getEncoding());
contentLength += bytes.length;
ctx.setResponseContentLength(bytes.length);
Context.writeToOutpuStream(out, bytes);
out.flush();
}catch (Exception ex){
logger.error("Error when writing data to outputstream",ex);
}finally{
if (closeStream)
IOTools.close(out);
}
}
/**
* 获取JSON结构的根节点
*
* @return JSON结构的根节点
*/
public Map getRoot(){
return root;
}
public String toString(){
return provider.toJson(root);
}
@Override
public String getContentType() {
return contentType;
}
@Override
public long getContentLength() {
return contentLength;
}
}