Springboot实战——黑马点评之附近商铺

1 认识GEO存储

1.1 GEO是什么

1.2 GEO怎么在Redis中存储

2 数据库店铺导入Redis

将数据库中的店铺数据按店铺类型type为关键字，分类存入Redis里
数据结构：
key(shop_type) -- sortedSet
sortedSet序列中元素组成为
value(shopId) -- score(X,Y)...若干个店铺坐标点
可以实现按照店铺类型，以距离为条件查询店铺信息。

      List<Shop> list = shopService.list();
      // 1. 采用stream流里的groupingBy方法 按照店铺类型划分为hashMap
      Map<Long, List<Shop>> map = list.stream()
              .collect(Collectors.groupingBy(Shop::getTypeId));
      // 2. 将hashMap中的记录按键值对导入到Redis中
      for(Map.Entry<Long, List<Shop>> entry : map.entrySet()){
          // 2.1 提取存入键值对的关键字
          Long typeId = entry.getKey();
          String key = "shop:geo:" + typeId;
          // 2.2 获取同类型的店铺的集合
          List<Shop> value = entry.getValue();
          List<RedisGeoCommands.GeoLocation<String>> locations = new ArrayList<>(value.size());
          // 2.3 写入Redis
          for(Shop shop : value){
              locations.add(new RedisGeoCommands.GeoLocation<>(
                      shop.getId().toString(),
                      new Point(shop.getX(),shop.getY())
              ));
          }
          stringRedisTemplate.opsForGeo().add(key, locations);
      }

3 附近店铺业务实现

3.1 按店铺类型查找附近店铺

基于上述导入到Redis中的店铺GEO数据，实现需求为：
前端请求提供店铺类型shopType用来筛选Redis-set，还需提供当前用户请求的中心位置信息point(x,y)，不要忘记分页查询传入的current以及MaxCount

• 使用Redis的research功能作店铺位置查询

// 2.2 end表示分页的截止下标
int end = current * SystemConstants.DEFAULT_PAGE_SIZE;

String geoSearchKey = SHOP_GEO_KEY + typeId;
// 3. 搜索返回的是 集合中的member点标识以及搜索距离
GeoResults<RedisGeoCommands.GeoLocation<String>> searchResult = stringRedisTemplate.opsForGeo().search(geoSearchKey,
                GeoReference.fromCoordinate(x, y),
                new Distance(5000),
                // 该方法的分页查询默认from都是从第一条记录开始
   RedisGeoCommands.GeoSearchCommandArgs.newGeoSearchArgs().includeDistance().limit(end));

        if(searchResult == null){
            return Result.ok(Collections.emptyList());
        }

• 将全局查询出的结果作手动截取
  并对截取结果作提取 提取出需要的id集合以及距离

// 4. 解析出搜索结果中的id
List<GeoResult<RedisGeoCommands.GeoLocation<String>>> list = searchResult.getContent();

if(list.size() <= from){
     return Result.ok(Collections.emptyList());
}
// 4.1 创建存放id序列的集合，待从数据库中查找
List<Long> ids = new ArrayList<>(list.size());
Map<String,Distance> distanceMap = new HashMap<>(list.size());
// 4.2 先对Redis中search出的结果作分页截取 从from=(current-1)*size开始截取
//     对search出的结果中提取出id 存入id集合中
list.stream().skip(from).forEach(result -> {

      String shopIdStr = result.getContent().getName();
      ids.add(Long.valueOf(shopIdStr));

      Distance distance = result.getDistance();
      distanceMap.put(shopIdStr,distance);
});

• 根据id集合去数据库中查询店铺信息并将距离赋值

// 5. 根据id集合查询数据库
String idStr = StrUtil.join(",", ids);
List<Shop> shops = query().in("id", ids).last("ORDER BY FIELD(id," + idStr + ")").list();

// 6. 对shops中的每个shop设置与id对应的距离变量
for(Shop shop:shops){
     Distance distance = distanceMap.get(shop.getId().toString());
     shop.setDistance(distance.getValue());
}