package com.thhy.screen.modules.biz.bigscreen.controller;

import com.alibaba.nacos.api.annotation.NacosInjected;
import com.alibaba.nacos.api.exception.NacosException;
import com.alibaba.nacos.api.naming.NamingService;
import com.alibaba.nacos.api.naming.pojo.Instance;
import com.thhy.general.config.GlobalConfig;
import com.thhy.general.config.SysUserInfo;
import com.thhy.general.utils.SpringContextUtils;
import com.thhy.general.utils.UUIDUtils;
import com.thhy.general.utils.UserInfoUtils;
import com.thhy.screen.config.NetUtils;
import org.redisson.api.RBucket;
import org.redisson.api.RedissonClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArraySet;
import java.util.concurrent.TimeUnit;

@ServerEndpoint("/prolistenserver/{usertoken}")
@Component
public class ProListenSocketServer {

    private RedissonClient redissonClient;

    public RedissonClient getRedissonClient() {
        return SpringContextUtils.getBean(RedissonClient.class);
    }

    private GlobalConfig globalConfig;

    public GlobalConfig getGlobalConfig() {
        return SpringContextUtils.getBean(GlobalConfig.class);
    }

    private Logger logger = LoggerFactory.getLogger("SocketServer");

    /**
     * 线程安全的无序的集合
     */
    public static final CopyOnWriteArraySet<Session> SESSIONS = new CopyOnWriteArraySet<>();


    @OnOpen
    public void open(Session session, @PathParam("usertoken") String userToken){
        RedissonClient redissonClient = getRedissonClient();
        GlobalConfig globalConfig = getGlobalConfig();
        SysUserInfo userInfo = UserInfoUtils.getInstance().getUserInfo(userToken);
        logger.info(session.getId()+"连接成功");
        SESSIONS.add(session);
        String host = NetUtils.getIp();
        RBucket<Map<String,Object>> rBucketSess = redissonClient.getBucket(globalConfig.getKeyPrefix()+":ws:sessioninfo:"+host+"-"+globalConfig.getServerPort()+"-"+session.getId());
        Map<String,Object> map = new HashMap<>();
        map.put("host",host);
        rBucketSess.set(map,365,TimeUnit.DAYS);
        RBucket<SysUserInfo> rBucket = redissonClient.getBucket(globalConfig.getKeyPrefix()+":ws:token:"+host+"-"+globalConfig.getServerPort()+"-"+session.getId());
        rBucket.set(userInfo,365, TimeUnit.DAYS);
        RBucket<String> companyRBucket = redissonClient.getBucket(globalConfig.getKeyPrefix()+":ws:company:"+host+"-"+globalConfig.getServerPort()+"-"+session.getId());
        companyRBucket.set(userInfo.getCompanyId(),365,TimeUnit.DAYS);
        logger.info(session.getId()+"连接信息保存成功");
    }

    /**
     * 连接关闭
     *
     * @param session
     */
    @OnClose
    public void onClose(Session session) {
        logger.info(session.getId()+"连接关闭");
        RedissonClient redissonClient = getRedissonClient();
        GlobalConfig globalConfig = getGlobalConfig();
        SESSIONS.remove(session);
        String host = NetUtils.getIp();
        RBucket<Map<String,Object>> rBucketSess = redissonClient.getBucket(globalConfig.getKeyPrefix()+":ws:sessioninfo:"+host+"-"+globalConfig.getServerPort()+"-"+session.getId());
        rBucketSess.delete();
        RBucket<SysUserInfo> rBucket = redissonClient.getBucket(globalConfig.getKeyPrefix()+":ws:token:"+host+"-"+globalConfig.getServerPort()+"-"+session.getId());
        rBucket.delete();
        RBucket<String> companyRBucket = redissonClient.getBucket(globalConfig.getKeyPrefix()+":ws:company:"+host+"-"+globalConfig.getServerPort()+"-"+session.getId());
        companyRBucket.delete();
        logger.info(session.getId()+"连接信息清除成功");
    }

    /**
     * 接收到消息
     *
     * @param text
     */
    @OnMessage
    public String onMsg(String text) throws IOException {
        return "servet 发送：" + text;
    }


}