千峰商城-springboot项目搭建-84-订单提交及支付-显示支付结果(websocket代码实现)...

1.引入依赖

        
        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-websocketartifactId>
            <version>2.3.12.RELEASEversion>
        dependency>

2.添加websocket服务节点配置(java配置方式)

 

WebSocketConfig ：

@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter getServerEndpointExporter(){
        return new ServerEndpointExporter();
    }
}

3.创建websocket服务器

WebSocketServer ：

@Component@ServerEndpoint("/webSocket/{oid}")public class WebSocketServer { private static ConcurrentHashMap sessionMap = new ConcurrentHashMap<>(); //前端发送请求建立webSocket连接，就会执行OnOpen方法 @OnOpen public void open(@PathParam("oid") String orderId, Session session){ sessionMap.put(orderId,session); } //前端关闭页面或者主动关闭webSocket连接都会执行close @OnClose public void close(@PathParam("oid") String orderId){ sessionMap.remove(orderId); } public static void sendMsg(String orderId,String msg){ try { Session session = sessionMap.get(orderId); session.getBasicRemote().sendText(msg); }catch (Exception e){ e.printStackTrace(); } }}

 PayController：

@RestController
@RequestMapping("/pay")
public class PayController {
 
    @Autowired
    private OrderService orderService;
 
    //回调接口:当用户支付成功之后，微信支付平台就会请求这个接口，将支付状态的数据传递过来
    //1.接收微信支付平台传递的数据(传递的数据格式为request的输入流)
    @RequestMapping("/callback")
    public String success(HttpServletRequest request) throws Exception {
        System.out.println("------------callback");
        ServletInputStream is = request.getInputStream();
        byte[] bs = new byte[1024];
        int len = -1;
        StringBuilder builder = new StringBuilder();
        while ( (len = is.read(bs)) != -1){
            builder.append(new  String(bs,0,len));
 
        }
        String s = builder.toString();
        //使用帮助类将xml接口的字符串转换成map
        Map map = WXPayUtil.xmlToMap(s);
 
        if (map != null && "success".equalsIgnoreCase( map.get("result_code")) ){
            //支付成功
            //2.修改订单状态为“待发货/已支付”
            String orderId = map.get("out_trade_no");
            int i = orderService.updateOrderStatus(orderId, "2");
 
            System.out.println("----orderId:"+orderId);
            //3.通过websocket连接向前端推送消息
//            Session session = WebSocketServer.sessionMap.get(orderId);
//            session.getBasicRemote().sendText("1");
            WebSocketServer.sendMsg(orderId,"1");
 
            //4.响应微信支付平台
            if (i>0){
                HashMap resp = new HashMap<>();
                resp.put("return_code","success");
                resp.put("return_msg","OK");
                resp.put("app_id",map.get("appid"));
                resp.put("result_code","success");
 
                String s1 = WXPayUtil.mapToXml(resp);
 
                return s1;
            }else {
                return null;
            }
 
        }else {
            //支付失败
            return null;
        }
 
    }
}