Springboot实战——黑马点评之附近商铺
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());
}