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.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; @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; @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) { Map map = jsonToMap(args[0].toString()); if(map.get("pid").equals("00000001")){ System.out.println("接收时间:" + new SimpleDateFormat("HH:mm:ss").format(new Date()) + " " + args[0].toString()); } try { ApplicationContext act = ApplicationContextRegisterUtil.getApplicationContext(); MandatoryLiquidationService liquidationService = (MandatoryLiquidationService) act.getBean(IMandatoryLiquidationService.class); StockRealTimeBean stockDetailBean = new Gson().fromJson(args[0].toString(), StockRealTimeBean.class); liquidationService.RealTimeDataProcess(EStockType.IN,stockDetailBean); }catch (Exception e){ log.error("socket数据存入缓存错误:", e.getMessage()); } } }); 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); } } } }