可以利用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()));
}
}