微信公共平台(二)自动回复消息

package com.chaowen;



import java.io.IOException;

import java.io.InputStream;

import java.io.PrintWriter;

import java.io.StringReader;

import java.io.Writer;

import java.net.MalformedURLException;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import java.util.Arrays;

import java.util.Date;

import java.util.Scanner;



import javax.servlet.ServletException;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.xml.bind.JAXBContext;

import javax.xml.bind.JAXBException;

import javax.xml.bind.Marshaller;

import javax.xml.bind.Unmarshaller;



import com.chaowen.bean.WeChatReqBean;

import com.chaowen.bean.WeChatRespBean;

import com.chaowen.common.Constants;

import com.sun.xml.bind.marshaller.CharacterEscapeHandler;



/**

 * Servlet implementation class WeiXinServlet

 */

public class WeiXinServlet extends HttpServlet {

	private static final long serialVersionUID = 1L;

	public static final String Token = "token";   







	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

		response.setCharacterEncoding("UTF-8"); 

		PrintWriter   out   =   response.getWriter();



		String signature = request.getParameter("signature");

		String timestamp = request.getParameter("timestamp");

		String nonce = request.getParameter("nonce");

		String[] ArrTmp = { Token, timestamp, nonce };

		Arrays.sort(ArrTmp);

		StringBuffer sb = new StringBuffer();

		for (int i = 0; i < ArrTmp.length; i++) {

			sb.append(ArrTmp[i]);

		}

		String pwd = Encrypt(sb.toString());

		String echostr = request.getParameter("echostr");

		System.out.println("pwd=="+pwd);

		System.out.println("echostr=="+echostr);

		if(pwd.equals(signature)){

			if(!"".equals(echostr) && echostr != null){

				response.getWriter().print(echostr);

			}

		}

		

		out.flush();

		out.close();

		//this.doPost(request, response);

	}





	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

		 Scanner scanner = new Scanner(request.getInputStream());

         response.setContentType("application/xml");

         response.setCharacterEncoding("UTF-8"); 

         PrintWriter out = response.getWriter();

         try {

        	// 1、获取用户发送的信息

        	

        	 InputStream is = request.getInputStream();

         	// 取HTTP请求流长度

              int size = request.getContentLength();

              // 用于缓存每次读取的数据

              byte[] buffer = new byte[size];

              // 用于存放结果的数组

              byte[] xmldataByte = new byte[size];

              int count = 0;

              int rbyte = 0;

              // 循环读取

              while (count < size) { 

                  // 每次实际读取长度存于rbyte中

                  rbyte = is.read(buffer); 

                  for(int i=0;i<rbyte;i++) {

                      xmldataByte[count + i] = buffer[i];

                  }

                  count += rbyte;

              }

              is.close();

              String requestStr = new String(xmldataByte, "UTF-8");

        	

        	

        	 // 2、解析用户的信息

        	 JAXBContext jc = JAXBContext.newInstance(WeChatReqBean.class);

        	 Unmarshaller u = jc.createUnmarshaller();

        	 WeChatReqBean reqBean = (WeChatReqBean) u.unmarshal(new StringReader(requestStr));

        	 

        	// 3、判定用户了解动作

        	 String content = getContent(reqBean);

        	// 4、创建一个回复消息

        	 jc = JAXBContext.newInstance(WeChatRespBean.class);

        	 Marshaller m = jc.createMarshaller();

        	 m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);

        	 m.setProperty(CharacterEscapeHandler.class.getName(), new CharacterEscapeHandler() {

        		 @Override

        		 public void escape(char[] arg0, int arg1, int arg2, boolean arg3,

        				 Writer arg4) throws IOException {

        			 arg4.write(arg0, arg1, arg2);

        		 }

        	 });

        	 m.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);

        	 WeChatRespBean respBean = createRespBean(reqBean, content);

        	 m.marshal(respBean, out);

        	 out.flush();





		}catch (JAXBException e) {

           e.printStackTrace();

        } finally {

            if (scanner != null) {

                scanner.close();

                scanner = null;

            }

            if (out != null) {

                out.close();

                out = null;

            }

        }

	}



	/**

	 * @param reqBean

	 * @param content

	 * @return

	 */

	private WeChatRespBean createRespBean(WeChatReqBean reqBean, String content) {

		WeChatRespBean respBean = new WeChatRespBean();

		respBean.setFromUserName(reqBean.getToUserName());

		respBean.setToUserName(reqBean.getFromUserName());

		respBean.setMsgType("text");

		respBean.setCreateTime(new Date().getTime());

		respBean.setContent(content);

		return respBean;

	}

	

	public String Encrypt(String strSrc) {

		MessageDigest md = null;

		String strDes = null;



		byte[] bt = strSrc.getBytes();

		try {

			md = MessageDigest.getInstance("SHA-1");

			md.update(bt);

			strDes = bytes2Hex(md.digest()); //to HexString

		} catch (NoSuchAlgorithmException e) {

			System.out.println("Invalid algorithm.");

			return null;

		}

		return strDes;

	}



	public String bytes2Hex(byte[] bts) {

		String des = "";

		String tmp = null;

		for (int i = 0; i < bts.length; i++) {

			tmp = (Integer.toHexString(bts[i] & 0xFF));

			if (tmp.length() == 1) {

				des += "0";

			}

			des += tmp;

		}

		return des;

	}

	

	/**

	 * @param reqBean

	 * @throws JAXBException

	 * @throws MalformedURLException

	 */

	private String getContent(WeChatReqBean reqBean) throws JAXBException, MalformedURLException {

		StringBuffer content = new StringBuffer("");

		

		switch (Constants.getType(reqBean.getMsgType())){

		case text:

			if (reqBean.getContent().contains("成长档案")) {

				// 这是新用户关注时默认发的一条信息。可以做一个欢迎处理。

				content.append("这是成长档案的介绍!"+"\r\n");

				content.append(""+"\r\n");

				content.append(""+"\r\n");

				content.append("回复h获取更多内容!"+"\r\n");

			}else if(reqBean.getContent().contains("h")){

				content.append("这是帮助信息!"+"\r\n");

				content.append("如果你还有疑问或建议:请以#开头发送文字内容，我们的客服收到留言后会尽快跟你联系。"+"\r\n");

			}else if(reqBean.getContent().contains("cp")){

				content.append("产品介绍:"+"\r\n");

				content.append("001.家园共育平台"+"\r\n");

				content.append("002.运营分析系统"+"\r\n");

				content.append("003.互动学堂"+"\r\n");

				content.append(""+"\r\n");

				content.append(""+"\r\n");

				content.append("回复数字获取更多内容!"+"\r\n");

			}

			

			else {

				content.append("你的信息已收到，我们会第一时间处理.");

			}

			break;

		case event:

			if("subscribe".equals(reqBean.getEvent())){

				// 这是新用户关注时默认发的一条信息。可以做一个欢迎处理。

				content.append("非常感谢你的添加，我们定时为你提供最新的儿童内容服务哟!"+"\r\n");

				content.append("1.想了解我们的产品请直接将产品名称回复给我(如\"成长档案\"）"+"\r\n");

				content.append("2.想知道已经上线了哪些产品请回复\"cp\"或直接点击链接www.yey.com进行查看"+"\r\n");

			    content.append("3.如需帮助或了解请回复\"h\"。"+"\r\n");

			}

			break;

		case image:

			

			break;

		default:

			content.append("这是一个测试!!!");

			break;

		}

		return content.toString();

	}

}