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方法 按照店铺类型划分为hashMapMap<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 写入Redisfor(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());
}
