Redis GEO特性实现获取附近门店功能

目前项目有需要获取用户位置，然后推荐距离最近的5家门店的需求，目前无需做任何开发可以满足需求的方案是：把用户位置经纬度作为参数丢进数据库通过sql按照距离远近进行排序，大致sql如下：

select * from 柜台表 order by ROUND(6378.138*2*ASIN(SQRT(POW(SIN((latitude*PI()/180-#{用户位置纬度}*PI()/180)/2),2)+COS(latitude*PI()/180)*COS(#{用户位置纬度}*PI()/180)*POW(SIN((longitude*PI()/180-#{用户位置经度}*PI()/180)/2),2)))*1000) asc limit 0, 5

弊端也很明显：数据运算量大，设想品牌有一万家门店，同时有一万个人参与活动，这种并发情况下数据库扛不住会导致系统崩溃，这种方案行不通。

经人推荐说Redis GEO特性可支持位置相关操作，这个功能可以将用户给定的地理位置信息储存起来，并对这些信息进行操作。那就开始引入：

1 首先导入pom依赖：

<dependency>
    <groupId>org.springframework.data</groupId>
    <artifactId>spring-data-redis</artifactId>
    <version>1.8.6.RELEASE</version>
</dependency>

项目里之前引入的依赖是1.7.x版本，没有操作geo的api，网上资料说是在1.8版本之后才支持。

2 初始化柜台信息到redis

List<CounterDto> counterDtos = 获取所有门店;
GeoOperations opsForGeo = redisTemplate.opsForGeo();
Set<RedisGeoCommands.GeoLocation<Integer>> locations = new HashSet<>();
counterDtos.forEach(ci -> locations.add(new RedisGeoCommands.GeoLocation<Integer>(
        ci.getId(), new Point(ci.getLongitude(), ci.getLatitude())
)));
String key = "counterlocation";
opsForGeo.geoAdd(key, locations);

先要获取所有门店信息，其次拿到一个操作redis的 org.springframework.data.redis.core.RedisTemplate类，我采用的方案是把所有柜台的经纬度和id存入一个set集合，然后放到redis中，因为柜台有些信息是会有更新操作的，为了减小复杂度（在更新柜台信息的时候不更新redis数据）redis中存了柜台id，获取到最近的5家门店之后会再去请求数据库查询完整信息，到这柜台信息就已经成功存入redis中了

3 获取距离最近的门店

//默认以用户位置为中心获取附近1000公里的柜台，
static Distance initDistance() {
    Distance distance = new Distance(1000, Metrics.KILOMETERS);
    return distance;
}

//对返回的数据内容及排序规则进行设置
private RedisGeoCommands.GeoRadiusCommandArgs initArgs(int count) {
    count= 0==count?5:count;
    RedisGeoCommands.GeoRadiusCommandArgs args = RedisGeoCommands.GeoRadiusCommandArgs.newGeoRadiusArgs();
    //取前n条
    args.limit(count);
    //升序排序
    args.sortAscending();
    //返回的包括距离
    args.includeDistance();
    return args;
}

//下面代码片段是获取redis中距离最近5家门店
GeoOperations opsForGeo = redisTemplate.opsForGeo();
String key = ”counterlocation“;//跟初始化时key一致
Point point = new Point(param.getLongitude(), param.getLatitude());
Circle circle = new Circle(point, initDistance());
GeoResults geoResults = opsForGeo.geoRadius(key, circle, initArgs(count));
List<GeoResult> geoResultList = geoResults.getContent();
//下面拿到结果之后根据需求再做相应的数据组装即可
geoResultList.stream().forEach(c -> {
    RedisGeoCommands.GeoLocation<Integer> content = (RedisGeoCommands.GeoLocation<Integer>) c.getContent();
    int counterId = content.getName();//初始化存入redis的柜台id
    double distance=c.getDistance().getValue();//距离用户距离 
});

以上是完整流程，mark一下。

来源：CSDN

作者：无雨也无情

链接：https://blog.csdn.net/ItXiaoBird/article/details/104389762

标签

Redis

geo