From 0232c08966ec4a58c3e510269bafea9a0f789dc3 Mon Sep 17 00:00:00 2001 From: zvv <508748680@qq.com> Date: Mon, 6 Apr 2026 16:21:58 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9D=A5=E5=8D=95=E6=8F=90=E9=86=92=20+=20?= =?UTF-8?q?=E5=AE=A2=E6=88=B7=E5=82=AC=E5=8D=95=20=E4=B8=9A=E5=8A=A1?= =?UTF-8?q?=E4=BB=A3=E7=A0=81=E5=BC=80=E5=8F=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../sky/config/WebSocketConfiguration.java | 18 +++++ .../sky/controller/user/OrderController.java | 13 ++++ .../java/com/sky/service/OrderService.java | 7 ++ .../sky/service/impl/OrderServiceImpl.java | 39 ++++++++++ .../com/sky/websocket/WebSocketServer.java | 71 +++++++++++++++++++ 5 files changed, 148 insertions(+) create mode 100644 sky-server/src/main/java/com/sky/config/WebSocketConfiguration.java create mode 100644 sky-server/src/main/java/com/sky/websocket/WebSocketServer.java diff --git a/sky-server/src/main/java/com/sky/config/WebSocketConfiguration.java b/sky-server/src/main/java/com/sky/config/WebSocketConfiguration.java new file mode 100644 index 0000000..a46c2cb --- /dev/null +++ b/sky-server/src/main/java/com/sky/config/WebSocketConfiguration.java @@ -0,0 +1,18 @@ +package com.sky.config; + +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.web.socket.server.standard.ServerEndpointExporter; + +/** + * WebSocket配置类，用于注册WebSocket的Bean + */ +@Configuration +public class WebSocketConfiguration { + + @Bean + public ServerEndpointExporter serverEndpointExporter() { + return new ServerEndpointExporter(); + } + +} diff --git a/sky-server/src/main/java/com/sky/controller/user/OrderController.java b/sky-server/src/main/java/com/sky/controller/user/OrderController.java index 4c4f9e3..757a5f9 100644 --- a/sky-server/src/main/java/com/sky/controller/user/OrderController.java +++ b/sky-server/src/main/java/com/sky/controller/user/OrderController.java @@ -97,4 +97,17 @@ public class OrderController { orderService.repetition(id); return Result.success(); } + + + /** + * 客户催单 + * @param id + * @return + */ + @GetMapping("/reminder/{id}") + @ApiOperation("客户催单") + public Result reminder(@PathVariable Long id){ + orderService.reminder(id); + return Result.success(); + } } diff --git a/sky-server/src/main/java/com/sky/service/OrderService.java b/sky-server/src/main/java/com/sky/service/OrderService.java index a3b7fbb..5b8fe84 100644 --- a/sky-server/src/main/java/com/sky/service/OrderService.java +++ b/sky-server/src/main/java/com/sky/service/OrderService.java @@ -111,4 +111,11 @@ public interface OrderService { * @param id */ void complete(Long id); + + + /** + * 客户催单 + * @param id + */ + void reminder(Long id); } diff --git a/sky-server/src/main/java/com/sky/service/impl/OrderServiceImpl.java b/sky-server/src/main/java/com/sky/service/impl/OrderServiceImpl.java index 5c21914..cb0d8c8 100644 --- a/sky-server/src/main/java/com/sky/service/impl/OrderServiceImpl.java +++ b/sky-server/src/main/java/com/sky/service/impl/OrderServiceImpl.java @@ -1,5 +1,6 @@ package com.sky.service.impl; +import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.github.pagehelper.Page; @@ -22,6 +23,7 @@ import com.sky.vo.OrderPaymentVO; import com.sky.vo.OrderStatisticsVO; import com.sky.vo.OrderSubmitVO; import com.sky.vo.OrderVO; +import com.sky.websocket.WebSocketServer; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; import org.apache.xmlbeans.impl.xb.xmlconfig.Extensionconfig; @@ -67,6 +69,9 @@ public class OrderServiceImpl implements OrderService { @Value("${sky.baidu.ak}") private String ak; + @Autowired + private WebSocketServer webSocketServer; + /** * 用户下单 * @param ordersSubmitDTO @@ -191,6 +196,16 @@ public class OrderServiceImpl implements OrderService { .build(); orderMapper.update(orders); + + //通过websocket向客户端浏览器推送消息 type orderId content + Map map = new HashMap(); + map.put("type", 1); //1表示来单提醒 2表示客户催单 + map.put("orderId", ordersDB.getId()); + map.put("content", "订单号：" + outTradeNo); + String json = JSON.toJSONString(map); + + webSocketServer.sendToAllClient(json); + } @@ -531,6 +546,30 @@ public class OrderServiceImpl implements OrderService { } + /** + * 客户催单 + * @param id + */ + @Override + public void reminder(Long id) { + + //根据id查询订单 + Orders ordersDB = orderMapper.getById(id); + //校验订单是否存在 + if(ordersDB == null){ + throw new OrderBusinessException(MessageConstant.ORDER_STATUS_ERROR); + } + + Map map = new HashMap(); + map.put("type", 2); //1表示来单提醒 2表示客户催单 + map.put("orderId", id); + map.put("content", ordersDB.getNumber()); + + //通过websocket向客户端浏览器推送消息 + webSocketServer.sendToAllClient(JSON.toJSONString(map)); + } + + /** * 检查客户的收货地址是否超出配送范围 * @param address diff --git a/sky-server/src/main/java/com/sky/websocket/WebSocketServer.java b/sky-server/src/main/java/com/sky/websocket/WebSocketServer.java new file mode 100644 index 0000000..9a5fc83 --- /dev/null +++ b/sky-server/src/main/java/com/sky/websocket/WebSocketServer.java @@ -0,0 +1,71 @@ +package com.sky.websocket; + +import org.springframework.stereotype.Component; +import javax.websocket.OnClose; +import javax.websocket.OnMessage; +import javax.websocket.OnOpen; +import javax.websocket.Session; +import javax.websocket.server.PathParam; +import javax.websocket.server.ServerEndpoint; +import java.util.Collection; +import java.util.HashMap; +import java.util.Map; + +/** + * WebSocket服务 + */ +@Component +@ServerEndpoint("/ws/{sid}") +public class WebSocketServer { + + //存放会话对象 + private static Map sessionMap = new HashMap(); + + /** + * 连接建立成功调用的方法 + */ + @OnOpen + public void onOpen(Session session, @PathParam("sid") String sid) { + System.out.println("客户端：" + sid + "建立连接"); + sessionMap.put(sid, session); + } + + /** + * 收到客户端消息后调用的方法 + * + * @param message 客户端发送过来的消息 + */ + @OnMessage + public void onMessage(String message, @PathParam("sid") String sid) { + System.out.println("收到来自客户端：" + sid + "的信息:" + message); + } + + /** + * 连接关闭调用的方法 + * + * @param sid + */ + @OnClose + public void onClose(@PathParam("sid") String sid) { + System.out.println("连接断开:" + sid); + sessionMap.remove(sid); + } + + /** + * 群发 + * + * @param message + */ + public void sendToAllClient(String message) { + Collection sessions = sessionMap.values(); + for (Session session : sessions) { + try { + //服务器向客户端发送消息 + session.getBasicRemote().sendText(message); + } catch (Exception e) { + e.printStackTrace(); + } + } + } + +}