Redis实现附件店肆

Geospatial
焦点操作

2. // 添加店铺数据预热操作，GEOADD key 经度 纬度 member
3. stringRedisTemplate.opsForGeo().add(key, new Point(shop.getX(), shop.getY()), shop.getId().toString());
4. ​
5. // 具体查询，GEOSEARCH key BYLONLAT x y BYRADIUS 10 WITHDISTANCE
6. GeoResults<RedisGeoCommands.GeoLocation<String>> results = stringRedisTemplate.opsForGeo()
7.                .search(
8.                        key,
9.                        GeoReference.fromCoordinate(x, y),// 圆心
10.                        new Distance(5000),//距离圆心的距离
11.                       RedisGeoCommands.GeoSearchCommandArgs.newGeoSearchArgs().includeDistance().limit(end) //
12.                );

复制代码

做数据预热：将商户信息添加到Redis里才华查到附件商户信息，每个商户包罗店肆id、经纬度 x , y。由于前端是按类型查询一群店肆信息的，所以要将同一类型的商户添加到同一个组里。

2. void loadShopData() {
3.    // 1.查询店铺信息
4.    List<Shop> list = shopService.list();
5.    // 2.把店铺分组，按照typeId分组，typeId一致的放到一个集合
6.    Map<Long, List<Shop>> map = list.stream().collect(Collectors.groupingBy(Shop::getTypeId));
7.    // 3.分批完成写入Redis
8.    for (Map.Entry<Long, List<Shop>> entry : map.entrySet()) {
9.        // 3.1.获取类型id
10.        Long typeId = entry.getKey();
11.        String key = SHOP_GEO_KEY + typeId;
12.        // 3.2.获取同类型的店铺的集合
13.        List<Shop> value = entry.getValue();
14.        List<RedisGeoCommands.GeoLocation<String>> locations = new ArrayList<>(value.size());
15.        // 3.3.写入redis GEOADD key 经度 纬度 member
16.        for (Shop shop : value) {
17.            // stringRedisTemplate.opsForGeo().add(key, new Point(shop.getX(), shop.getY()), shop.getId().toString());
18.            locations.add(new RedisGeoCommands.GeoLocation<>(
19.                    shop.getId().toString(),
20.                    new Point(shop.getX(), shop.getY())
21.            ));
22.        }
23.        stringRedisTemplate.opsForGeo().add(key, locations);
24.    }
25. }

复制代码

依靠

2. <dependency>
3.    <groupId>org.springframework.boot</groupId>
4.    <artifactId>
5.        spring-boot-starter-data-redis
6.    </artifactId>
7.    <exclusions>
8.        <exclusion>
9.            <artifactId>spring-data-redis</artifactId>
10.            <groupId>org.springframework.data</groupId>
11.        </exclusion>
12.        <exclusion>
13.            <artifactId>lettuce-core</artifactId>
14.            <groupId>io.lettuce</groupId>
15.        </exclusion>
16.    </exclusions>
17. </dependency>
18. <dependency>
19.    <groupId>org.springframework.data</groupId>
20.    <artifactId>spring-data-redis</artifactId>
21.    <version>2.6.2</version>
22. </dependency>
23. <dependency>
24.    <groupId>io.lettuce</groupId>
25.    <artifactId>lettuce-core</artifactId>
26.    <version>6.1.6.RELEASE</version>
27. </dependency>

复制代码

ShopController

2. @GetMapping("/of/type")
3. public Result queryShopByType(
4.        @RequestParam("typeId") Integer typeId,
5.        @RequestParam(value = "current", defaultValue = "1") Integer current,
6.        @RequestParam(value = "x", required = false) Double x,
7.        @RequestParam(value = "y", required = false) Double y
8. ) {
9.   return shopService.queryShopByType(typeId, current, x, y);
10. }

复制代码

ShopServiceImpl

2. @Override
3.    public Result queryShopByType(Integer typeId, Integer current, Double x, Double y) {
4.        // 1.判断是否需要根据坐标查询
5.        if (x == null || y == null) {
6.            // 不需要坐标查询，按数据库查询
7.            Page<Shop> page = query()
8.                    .eq("type_id", typeId)
9.                    .page(new Page<>(current, SystemConstants.DEFAULT_PAGE_SIZE));
10.            // 返回数据
11.            return Result.ok(page.getRecords());
12.        }
13. ​
14.        // 2.计算分页参数
15.        int from = (current - 1) * SystemConstants.DEFAULT_PAGE_SIZE;
16.        int end = current * SystemConstants.DEFAULT_PAGE_SIZE;
17. ​
18.        // 3.查询redis、按照距离排序、分页。结果：shopId、distance
19.        String key = SHOP_GEO_KEY + typeId;
20.        GeoResults<RedisGeoCommands.GeoLocation<String>> results = stringRedisTemplate.opsForGeo() // GEOSEARCH key BYLONLAT x y BYRADIUS 10 WITHDISTANCE
21.                .search(
22.                        key,
23.                        GeoReference.fromCoordinate(x, y),
24.                        new Distance(5000),
25.                        RedisGeoCommands.GeoSearchCommandArgs.newGeoSearchArgs().includeDistance().limit(end) //这里是从第一条到结束，所以要截取需要的数据返回
26.                );
27.        // 4.解析出id
28.        if (results == null) {
29.            return Result.ok(Collections.emptyList());
30.        }
31.        List<GeoResult<RedisGeoCommands.GeoLocation<String>>> list = results.getContent();
32.        if (list.size() <= from) {
33.            // 没有下一页了，结束
34.            return Result.ok(Collections.emptyList());
35.        }
36.        // 4.1.截取 from ~ end的部分
37.        List<Long> ids = new ArrayList<>(list.size());
38.        Map<String, Distance> distanceMap = new HashMap<>(list.size());
39.        
40.        list.stream().skip(from).forEach(result -> {
41.            // 4.2.获取店铺id
42.            String shopIdStr = result.getContent().getName();
43.            ids.add(Long.valueOf(shopIdStr));
44.            // 4.3.获取距离
45.            Distance distance = result.getDistance();
46.            distanceMap.put(shopIdStr, distance);
47.        });
48.        // 5.根据id查询Shop
49.        String idStr = StrUtil.join(",", ids);
50.        List<Shop> shops = query().in("id", ids).last("ORDER BY FIELD(id," + idStr + ")").list();
51.        for (Shop shop : shops) {
52.            shop.setDistance(distanceMap.get(shop.getId().toString()).getValue());
53.        }
54.        // 6.返回
55.        return Result.ok(shops);
56.    }

复制代码