package com.ruoyi.other.webSocket;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandlerContext;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


public class NettyMsg {
	
	/**
	 * 返回一个正确数据
	 *
	 * @param method
	 * @param data
	 * @return
	 * @author TaoNingBo
	 */
	public static String setMsg(String method, Map<String, Object> data) {
		StringBuffer json = new StringBuffer();
		json.append(getHeader(200, "SUCCESS", method));
		json.append(JSON.toJSONString(data));
		json.append("}");
		//return JSON.toJSONString(json);
		return json.toString();
	}
	
	/**
	 * 返回一个正确数据
	 *
	 * @param method
	 * @param data
	 * @return
	 */
	public static String setMsg(String method, List<Map<String, Object>> data) {
		StringBuffer json = new StringBuffer();
		json.append(getHeader(200, "SUCCESS", method));
		List<JSONObject> jsonList = new ArrayList<JSONObject>();
		for (Map<String, Object> map : data) {
			JSONObject dataJson = new JSONObject(map);
			jsonList.add(dataJson);
		}
		json.append(jsonList);
		json.append("}");

//		return JSON.toJSONString(json);
		return json.toString();
	}
	
	/**
	 * 返回一个错误数据
	 *
	 * @param method
	 * @param data
	 * @return
	 * @author TaoNingBo
	 */
	public static String setErrMsg(String method, String data) {
		StringBuffer json = new StringBuffer();
		json.append(getHeader(-1, "FAILURE", method));
		json.append("\"" + data + "\"");
		json.append("}");
//		return JSON.toJSONString(json);
		return json.toString();
	}
	
	/**
	 * 生成一个返回JSON的头
	 *
	 * @param code
	 * @param msg
	 * @param method
	 * @return
	 * @author TaoNingBo
	 */
	private static String getHeader(int code, String msg, String method) {
		StringBuffer header = new StringBuffer();
		header.append("{");
		header.append("\"code\":\"" + code);
		header.append("\",\"msg\":\"" + msg);
		header.append("\",\"method\":\"" + method);
		header.append("\",\"data\":");
		return header.toString();
	}
	
	
}