package com.webserver.webservice;

 import java.io.ByteArrayInputStream;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.net.SocketTimeoutException;

 import java.text.SimpleDateFormat;

 import java.util.Date;

 import java.util.Iterator;

 import java.util.Map;

 import java.util.Set;

 import java.util.concurrent.ConcurrentHashMap;

 import javax.servlet.ServletException;

 import javax.servlet.annotation.WebServlet;

 import javax.servlet.http.HttpServlet;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import org.apache.http.Header;

 import org.apache.http.HttpResponse;

 import org.apache.http.client.ClientProtocolException;

 import org.apache.http.client.HttpClient;

 import org.apache.http.client.methods.HttpPost;

 import org.apache.http.entity.BasicHttpEntity;

 import org.apache.http.impl.client.DefaultHttpClient;

 import org.apache.http.params.CoreConnectionPNames;

 import org.apache.http.util.EntityUtils;

 /**

  * Servlet implementation class Prompt

  */

 @WebServlet("/Prompt")

 public class Prompt extends HttpServlet {

     private static final long serialVersionUID = 1L;

     private static Map<String, CookieData> cookieMap  = new ConcurrentHashMap<String,CookieData>();

     public static String yuyinServer = "http://10.3.172.205:8080/RecognizePY/directRecognize";

     /**

      * @see HttpServlet#HttpServlet()

      */

     public Prompt() {

         super();

         // TODO Auto-generated constructor stub

     }

     /**

      * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)

      */

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

         // TODO Auto-generated method stub

         this.doPost(request, response);

     }

     /**

      * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)

      */

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

         String session = request.getParameter("session").trim();

         //String token = request.getParameter("token").trim();

         response.setCharacterEncoding("UTF-8");

         response.setContentType("text/json;charset=utf-8");

         String firstCookie = "";

         //从原型管理器中获取原型对象

         CookieData cookie = cookieMap.get(session);

         //原型管理器包含该对象

         if(cookie!=null){

             firstCookie = cookie.getCookie();

             cookie.setAccessDate(System.currentTimeMillis());

         } else {

             //不包含就创建一个对象 并存到原型管理器中

             CookieData tmpCookie = new CookieData();

             firstCookie = getCookie();

             tmpCookie.setCookie(firstCookie);

             tmpCookie.setAccessDate(System.currentTimeMillis());

             cookieMap.put(session, tmpCookie);

             //根据具体情况 原型对象在原型管理器中存放时间超过规定时间移除改原型对象

             cleanCookieMap();

         }

         //TODO

         //....

     }

     /**

      * 客户端要记住返回的cookie

      * @return

      * @throws ClientProtocolException

      * @throws IOException

      */

     public String getCookie() throws ClientProtocolException, IOException{

         String firstCookie = "";

         try{

             HttpClient client = new DefaultHttpClient();

             HttpPost post = new HttpPost(yuyinServer);

             //设置连接超时时间

             client.getParams().setParameter(CoreConnectionPNames.CONNECTION_TIMEOUT, 2000);

             //设置数据传输超时时间

             client.getParams().setParameter(CoreConnectionPNames.SO_TIMEOUT, 6000);

             HttpResponse response = null;

             response = client.execute(post);

             if(response.getStatusLine().getStatusCode()==200){

                 //如果状态码为200,就是正常返回

                 Header[] cookies = response.getHeaders("Set-Cookie");

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

                     Header curr = cookies[i];

                     firstCookie += curr.getValue();

                 }

             }

         }catch(SocketTimeoutException e){

             firstCookie = null;

         }

         return firstCookie;

     }

     /**

      * 送语音数据，拿到识别结果

      * @param out

      * @return

      * @throws IOException

      * @throws ClientProtocolException

      */

     public String getReconizedResult(byte[] wavbt,String cookie) throws ClientProtocolException, IOException{

         String retStr = "";

         try {

             HttpClient client = new DefaultHttpClient();

             client.getParams().setParameter(CoreConnectionPNames.CONNECTION_TIMEOUT, 2000);

             client.getParams().setParameter(CoreConnectionPNames.SO_TIMEOUT, 6000);

             HttpPost post = new HttpPost(yuyinServer);

             post.addHeader("Cookie",cookie);

             //以流的方式传参数（有两种方式）

             //第一中方式

             //InputStreamEntity entity  = new InputStreamEntity(new ByteArrayInputStream(wavbt), wavbt.length);

             //第二种方式

             BasicHttpEntity entity = new BasicHttpEntity();

             //设置内容

             entity.setContent(new ByteArrayInputStream(wavbt));

             //设置长度

             entity.setContentLength(wavbt.length);

             post.setEntity(entity);

             HttpResponse response= null;

             response = client.execute(post);

             String respStr = EntityUtils.toString(response.getEntity());

             org.json.JSONObject retJsonObj = new org.json.JSONObject(respStr);

             retStr = retJsonObj.getString("result");

             saveFile("success",wavbt);

         } catch (Exception e) {

             System.out.println(e.getMessage());

             saveFile("fail",wavbt);

         }

         return retStr;

     }

     public static void saveFile(String flag,byte[] wavbt) {

         SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd-HH-mm-ss");

         try (

             FileOutputStream fos = new FileOutputStream("d:/test"+flag+ df.format(new Date()) +".wav");

         ) {

             fos.write(wavbt);

         } catch (Exception e) {

         }

     }

     //自定义的原型对象

     class CookieData{

         /** 访问cookie */

         private String cookie;

         /** 访问时间 */

         private long accessDate;

         /**

          * 返回 cookie

          * @return cookie

          */

         public String getCookie() {

             return cookie;

         }

         /**

          * 设置 cookie

          * @param cookie

          */

         public void setCookie(String cookie) {

             this.cookie = cookie;

         }

         /**

          * 返回 accessDate

          * @return accessDate

          */

         public long getAccessDate() {

             return accessDate;

         }

         /**

          * 设置 accessDate

          * @param accessDate

          */

         public void setAccessDate(long accessDate) {

             this.accessDate = accessDate;

         }

     }

     //存放时间超过规定时间后清除原型对象

     public void cleanCookieMap(){

         long currTime = System.currentTimeMillis();

         Set<String> keySet = cookieMap.keySet();

         Iterator<String> it = keySet.iterator();

         while(it.hasNext()){

             String key = it.next();

             CookieData cookie = cookieMap.get(key);

             //如果cookie保存时间超过一小时 则移除

             if(currTime - cookie.getAccessDate()> 3600000 ){

                 cookieMap.remove(key);

             }

         }

     }

 }