首页 > 软件编程 > java > SpringBoot Redis查找附近用户

SpringBoot+Redis实现查找附近用户的示例代码

2024-02-03 09:55:34 作者：帅龍之龍

SpringDataRedis提供了十分简单的地理位置定位的功能,本文主要介绍了SpringBoot+Redis实现查找附近用户的示例代码,具有一定的参考价值,感兴趣的可以了解一下

前言

简单记录一下使用Redis的GEO命令，SpringDataRedis提供了十分简单的地理位置定位的功能，实现查找附近用户的功能。

一、Redis的GEO命令之GEOADD、GEORADIUS命令

1.GEOADD 命令

（1）用法：GEOADD key [longitude][ latitude ][member]
（2）作用：用于存储地理位置信息，以便进行地理位置搜索和距离计算等操作。
（3）返回值：成功添加的成员数量。
（4）示例：

redis > GEOADD cities 116.4074 39.9042 Beijing
redis > GEOADD cities NX 121.4737 31.2304 Shanghai

（5）可选参数：
- NX：只在 key 不存在时才执行操作。
- XX：只在 key 存在时才执行操作。
- CH：修改成功的成员数量将被返回

2.GEORADIUS 命令

（1）用法：GEORADIUS key longitude latitude radius
（2）作用：用于查询指定地理位置附近的其他地理位置的命令。
（3）返回值：返回成员列表。
（4）示例：

redis > GEORADIUS User-Location 116.4074 39.9042 100 km

二、示例代码

1.控制层

（1）UserController.java

/**
 * 更新用户位置信息
 * {
 *     "longitude": 113.936099,
 *     "latitude": 22.542364
 * }
 */
@PostMapping("updateUserLocation")
@ResponseBody
@CrossOrigin
public <T> T updateUserLocation(@RequestBody HashMap<String, Object> data) {
    return userService.updateUserLocation(data);
}

/**
 * 更新用户位置信息
 * {
 *     "longitude": 113.936099,
 *     "latitude": 22.542364,
 *     "radius": 10
 * }
 */
@PostMapping("nearby")
@ResponseBody
@CrossOrigin
public <T> T nearby(@RequestBody HashMap<String, Object> data) {
    return userService.nearby(data);
}

2.接口层

（1）IUserService.java

<T> T updateUserLocation(HashMap<String, Object> data);

<T> T nearby(HashMap<String, Object> data);

3.实现层

（1）UserServiceImpl.java

@Override
public <T> T updateUserLocation(HashMap<String, Object> data) {
    HashMap<String, Object> responseObj = new HashMap<>();
    // 获取登录用户
    UserDTO userDTO = RequestHolder.getUser();

    // 获取经纬度
    Double longitude = (Double) data.get("longitude"); // 经度
    Double latitude = (Double) data.get("latitude"); // 维度

    String USER_LOCATION_KEY = "User-Location";
    String phone = userDTO.getPhone();
    stringRedisTemplate.opsForGeo().add(USER_LOCATION_KEY, new Point(longitude, latitude), phone);
    responseObj.put("code", 200);
    responseObj.put("success", true);
    responseObj.put("msg", "更新完成");
    return (T) responseObj;
}

@Override
public <T> T nearby(HashMap<String, Object> data) {
    HashMap<String, Object> responseObj = new HashMap<>();
    // 获取登录用户
    UserDTO userDTO = RequestHolder.getUser();

    // 获取经纬度，以及半径
    Double longitude = (Double) data.get("longitude"); // 经度
    Double latitude = (Double) data.get("latitude"); // 维度
    Integer radius = (Integer) data.get("radius"); // 半径

    String USER_LOCATION_KEY = "User-Location";
    String phone = userDTO.getPhone();
    Distance distance = new Distance(radius, Metrics.KILOMETERS); // 距离，单位为千米
    Circle circle = new Circle(new Point(longitude, latitude), distance); // 圆心

    // 使用Redis的地理位置操作对象，在指定范围内查询附近的用户位置信息
    GeoResults<RedisGeoCommands.GeoLocation<String>> geoResults = stringRedisTemplate.opsForGeo().radius(USER_LOCATION_KEY, circle);
    List<Object> nearbyUsers = new ArrayList<>();
    for (GeoResult<RedisGeoCommands.GeoLocation<String>> geoResult : geoResults.getContent()) {
        Object memberId = geoResult.getContent().getName();
        // 排除查询用户本身
        if (!memberId.equals(phone)) {
            nearbyUsers.add(memberId);
        }
    }

    responseObj.put("code", 200);
    responseObj.put("success", true);
    responseObj.put("msg", "更新完成");
    responseObj.put("data", nearbyUsers);
    return (T) responseObj;
}

到此这篇关于SpringBoot+Redis实现查找附近用户的示例代码的文章就介绍到这了,更多相关SpringBoot Redis查找附近用户内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！