From 74f2dd24eb7241020319771c794d9e9f06264d7e Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 13 Jan 2026 18:49:06 +0800
Subject: [PATCH] 1
---
src/main/java/com/nq/ws/USWebsocketRunClient.java | 130 +++++++++++++++++++++++++++++++++++++++++++
1 files changed, 130 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/nq/ws/USWebsocketRunClient.java b/src/main/java/com/nq/ws/USWebsocketRunClient.java
new file mode 100644
index 0000000..8f15b19
--- /dev/null
+++ b/src/main/java/com/nq/ws/USWebsocketRunClient.java
@@ -0,0 +1,130 @@
+package com.nq.ws;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.gson.Gson;
+import com.google.gson.reflect.TypeToken;
+import com.nq.enums.EStockType;
+import com.nq.pojo.StockRealTimeBean;
+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 java.io.IOException;
+import java.lang.reflect.Type;
+import java.net.URI;
+import java.nio.charset.StandardCharsets;
+import java.util.*;
+
+@Slf4j
+public class USWebsocketRunClient extends WebSocketClient {
+
+ private EStockType eStockType;
+ public USWebsocketRunClient(URI serverUri,
+ EStockType eStockType
+ ) {
+ super(serverUri);
+ this.eStockType = eStockType;
+ }
+
+ private static HttpClient httpClient = HttpClients.createDefault(); // 单例化 HttpClient
+
+
+ private static HttpPost httpPost;
+ static {
+ httpPost = new HttpPost("http://127.0.0.1:8001/api/sendNotification"); // 初始化 HttpPost
+ }
+
+ @Override
+ public void onOpen(ServerHandshake serverHandshake) {
+ send(("key:"+ eStockType.getStockKey()+":"+eStockType.getContryId()).getBytes());
+ Timer heartbeatTimer;
+ // 启动心跳定时器
+ heartbeatTimer = new Timer();
+ heartbeatTimer.schedule(new TimerTask() {
+ @Override
+ public void run() {
+ if (isOpen()) {
+ //send(("key:"+ eStockType.getStockKey()+":"+eStockType.getContryId()).getBytes());
+ send("heartbeat");
+ }
+ }
+ }, 0, 3000); // 每3秒发送一次心跳消息
+ }
+
+ @Override
+ public void onMessage(String message) {
+ if (message.contains("身份验证成功") || message.contains("pong") || message.contains("身份验证失败") || message.contains("ws连接点只能有一个")) {
+ System.out.println("US" + message);
+ return;
+ }
+ System.out.println("US2" + message);
+ Map<String, String> stringObjectMap = jsonToMap(message);
+ StockRealTimeBean stockRealTimeBean = new StockRealTimeBean();
+ stockRealTimeBean.setPid(stringObjectMap.get("Id").toString());
+ stockRealTimeBean.setLast(stringObjectMap.get("Last").toString());
+ stockRealTimeBean.setBid(stringObjectMap.get("Bid").toString());
+ stockRealTimeBean.setAsk(stringObjectMap.get("Ask").toString());
+ stockRealTimeBean.setHigh(stringObjectMap.get("High").toString());
+ stockRealTimeBean.setLow(stringObjectMap.get("Low").toString());
+ stockRealTimeBean.setPc(stringObjectMap.get("Chg").toString());
+ stockRealTimeBean.setPcp(stringObjectMap.get("ChgPct").toString()+"%");
+ stockRealTimeBean.setTime(stringObjectMap.get("Time").toString());
+ RedisKeyUtil.setCacheRealTimeStock(EStockType.US,stockRealTimeBean);
+ ObjectMapper objectMapper = new ObjectMapper();
+ try {
+ if(!stockRealTimeBean.getPcp().contains("-")){
+ stockRealTimeBean.setPcp("+"+stringObjectMap.get("ChgPct").toString()+"%");
+ }
+ String json = objectMapper.writeValueAsString(stockRealTimeBean);
+ sendLoca(json);
+ StockRealTimeBean stockDetailBean = new Gson().fromJson(message, StockRealTimeBean.class);
+ RedisKeyUtil.setCacheRealTimeStock(EStockType.US,stockDetailBean);
+ } catch (JsonProcessingException e) {
+ log.error("websocket 美股股票 消息错误：{}", e.getMessage());
+ }
+ }
+
+ public static Map<String, String> jsonToMap(String json) {
+ Gson gson = new Gson();
+ Type type = new TypeToken<Map<String, String>>(){}.getType();
+ return gson.fromJson(json, type);
+ }
+
+ @Override
+ public void onClose(int i, String s, boolean b) {
+ log.info("websocket 美股股票 关闭 {} ", i);
+ }
+
+ @Override
+ public void onError(Exception e) {
+ log.info("websocket 美股股票 错误{}", e.getMessage());
+ }
+
+ public void sendLoca(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);
+ }
+ }
+}
--
Gitblit v1.9.3