1

zj

2024-08-29 2a13fc3d05c012766eb677cb0486e6dcf9ed1bf2

 src/main/java/com/nq/ws/WebsocketRunClient.java

@@ -1,6 +1,7 @@
package com.nq.ws;

import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import com.nq.enums.EStockType;
import com.nq.pojo.StockRealTimeBean;
import com.nq.service.IMandatoryLiquidationService;
@@ -8,45 +9,88 @@
import com.nq.utils.ApplicationContextRegisterUtil;
import com.nq.utils.redis.RedisKeyUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;
import org.springframework.context.ApplicationContext;

import java.io.IOException;
import java.lang.reflect.Type;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Slf4j
public class WebsocketRunClient  extends WebSocketClient {

    public WebsocketRunClient(URI serverUri) {
    private static HttpClient httpClient = HttpClients.createDefault(); // 单例化 HttpClient


    private static HttpPost httpPost;
    static {
        httpPost = new HttpPost("http://127.0.0.1:8002/api/sendNotification"); // 初始化 HttpPost
    }

    private EStockType eStockType;
    public WebsocketRunClient(URI serverUri,
    EStockType eStockType
    ) {
        super(serverUri);
        this.eStockType = eStockType;
    }

    @Override
    public void onOpen(ServerHandshake serverHandshake) {
        send(("key:"+ EStockType.IN.getStockKey()+":"+EStockType.IN.getContryId()).getBytes());
        send(("key:"+ eStockType.getStockKey()+":"+eStockType.getContryId()).getBytes());
    }

    @Override
    public void onMessage(String s) {
        try {


            ApplicationContext act = ApplicationContextRegisterUtil.getApplicationContext();
            MandatoryLiquidationService liquidationService = (MandatoryLiquidationService) act.getBean(IMandatoryLiquidationService.class);
            StockRealTimeBean stockDetailBean =  new Gson().fromJson(s, StockRealTimeBean.class);
            liquidationService.RealTimeDataProcess(EStockType.IN,stockDetailBean);
            RedisKeyUtil.setCacheRealTimeStock(eStockType,stockDetailBean);
            send(s);
        }catch (Exception e){

            log.error("websocket  新加坡 错误："+s);
            e.printStackTrace();
        }
    }


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

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

    public void send(String message) {
        try {
            // 准备 form-data 参数
            List<BasicNameValuePair> params = new ArrayList<>();
            params.add(new BasicNameValuePair("message", message));

            // 设置编码格式为 UTF-8
            UrlEncodedFormEntity entity = new UrlEncodedFormEntity(params, StandardCharsets.UTF_8);
            httpPost.setEntity(entity); // 设置 HttpPost 对象的参数

            // 发送请求
            HttpResponse response = httpClient.execute(httpPost);

            // 处理响应
            int statusCode = response.getStatusLine().getStatusCode();
        } catch (IOException e) {
            log.error("Http 请求错误", e);
        }
    }
}