package com.nq.ws;

import cn.hutool.log.Log;
import com.google.gson.Gson;
import com.nq.pojo.StockDetailBean;
import com.nq.utils.StockUtil;
import com.nq.utils.redis.RedisShardedPoolUtils;
import lombok.extern.slf4j.Slf4j;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;

import java.net.URI;

@Slf4j
public class WebsocketRunClient  extends WebSocketClient {


    public WebsocketRunClient(URI serverUri) {
        super(serverUri);
    }

    @Override
    public void onOpen(ServerHandshake serverHandshake) {
        log.info("webSocket 客服端链接成功");
        send(("key:"+ StockUtil.KEY +":"+StockUtil.COUNTRY_ID).getBytes());
    }

    @Override
    public void onMessage(String s) {
        try {
            StockDetailBean stockDetailBean =  new Gson().fromJson(s, StockDetailBean.class);
            RedisShardedPoolUtils.set("instockDetail:"+stockDetailBean.getPid(),s);
        }catch (Exception e){
        }
    }

    @Override
    public void onClose(int i, String s, boolean b) {
        log.info("websocket 关闭"+1);
    }

    @Override
    public void onError(Exception e) {
        log.info("websocket 错误");
    }
}