小鹏汽车Robotaxi
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
xuhy
2026-04-22 746517aca9d0b7ae334c70b3041a9b1de795addb 
 xp-robotaxi-push-message-boot/src/main/java/com/xiaopeng/pushMessage/webSocket/handler/WebSocketTextFrameHandler.java


@@ -2,6 +2,8 @@




import com.alibaba.fastjson.JSON;


import com.alibaba.fastjson.JSONObject;


import com.xiaopeng.api.vo.automaticPilot.AppUserPosition;


import com.xiaopeng.common.redis.service.RedisService;


import com.xiaopeng.pushMessage.webSocket.service.WebSocketUserConnectionManager;


import io.netty.channel.ChannelHandler;


import io.netty.channel.ChannelHandlerContext;


@@ -9,12 +11,19 @@


import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;


import lombok.extern.slf4j.Slf4j;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.data.mongodb.core.MongoTemplate;


import org.springframework.data.mongodb.core.geo.GeoJsonPoint;


import org.springframework.data.mongodb.core.query.Criteria;


import org.springframework.data.mongodb.core.query.Query;


import org.springframework.data.mongodb.core.query.Update;


import org.springframework.stereotype.Component;


import org.springframework.util.StringUtils;




import javax.annotation.Resource;


import java.time.LocalDateTime;


import java.util.Objects;


import java.util.UUID;


import java.util.*;


import java.util.function.Function;


import java.util.stream.Collectors;




/**


 * TextWebSocketFrame 表示一个文本桢


@@ -32,6 +41,19 @@




    @Autowired


    private WebSocketAuthHandler authHandler;




    @Resource


    private MongoTemplate mongoTemplate;




    Map<String, BusinessMsgHandler> businessMsgHandlers;




    public WebSocketTextFrameHandler(List<BusinessMsgHandler> handlers) {


        this.businessMsgHandlers = handlers.stream()


                .collect(Collectors.toMap(


                        BusinessMsgHandler::getMessageType,


                        Function.identity()


                ));


    }




    @Override


    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {


@@ -143,6 +165,15 @@


        if (appUserId != null) {


            log.info("收到用户 {} 的业务消息: {}", appUserId, messageObj.toJSONString());




            businessMsgHandlers.get(messageObj.getString("type")).handle(appUserId, messageObj.getString("message"));


            //判断处理用户实时位置


            if ("location".equals(messageObj.getString("type"))) {


                log.info("收到用户 {} 的实时位置消息: {}", appUserId, messageObj.toJSONString());


                Double latitude = messageObj.getDouble("latitude");


                Double longitude = messageObj.getDouble("longitude");


                saveAppUserLocation(Long.parseLong(appUserId), latitude, longitude);


            }




            // 回复消息


            JSONObject response = new JSONObject();


            response.put("type", "message_response");


@@ -192,4 +223,21 @@


        log.error("WebSocket连接异常: {}", cause.getMessage(), cause);


        ctx.channel().close();


    }








    private void saveAppUserLocation(Long appUserId, Double latitude, Double longitude) {


        AppUserPosition userPosition = mongoTemplate.findOne(Query.query(Criteria.where("appUserId").is(appUserId)), AppUserPosition.class);


        GeoJsonPoint point = new GeoJsonPoint(longitude, latitude);


        if(null != userPosition){


            mongoTemplate.updateFirst(Query.query(Criteria.where("appUserId").is(appUserId)),


                    Update.update("point", point).set("addTime", System.currentTimeMillis()), AppUserPosition.class);


        }else{


            userPosition = new AppUserPosition();


            userPosition.setAppUserId(appUserId);


            userPosition.setPoint(point);


            userPosition.setAddTime(System.currentTimeMillis());


            mongoTemplate.save(userPosition);


        }


    }


}