查找附近人实现方案(Redis)

可以利用Redis的Geospatial数据类型,结合用户经纬度信息进行存储和查询。代码如下:

package net.rakan.distributedservice.testserver.test.nearby.controller;

import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Parameters;
import io.swagger.v3.oas.annotations.enums.ParameterIn;
import io.swagger.v3.oas.annotations.tags.Tag;
import net.rakan.distributedservice.common.util.ListUtils;
import net.rakan.distributedservice.common.vo.Result;
import net.rakan.distributedservice.testserver.test.nearby.dto.SaveUserLocationDTO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.geo.*;
import org.springframework.data.redis.connection.RedisGeoCommands;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.util.Objects;
import java.util.stream.Collectors;

/**
 * 附近人
 *
 * @author LiChangRui on 2024/3/15 10:14
 */
@Tag(name = "附近人")
@RestController
@RequestMapping("/nearBy")
public class NearByController {

    private static final String USER_LOCATION_KEY = "user_location";

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 存储用户经纬度
     *
     * @author LiChangRui on 2024/9/23 20:33
     */
    @Operation(summary = "存储用户经纬度")
    @PostMapping("/saveUserLocation")
    public Result<?> saveUserLocation(@RequestBody @Validated SaveUserLocationDTO dto) {
        redisTemplate
                .opsForGeo()
                .add(USER_LOCATION_KEY,
                        new Point(dto.getLongitude(), dto.getLatitude()),
                        dto.getUserId()
                );
        return Result.ok();
    }


    /**
     * 查询附近的人
     *
     * @author LiChangRui on 2024/9/23 20:33
     */
    @Operation(summary = "查询附近的人")
    @Parameters({
            @Parameter(name = "longitude", description = "经度", required = true, in = ParameterIn.QUERY),
            @Parameter(name = "latitude", description = "纬度", required = true, in = ParameterIn.QUERY),
            @Parameter(name = "radius", description = "半径(KM)", required = true, in = ParameterIn.QUERY)
    })
    @GetMapping("/queryNearbyUserList")
    public Result<?> queryLikeNum(@RequestParam("longitude") Double longitude,
                                  @RequestParam("latitude") Double latitude,
                                  @RequestParam("radius") Double radius) {
        // 创建一个表示地理位置的圆
        Circle circle = new Circle(new Point(longitude, latitude),
                new Distance(radius, Metrics.KILOMETERS)
        );

        GeoResults<RedisGeoCommands.GeoLocation<Object>> nearbyLocations = redisTemplate
                .opsForGeo()
                .radius(USER_LOCATION_KEY, circle);

        if (Objects.isNull(nearbyLocations)) {
            return Result.ok(ListUtils.empty());
        }

        return Result.ok(nearbyLocations.getContent().stream()
                .map(GeoResult::getContent)
                .collect(Collectors.toList()));
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值