package com.terra.lfdcexp.config;

import com.terra.lfdcexp.entity.FenceEntityVo;
import com.terra.lfdcexp.util.HttpClientUtil;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.enums.ReadyState;
import org.java_websocket.handshake.ServerHandshake;

import java.net.URISyntaxException;


import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.java_websocket.WebSocket;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.handshake.ServerHandshake;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.stream.Collectors;


public class WebSocketC {

    private static Logger logger = LoggerFactory.getLogger(WebSocketC.class);

    public static WebSocketClient client;
    public static boolean isConnect = false;

    @Value("${http.websocket}")
    private String http_websocket;

    // 单例线程池

    ExecutorService executorService = Executors.newSingleThreadExecutor();

    @Bean
    public WebSocketClient getWebSocketClient() {
        try {
            logger.info("websocket:",http_websocket);
            client = new WebSocketClient(new URI(http_websocket), new Draft_6455()) {
                @Override
                public void onOpen(ServerHandshake serverHandshake) {
                    logger.info("握手成功");
                    isConnect = true;
                }

                @Override
                public void onMessage(String msg) {
                    logger.info("收到消息==========" + msg);

                    if (null != msg && !msg.trim().equals("")) {
                        // 异步线程加入线程池，进行发送操作
                        executorService.submit(() -> {
                            FenceEntityVo fenceEntityVo = new FenceEntityVo();
                            // code=21 的记录取出
                            fenceEntityVo.isAlert(msg);

                        });
                    }

                }

                @Override
                public void onClose(int i, String s, boolean b) {
                    logger.info("链接已关闭");
                }

                @Override
                public void onError(Exception e) {
                    e.printStackTrace();
                    logger.info("发生错误已关闭");
                }
            };

            Timer timer = new Timer();
            timer.schedule(new TimerTask() {
                @Override
                public void run() {
                    if (!isConnect) {
                        WebSocketClient client = getWebSocketClient();
                        logger.info("正在连接中...");
                        if (client.getReadyState().equals(ReadyState.NOT_YET_CONNECTED)) {
                            // 未连接状态
                            try {
                                client.connect();
                            } catch (Exception e) {
                                e.printStackTrace();
                            }
                        } else if (client.getReadyState().equals(ReadyState.CLOSING) || client.getReadyState().equals(ReadyState.CLOSED)) {
                            // 正关闭状态 或者 关闭状态
                            try {
                                client.reconnect();
                            } catch (IllegalArgumentException e) {
                                e.printStackTrace();
                            }
                        }

                    }
                }
            }, 1000, 5000);
            // client.connect();// 连接
            return client;

        } catch (URISyntaxException e) {
            e.printStackTrace();

        }



        return null ;
    }

}