微服务 SpringBoot 整合 Redis GEO 实现附近商户功能

@Resourceprivate IShopService shopService;
@Resourceprivate StringRedisTemplate stringRedisTemplate;
@Testvoid loadShopData() {    //1. 查询店铺信息    List<Shop> shopList = shopService.list();    //2. 把店铺分组，按照typeId分组、typeId一致的放在一个集合    Map<Long, List<Shop>> map = shopList.stream().collect(Collectors.groupingBy(Shop::getTypeId));    //3. 分批完成写入redis    for (Map.Entry<Long, List<Shop>> entry : map.entrySet()) {        //3.1 获取类型id        Long typeId = entry.getKey();        String key = RedisConstants.SHOP_GEO_KEY + typeId;        //3.2 获取同类型的店铺的集合        List<Shop> value = entry.getValue();        List<RedisGeoCommands.GeoLocation<String>> locations = new ArrayList<>(value.size());        //3.3 写入redis GEOADD key 经度 维度 member        for (Shop shop : value) {            locations.add(new RedisGeoCommands.GeoLocation<>(                shop.getId().toString(),                new Point(shop.getX(), shop.getY())));        }        stringRedisTemplate.opsForGeo().add(key, locations);    }
}

@GetMapping("/of/type")public Result queryShopByType(    @RequestParam("typeId") Integer typeId,    @RequestParam(value = "current", defaultValue = "1") Integer current,    @RequestParam(value = "x", required = false) Double x,    @RequestParam(value = "y", required = false) Double y) {    return shopService.queryShopByType(typeId, current, x, y);}

@Overridepublic Result queryShopByType(Integer typeId, Integer current, Double x, Double y) {    //1. 判断是否需要坐标查询    if (x == null || y == null) {        // 不需要坐标查询，按数据库查询        Page<Shop> page = query()            .eq("type_id", typeId)            .page(new Page<>(current, SystemConstants.DEFAULT_PAGE_SIZE));        // 返回数据        return Result.ok(page.getRecords());    }    //2. 计算分页参数    int form = (current - 1) * SystemConstants.DEFAULT_PAGE_SIZE;    int end = current * SystemConstants.DEFAULT_PAGE_SIZE;    //3. 查询redis，按照距离排序、分页 结果：shopId、distance    String key = RedisConstants.SHOP_GEO_KEY + typeId;    GeoResults<RedisGeoCommands.GeoLocation<String>> results = stringRedisTemplate.opsForGeo().search(        key,        GeoReference.fromCoordinate(x, y),        new Distance(5000),        RedisGeoCommands.GeoSearchCommandArgs.newGeoSearchArgs().includeDistance().limit(end));    //4. 解析id    if (results == null) {        return Result.ok(Collections.emptyList());    }    List<GeoResult<RedisGeoCommands.GeoLocation<String>>> content = results.getContent();    //4.1 截取from => end    List<Long> ids = new ArrayList<>(content.size());    Map<String, Distance> distanceMap = new HashMap<>(content.size());    if (content.size() <= form) {        return Result.ok(Collections.emptyList());    }    content.stream().skip(form).forEach(result -> {        //4.2 获取店铺id        String shopIdStr = result.getContent().getName();        ids.add(Long.valueOf(shopIdStr));        //4.2 获取距离        Distance distance = result.getDistance();        distanceMap.put(shopIdStr, distance);    });    //5. 根据id查询shop    String idStr = StrUtil.join(",", ids);    List<Shop> shops = query().in("id", ids).last("ORDER BY FIELD(id," + idStr + ")").list();    // 循环将 商品距离放入对象距离属性中    shops.forEach(shop -> {        shop.setDistance(distanceMap.get(shop.getId().toString()).getValue());    });    //6. 返回结果    return Result.ok(shops);}