Redis：GEO(Geolocation)数据结构使用

程序员无羡

已于 2023-03-27 07:31:33 修改

阅读量153

点赞数

分类专栏： Redis Java 文章标签： redis 数据库 java Geolocation geo

于 2023-03-27 07:27:41 首次发布

本文链接：https://blog.csdn.net/weixin_45427648/article/details/129782128

版权

Java 同时被 2 个专栏收录

89 篇文章 0 订阅

订阅专栏

Redis

22 篇文章 0 订阅

订阅专栏

文章介绍了如何在SpringBoot应用中排除默认的spring-data-redis和lettuce-core依赖，然后手动引入特定版本的库。接着展示了如何利用SpringDataRedis的opsForGeo方法添加和查询地理位置数据，包括按距离排序和分页的方法。

摘要由CSDN通过智能技术生成

依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
    <exclusions>
        <exclusion>
            <artifactId>spring-data-redis</artifactId>
            <groupId>org.springframework.data</groupId>
        </exclusion>
        <exclusion>
            <artifactId>lettuce-core</artifactId>
            <groupId>io.lettuce</groupId>
        </exclusion>
    </exclusions>
</dependency>
<dependency>
    <groupId>org.springframework.data</groupId>
    <artifactId>spring-data-redis</artifactId>
    <version>2.6.2</version>
</dependency>
<dependency>
    <groupId>io.lettuce</groupId>
    <artifactId>lettuce-core</artifactId>
    <version>6.1.6.RELEASE</version>
</dependency>

import org.springframework.data.geo.Point;
import org.springframework.data.redis.connection.RedisGeoCommands;
import org.springframework.data.redis.core.StringRedisTemplate;

// 向redis添加
stringRedisTemplate.opsForGeo().add(key, new Point(shop.getX(), shop.getY()), shopId);
// 或者
List<RedisGeoCommands.GeoLocation<String>> locations = new ArrayList<>(value.size());
locations.add(new RedisGeoCommands.GeoLocation<>(
        shop.getId().toString(),
        new Point(shop.getX(), shop.getY())
));
stringRedisTemplate.opsForGeo().add(key, locations);


// 查询redis、按照距离排序、分页。结果：shopId、distance
// 计算分页参数 current当前页
int from = (current - 1) * 5;
int end = current * 5;

GeoResults<RedisGeoCommands.GeoLocation<String>> results = stringRedisTemplate.opsForGeo() 
        .search(
               key,
               GeoReference.fromCoordinate(x, y),
               new Distance(5000),                                                
               RedisGeoCommands.GeoSearchCommandArgs.newGeoSearchArgs().includeDistance().limit(end)
        );

// 判断是否有数据
if (results == null) {
    return Result.ok(Collections.emptyList());
}

List<GeoResult<RedisGeoCommands.GeoLocation<String>>> list = results.getContent();
if (list.size() <= from) {
    // 没有下一页了，结束
    return Result.ok(Collections.emptyList());
}

// 截取 from ~ end的部分
List<Long> ids = new ArrayList<>(list.size());
Map<String, Distance> distanceMap = new HashMap<>(list.size());
list.stream().skip(from).forEach(result -> {
    // 获取shopId
    String shopIdStr = result.getContent().getName();
    ids.add(Long.valueOf(shopIdStr));
    // 获取距离
    Distance distance = result.getDistance();
    distanceMap.put(shopIdStr, distance);
});