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; import com.nq.service.impl.MandatoryLiquidationService; import com.nq.utils.ApplicationContextRegisterUtil; import com.nq.utils.PropertiesUtil; import com.nq.utils.redis.RedisKeyUtil; import com.nq.ws.client.IO; import com.nq.ws.client.Socket; import io.socket.emitter.Emitter; 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.json.JSONException; import org.json.JSONObject; import org.springframework.context.ApplicationContext; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import java.io.IOException; import java.lang.reflect.Type; import java.net.URI; import java.net.URISyntaxException; import java.nio.charset.StandardCharsets; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Map; /** * soket-io */ @Slf4j @Configuration public class WsClientConfig { private static final String SERVER_URL = PropertiesUtil.getProperty("JS_IN_HTTP_API"); private static final String AUTH_KEY = PropertiesUtil.getProperty("JS_IN_KEY"); private static final String ROOM_ID = "14"; private static Socket socket; private static HttpClient httpClient = HttpClients.createDefault(); // 单例化 HttpClient private static HttpPost httpPost; static { httpPost = new HttpPost("http://127.0.0.1:8001/api/sendNotification"); // 初始化 HttpPost } @Bean public void websocketRunClientMap() { connectToServer(); } private static void connectToServer() { IO.Options options = new IO.Options(); options.reconnection = true; options.reconnectionDelay = 1000; try { socket = IO.socket(new URI(SERVER_URL), options); } catch (URISyntaxException e) { log.error("Invalid URI", e); return; } socket.on(Socket.EVENT_CONNECT, new Emitter.Listener() { @Override public void call(Object... args) { authenticate(); } }); socket.on("marketData", new Emitter.Listener() { @Override public void call(Object... args) { send(args[0].toString()); StockRealTimeBean stockDetailBean = new Gson().fromJson(args[0].toString(), StockRealTimeBean.class); RedisKeyUtil.setCacheRealTimeStock(EStockType.IN,stockDetailBean); } }); socket.on(Socket.EVENT_CONNECT_ERROR, new Emitter.Listener() { @Override public void call(Object... args) { log.error("socketIo连接错误: " + args[0]); } }); socket.on(Socket.EVENT_DISCONNECT, new Emitter.Listener() { @Override public void call(Object... args) { log.error("socketIo 断开连接: " + args[0]); if (!socket.connected()) { log.error("socketIo 开始重连: " + args[0]); reconnect(); } } }); socket.connect(); } public static Map jsonToMap(String json) { Gson gson = new Gson(); Type type = new TypeToken>(){}.getType(); return gson.fromJson(json, type); } private static void reconnect() { new Thread(new Runnable() { @Override public void run() { while (!socket.connected()) { try { log.error("socketIo 开始重连"); socket.connect(); Thread.sleep(5000); // 重连间隔,单位毫秒 } catch (InterruptedException e) { log.error("重连被中断", e); } } } }).start(); } private static void authenticate() { log.info("socketIo---->开始连接"); if (socket != null && socket.connected()) { JSONObject authData = new JSONObject(); try { authData.put("key", AUTH_KEY); authData.put("roomId", ROOM_ID); socket.emit("authenticate", authData); log.info("socketIo---->连接成功"); } catch (JSONException e) { log.error("socketIo认证错误:"+e.getMessage(), e); } } } private static void send(String message) { try { // 准备 form-data 参数 List 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); } } }