package com.cloud.retrieve.websocket;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.cloud.common.utils.DateUtil;
import com.cloud.retrieve.service.EsDataService;
import jdk.nashorn.internal.runtime.logging.Loggable;
import jdk.nashorn.internal.runtime.logging.Logger;
import lombok.extern.slf4j.Slf4j;
import org.java_websocket.WebSocket;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.net.InetSocketAddress;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
//@Component
public class AlarmSocketServer extends WebSocketServer {

    @Autowired
    private EsDataService esDataService;

    private Map<WebSocket, Object> conns = new HashMap<>();

//    public AlarmSocketServer(){
//        super(new InetSocketAddress(7678));
//
//        new Thread(){
//            @Override
//            public void run(){
//
//                Object startTime =null;
//                Map<String,Object> map = new HashMap<>();
//
//                try {
//                    Thread.sleep(2000);
//                    AlarmSocketServer.this.start();
//                } catch (InterruptedException e) {
//                    e.printStackTrace();
//                }
//
//                while(true){
//                    try {
//                        Thread.sleep(2000);
//                        JSONObject alarmData = esDataService.findEsAllAlarmDataByPage(1, 200,null,null);
//
//                        for(WebSocket conn : conns.keySet()){
//                            try {
//                                String resourceDescriptor = conn.getResourceDescriptor();
//                                String[] params = resourceDescriptor.split("/");
//                                String method = params[1];
//
//                                try {
//                                    if(conns.get(conn) != null && conns.get(conn) instanceof Map){
//                                        startTime = ((Map<String, Object>)conns.get(conn)).get("startTime");
//                                        map = (Map<String, Object>)(((Map<String, Object>)conns.get(conn)).get("map"));
//                                        if(map == null){
//                                            map = new HashMap<>();
//                                        }
//                                    }
//                                 } catch (Exception e) {
//                                    e.printStackTrace();
//                                    startTime = null;
//                                    map = new HashMap<>();
//                                }
//
//                            } catch (Exception e) {
//                                e.printStackTrace();
//                                try {
//                                    if(conn.isClosed()){
//                                        conns.remove(conn);
//                                    }
//                                } catch (Exception e1) {
//                                    e1.printStackTrace();
//                                }
//                            }
//
//                            try {
//                                Map<String, Object> curMap = new HashMap<>();
//                                if((alarmData !=null && alarmData.get("total")!= null && Integer.parseInt(alarmData.getString("total")) > 0)){
//                                    JSONArray curAlarmList = alarmData.getJSONArray("equList");
//
//                                    List<Object> sendList = new ArrayList<>();
//                                    Timestamp lastLikeDate = null;
//                                    if(startTime !=null){//socket获取过数据
//                                        lastLikeDate = Timestamp.valueOf(startTime.toString());
//                                        for(int i=0;i < curAlarmList.size();i++){
//                                            JSONObject obj = (JSONObject) curAlarmList.get(i);
//                                            Timestamp sTime = Timestamp.valueOf(obj.getString("likeDate"));
//                                            String recordId = obj.get("Id").toString();
//                                                curMap.put(recordId, obj);
//                                            if(sTime.after(lastLikeDate) && !map.containsKey(recordId)){
//                                                sendList.add(obj);
//                                            }
//                                        }
//                                    }else{
//                                        for(int i=0;i < curAlarmList.size();i++){
//                                            JSONObject obj = (JSONObject) curAlarmList.get(i);
//                                            String recordId = obj.get("Id").toString();
//                                            curMap.put(recordId, obj);
//                                            sendList.add(obj);
//                                        }
//
//                                    }
//
//                                    if(sendList.size() > 0){
//                                        JSONObject sData = new JSONObject();
//                                        sData.put("total",alarmData.getString("total"));
//                                        sData.put("data", sendList);
//                                        log.info("websocket推送es数据："+sendList);
//                                        conn.send(JSON.toJSONString(sData, SerializerFeature.WriteMapNullValue).toString());
//
//                                        Map<String, Object> connMap = new HashMap<>();
//                                        lastLikeDate = Timestamp.valueOf(((JSONObject)sendList.get(sendList.size()-1)).getString("likeDate"));
//                                        log.info("websocket推送es数据 startTime："+lastLikeDate);
//                                        connMap.put("startTime", lastLikeDate);
//                                        connMap.put("map", curMap);
//                                        conns.put(conn, connMap);
//
//                                    }else{
//                                       /* JSONObject sData = new JSONObject();
//                                        sData.put("total",alarmData.getString("total"));
//                                        sData.put("data", sendList);
//                                        conn.send(JSON.toJSONString(sData, SerializerFeature.WriteMapNullValue).toString());
//                                        log.info("sendList.size=0");*/
//                                    }
//                                }
//                            } catch (Exception e) {
//                                e.printStackTrace();
//                            }
//                        }
//                    } catch (InterruptedException e) {
//                        e.printStackTrace();
//                    }
//                }
//            }
//        }.start();
//    }
    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        try {
            log.info("建立连接");
            conns.put(conn, new HashMap<String, Object>());
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        try {
            log.info("连接关闭");
            conns.remove(conn);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Override
    public void onMessage(WebSocket conn, String message) {
    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        log.info("exception:"+ex.getMessage());
    }

    @Override
    public void onStart() {
    }
}