From 5a8bbcb100f37b8e17d6fda0d851d96dbace46e4 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期一, 04 十一月 2019 20:17:42 +0800
Subject: [PATCH] fix recv msg

---
 src/main/java/com/cloud/count/listener/CountThread.java |  170 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 169 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/cloud/count/listener/CountThread.java b/src/main/java/com/cloud/count/listener/CountThread.java
index d2236b8..abd3000 100644
--- a/src/main/java/com/cloud/count/listener/CountThread.java
+++ b/src/main/java/com/cloud/count/listener/CountThread.java
@@ -1,4 +1,172 @@
 package com.cloud.count.listener;
 
-public class CountThread {
+import com.cloud.count.dao.CountDao;
+import com.cloud.count.model.CountState;
+import com.cloud.count.model.People;
+import com.cloud.count.service.serviceImpl.CountServiceImpl;
+import com.cloud.count.utils.XmlUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import java.io.*;
+import java.net.Socket;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+
+public class CountThread implements Runnable {
+    private Logger logger = LoggerFactory.getLogger(this.getClass());
+    static boolean isServerReboot = false;
+
+    private CountDao dao;
+
+    static final String CRLF = System.getProperty("line.separator");
+
+    private Socket client;
+
+    public CountThread(Socket socket) {
+        this.dao = BeanContext.getBean("countDao");
+        this.client = socket;
+    }
+
+    String encoding = "UTF-8";
+    BufferedReader iReader = null;
+    BufferedWriter iWriter = null;
+
+    @Override
+    public void run() {
+        try{
+            iReader = new BufferedReader(new InputStreamReader(client.getInputStream(), encoding));
+            iWriter = new BufferedWriter(new OutputStreamWriter(client.getOutputStream(), encoding));
+
+            while (true) {
+                String str = iReader.readLine();
+                sendAck(iWriter);
+
+                String xmlStr = XmlUtil.toXmlStr(str);
+
+                // 濡傛灉瀛樺湪TotalEnters銆傚垯璇佹槑杩斿洖鐨勬槸瀹炴椂鏁版嵁鐨剎ml
+                People people = null;
+                if(xmlStr.indexOf("TotalEnters")>-1){
+                    people = XmlUtil.realTimeXmlToPeople(xmlStr);
+                    int inCreEnter = CountServiceImpl.addEnter(people.getEnters());//杩涘叆浜烘暟澧為噺
+                    int inCreExit = CountServiceImpl.addExit(people.getExits());//鍑哄幓浜烘暟澧為噺
+                    CountServiceImpl.totalEnters += inCreEnter;
+                    CountServiceImpl.totalExits += inCreExit;
+
+                    people.setEnters(CountServiceImpl.totalEnters);//灏嗙疮鍔犲悗鐨勬暟閲忓瓨鍒版暟鎹簱涓�
+                    people.setExits(CountServiceImpl.totalExits);
+                    boolean resultB = dao.savePeople(people);
+                    System.out.println("totalEnters:"+CountServiceImpl.totalEnters);
+
+                    //鎬昏繘鍏ユ暟閲忔垨鑰呮�诲嚭鍘绘暟閲忥紝涓嶄竴瀹氬摢涓�涓厛鍒�2瀛楄妭int鐨勪笂闄�65535
+//                                if( CountServiceImpl.totalEnters < CountServiceImpl.baseEnters){
+//                                    //琛ㄧず缁熻璁惧閲嶆柊鍚姩浜�,缁熻鍙傛暟褰掗浂
+//                                    CountServiceImpl.initCountArgs(people);
+//                                }
+                    op: if(!isServerReboot){
+                        isServerReboot = true;
+                        CountState state = dao.getCountState();//鎭㈠缁熻鐘舵��
+                        Date nowDate = new Date();
+                        Date BaseDate = new Date(state.getBaseTime()*1000L);
+                        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
+                        if(!sdf.format(nowDate).equals(sdf.format(BaseDate))){//鍘嗗彶鏁版嵁涓嶇瓑浜庡綋澶�
+                            CountServiceImpl.baseEnters  = people.getEnters();
+                            CountServiceImpl.baseExits = people.getExits();
+                            CountServiceImpl.realtimeBaseEnters = people.getEnters();
+                            CountServiceImpl.realtimebBaseExits =  people.getExits();
+                            CountServiceImpl.errorCount = 0;
+                            break op;
+                        }
+                        //杩樺師鐘舵��
+                        CountServiceImpl.baseEnters  = state.getBaseEnters();
+                        CountServiceImpl.baseExits = state.getBaseExits();
+                        CountServiceImpl.totalEnters = state.getTotalEnters();
+                        CountServiceImpl.totalExits = state.getTotalExits();
+                        CountServiceImpl.realtimeBaseEnters = state.getRealtimeBaseEnters();
+                        CountServiceImpl.realtimebBaseExits = state.getRealtimeBaseExits();
+                        CountServiceImpl.errorCount = state.getErrorCount();
+                    }
+
+                    //鎬昏繘鍏ヤ汉鏁板皬浜庢�诲嚭鍘荤殑浜烘暟锛岃〃绀虹粺璁¤澶囧湪鏀惧鏃堕棿锛堝嚭鍘荤殑浜烘暟澶т簬杩涘叆鐨勪汉鏁帮級閲嶅惎浜�
+                    if(CountServiceImpl.totalEnters < CountServiceImpl.totalExits){
+                        CountServiceImpl.errorCount = CountServiceImpl.totalExits - CountServiceImpl.totalEnters;
+                    }else
+                        CountServiceImpl.errorCount = 0;
+
+                    CountState currentState = new CountState();
+                    currentState.setBaseEnters(CountServiceImpl.baseEnters);
+                    currentState.setBaseExits(CountServiceImpl.baseExits);
+                    currentState.setTotalEnters(CountServiceImpl.totalEnters);
+                    currentState.setTotalExits(CountServiceImpl.totalExits);
+                    currentState.setRealtimeBaseEnters(CountServiceImpl.realtimeBaseEnters);
+                    currentState.setRealtimeBaseExits(CountServiceImpl.realtimebBaseExits);
+                    currentState.setErrorCount(CountServiceImpl.errorCount);
+                    currentState.setBaseTime(people.getTimestamp());
+                    dao.updateCountState(currentState);//鏇存柊褰撳墠缁熻鐘舵��
+                } else {
+                    System.out.println("noTotalEners");
+                }
+                // 濡傛灉瀛樺湪ReportData銆傚垯璇佹槑杩斿洖鐨勬槸缁熻鏁版嵁鐨剎ml
+                if(xmlStr.indexOf("ReportData")>-1){
+                    people = XmlUtil.xmlToPeople(xmlStr);
+                    dao.savePeople(people);
+                }
+            }
+
+        }catch (IOException ex){
+
+        }catch (Exception e){
+            System.out.println("exception:"+e.getMessage());
+        }
+        finally {
+            try {
+                if(iWriter !=null)
+                    iWriter.close();
+                if(iReader !=null)
+                    iReader.close();
+                if(client !=null)
+                    client.close();
+
+            }catch (IOException e){
+                System.out.println("finally exception:"+e.getMessage());
+            }
+        }
+    }
+
+    public static void sendAck(BufferedWriter iWriter) {
+        String statusLine = "HTTP/1.0 200 OK"+CRLF;
+        String contentTypeLine = "Content-type: text/html"+CRLF ;
+        String connectionClose = "Connection: Keep-Alive"+CRLF ;
+        String contentLengthLine = "Content-Length: 0"+CRLF +CRLF;;
+
+        try {
+            iWriter.write(statusLine);
+            iWriter.write(contentTypeLine);
+            iWriter.write(connectionClose);
+            iWriter.write(contentLengthLine);
+            iWriter.flush();
+        } catch (IOException ex) {
+            ex.printStackTrace();
+        }
+    }
+
+    // Send nack
+    public static void sendNack(BufferedWriter iWriter) {
+        String statusLine = "HTTP/1.1 400 Bad Request" + CRLF;
+        String contentTypeLine = "Content-type: text/html" + CRLF;
+        String connectionClose = "Connection: close" + CRLF;
+        String content = "<ErrorList><Error>error</Error></ErrorList>";
+        String contentLengthLine = "Content-Length: "
+                + (new Integer(content.length())).toString() + CRLF + CRLF;
+
+        try {
+            iWriter.write(statusLine);
+            iWriter.write(contentTypeLine);
+            iWriter.write(connectionClose);
+            iWriter.write(contentLengthLine);
+            iWriter.write(content);
+            iWriter.flush();
+        } catch (IOException ex) {
+            ex.printStackTrace();
+        }
+    }
 }

--
Gitblit v1.8.0