快捷導(dǎo)航

redis GEO數(shù)據(jù)結(jié)構(gòu)、實(shí)現(xiàn)附近商鋪功能實(shí)踐

更新時(shí)間：2025年12月24日 08:49:21 作者：姓蔡小朋友

文章介紹了Redis中的GEO命令及其用途,包括地理坐標(biāo)存儲(chǔ)、距離計(jì)算、坐標(biāo)轉(zhuǎn)換和位置搜索等功能,還分享了如何使用Redis實(shí)現(xiàn)查詢附近商鋪的功能,包括導(dǎo)入商鋪信息和根據(jù)類型及距離進(jìn)行搜索

一、GEO

Grolocation意為地理坐標(biāo)，允許存儲(chǔ)地理坐標(biāo)信息，完成地理坐標(biāo)之間距離的計(jì)算。

geoadd key longitude1 latitude1 field1 longitude2 latitude2 field2...：向圖key中添加多個(gè)member，包含經(jīng)度longitude、緯度latitude 、member唯一標(biāo)識(shí)field
geodist key field1 field2 [m,km]：返回兩個(gè)member之間的距離
geohash key field1：將指定member的坐標(biāo)轉(zhuǎn)為hash字符串形式并返回
geopos key field1：返回指定member的坐標(biāo)
geoseartch key FROMLONLAT longitude1 latitude1 [BYRADIUS 10km] [ASC/DESC] [WITHDIST]：給定圓心和半徑，按照升序或降序返回指定范圍內(nèi)的member，并返回距離具體值WITHDIST
geosearchstore：同geoseratch，并將結(jié)果存儲(chǔ)到指定的key中

GEO底層是用sortedSet實(shí)現(xiàn)，其中經(jīng)緯度會(huì)被換算為score字段(score=a+b的意思)，member唯一標(biāo)識(shí)作為value字段。

二、redis實(shí)現(xiàn)查詢附近商鋪功能

1.按照商鋪類型導(dǎo)入商鋪信息

@Resource
StringRedisTemplate stringRedisTemplate;
@Resource
IShopService shopService;
@Test
public void testLoadShopData(){
    List<Shop> shops = shopService.query().list();
    for (Shop shop : shops) {
        stringRedisTemplate.opsForGeo().add(
                "geo:shop:"+shop.getTypeId(),
                new RedisGeoCommands.GeoLocation<>(
                        shop.getId().toString(),
                        new Point(shop.getX().doubleValue(),shop.getY().doubleValue())));
    }

2.根據(jù)商鋪類型和距離查詢附近商鋪

@Override
public Result queryShopByType(Integer typeId, Integer current, Double x, Double y) {
    if (x==null || y==null){
        // 根據(jù)類型分頁查詢
        Page<Shop> page = query()
                .eq("type_id", typeId)
                .page(new Page<>(current, 5));
        // 返回?cái)?shù)據(jù)
        return Result.ok(page.getRecords());
    }
    // 該頁起始和終止元素
    int from = (current-1)*5;
    int end = from + 5;
    // 查詢距離內(nèi)的店鋪
    GeoResults<RedisGeoCommands.GeoLocation<String>> radius = stringRedisTemplate.opsForGeo().radius(
            "geo:shop:" + typeId,
            new Circle(new Point(x, y), new Distance(5000)),
            RedisGeoCommands.GeoRadiusCommandArgs.newGeoRadiusArgs().includeDistance());//指定圓心半徑m查相關(guān)店鋪，并返回店鋪距離圓心的距離
    // 判空
    if (radius==null){
        return Result.ok(Collections.emptyList());
    }
    // 這里看類的結(jié)構(gòu)只能getContent
    List<GeoResult<RedisGeoCommands.GeoLocation<String>>> results = radius.getContent();

    // 判該頁是否有元素
    if (results.size()<=from){
        return Result.ok(Collections.emptyList());
    }

    // 截取from~end部分實(shí)現(xiàn)分頁
    List<GeoResult<RedisGeoCommands.GeoLocation<String>>> results1 = results.subList(from, Math.min(end, results.size()-1));

    ArrayList<Long> shopIds = new ArrayList<>();
    HashMap<Long, Distance> longDistanceHashMap = new HashMap<>();
    results1.forEach(r ->{
        String shopId = r.getContent().getName();
        shopIds.add(Long.valueOf(shopId));
        Distance distance = r.getDistance();
        longDistanceHashMap.put(Long.valueOf(shopId), distance);
    });

    List<Shop> shops = listByIds(shopIds);
    for (Shop s : shops) {
        double value = longDistanceHashMap.get(s.getId()).getValue();
        s.setDistance(value);
    }

    return Result.ok(shops);
}