Spring Boot实战分页查询附近的人: Redis+GeoHash+Lua

前言

最近在做社交的业务，用户进入首页后需要查询附近的人；

项目状况：前期尝试业务阶段；

特点：

快速实现（不需要做太重，满足初期推广运营即可）

快速投入市场去运营

收集用户的经纬度：

用户在每次启动时将当前的地理位置（经度，维度）上报给后台

提到附近的人，脑海中首先浮现特点：

需要记录每位用户的经纬度

查询当前用户附近的人，搜索在N公里内用户

架构设计

时序图
技术实现方案

SpringBoot

Redis(version>=3.2)

Redis原生命令实现

存入用户的经纬度
- geoadd 用于存储指定的地理空间位置，可以将一个或多个经度(longitude)、纬度(latitude)、位置名称(member)添加到指定的 key 中
- 命令格式:
```
GEOADD key longitude latitude member [longitude latitude member ...]
```
- 模拟五个用户存入经纬度，redis客户端执行如下命令：
```
GEOADD zhgeo 116.48105 39.996794 zhangsan
GEOADD zhgeo 116.514203 39.905409 lisi
GEOADD zhgeo 116.489033 40.007669 wangwu
GEOADD zhgeo 116.562108 39.787602 sunliu
GEOADD zhgeo 116.334255 40.027400 zhaoqi
```
- 通过redis客户端查看效果：
查找距当前用户由近到远附近100km用户
- georadiusbymember可以找出位于指定范围内的元素，georadiusbymember 的中心点是由给定的位置元素决定的
- 命令格式:
```
GEORADIUSBYMEMBER key member radius m|km|ft|mi [WITHCOORD] [WITHDIST] [WITHHASH] [COUNT count] [ASC|DESC] [STORE key] [STOREDIST key]
```
- 模拟查找100km里距离sunliu由近到远五个人
```
georadiusbymember zhgeo sunliu 100 km asc count 5
```
- 命令执行效果如下：
如何实现分页查询那？
- 每次分页查询的请求都计算一次然后拿到程序中在取相应的分页数据，优缺点：
  - 优点：实现简单，无需额外的存储空间
  - 缺点：当用户量大时，很显然不仅效率低，而且容易把程序的内存搞溢出
- 经过查找发现redis的github官网给出了分页Issues(参考:Will the GEORADIUS support pagination?)，解决方案如下：
  - 利用GEORADIUSBYMEMBER 命令中的 STOREDIST 将排好序的数据存入一个Zset集合中，以后分页查直接从Zset
  - 命令如下：
```
georadiusbymember zhgeo sunliu 100 km asc count 5 storedist sunliu
```
  - 有序集合效果如下:
  - 以后分页查询命令:
```
//首先删除本身元素
zrem sunliu sunliu
//分页查找元素（在此以:查找第1页，每页数量是3为例）
zrange sunliu 0 2 withscores
```
  - 效果如下：

代码实现

完整代码（GitHub，欢迎大家Star,Fork,Watch)

https://github.com/dangnianchuntian/springboot

主要代码展示

Controller

/*
 * Copyright (c) 2020. [email protected] All Rights Reserved.
 * 项目名称：Spring Boot实战分页查询附近的人: Redis+GeoHash+Lua
 * 类名称：GeoController.java
 * 创建人：张晗
 * 联系方式：[email protected]
 * 开源地址: https://github.com/dangnianchuntian/springboot
 * 博客地址: https://zhanghan.blog.csdn.net
 */

package com.zhanghan.zhnearbypeople.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;

import com.zhanghan.zhnearbypeople.controller.request.ListNearByPeopleRequest;
import com.zhanghan.zhnearbypeople.controller.request.PostGeoRequest;
import com.zhanghan.zhnearbypeople.service.GeoService;

@RestController
public class GeoController {

    @Autowired
    private GeoService geoService;

    /**
     * 记录用户地理位置
     */
    @RequestMapping(value = "/post/geo", method = RequestMethod.POST)
    public Object postGeo(@RequestBody @Validated PostGeoRequest postGeoRequest) {
        return geoService.postGeo(postGeoRequest);
    }

    /**
     * 分页查询当前用户附近的人
     */
    @RequestMapping(value = "/list/nearby/people", method = RequestMethod.POST)
    public Object listNearByPeople(@RequestBody @Validated ListNearByPeopleRequest listNearByPeopleRequest) {
        return geoService.listNearByPeople(listNearByPeopleRequest);
    }

}

service

/*
 * Copyright (c) 2020. [email protected] All Rights Reserved.
 * 项目名称：Spring Boot实战分页查询附近的人: Redis+GeoHash+Lua
 * 类名称：GeoServiceImpl.java
 * 创建人：张晗
 * 联系方式：[email protected]
 * 开源地址: https://github.com/dangnianchuntian/springboot
 * 博客地址: https://zhanghan.blog.csdn.net
 */

package com.zhanghan.zhnearbypeople.service.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.geo.Point;
import org.springframework.data.redis.connection.RedisGeoCommands;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Service;

import com.zhanghan.zhnearbypeople.controller.request.ListNearByPeopleRequest;
import com.zhanghan.zhnearbypeople.controller.request.PostGeoRequest;
import com.zhanghan.zhnearbypeople.dto.NearByPeopleDto;
import com.zhanghan.zhnearbypeople.service.GeoService;
import com.zhanghan.zhnearbypeople.util.RedisLuaUtil;
import com.zhanghan.zhnearbypeople.util.wrapper.WrapMapper;

@Service
public class GeoServiceImpl implements GeoService {

    private static Logger logger = LoggerFactory.getLogger(GeoServiceImpl.class);

    @Autowired
    private RedisTemplate<String, Object> objRedisTemplate;

    @Value("${zh.geo.redis.key:zhgeo}")
    private String zhGeoRedisKey;

    @Value("${zh.geo.zset.redis.key:zhgeozset:}")
    private String zhGeoZsetRedisKey;

    /**
     * 记录用户访问记录
     */
    @Override
    public Object postGeo(PostGeoRequest postGeoRequest) {

        //对应redis原生命令:GEOADD zhgeo 116.48105 39.996794 zhangsan
        Long flag = objRedisTemplate.opsForGeo().add(zhGeoRedisKey, new RedisGeoCommands.GeoLocation<>(postGeoRequest
                .getCustomerId(), new Point(postGeoRequest.getLatitude(), postGeoRequest.getLongitude())));

        if (null != flag && flag > 0) {
            return WrapMapper.ok();
        }

        return WrapMapper.error();
    }

    /**
     * 分页查询附近的人
     */
    @Override
    public Object listNearByPeople(ListNearByPeopleRequest listNearByPeopleRequest) {

        String customerId = listNearByPeopleRequest.getCustomerId();

        String strZsetUserKey = zhGeoZsetRedisKey + customerId;

        List<NearByPeopleDto> nearByPeopleDtoList = new ArrayList<>();

        //如果是从第1页开始查,则将附近的人写入zset集合，以后页直接从zset中查
        if (1 == listNearByPeopleRequest.getPageIndex()) {
            List<String> scriptParams = new ArrayList<>();
            scriptParams.add(zhGeoRedisKey);
            scriptParams.add(customerId);
            scriptParams.add("100");
            scriptParams.add(RedisGeoCommands.DistanceUnit.KILOMETERS.getAbbreviation());
            scriptParams.add("asc");
            scriptParams.add("storedist");
            scriptParams.add(strZsetUserKey);

            //用Lua脚本实现georadiusbymember中的storedist参数
            //对应Redis原生命令:georadiusbymember zhgeo sunliu 100 km asc count 5 storedist sunliu
            Long executeResult = objRedisTemplate.execute(RedisLuaUtil.GEO_RADIUS_STOREDIST_SCRIPT(), scriptParams);

            if (null == executeResult || executeResult < 1) {
                return WrapMapper.ok(nearByPeopleDtoList);
            }

            //zset集合中去除自己
            //对应Redis原生命令:zrem sunliu sunliu
            Long remove = objRedisTemplate.opsForZSet().remove(strZsetUserKey, customerId);

        }

        nearByPeopleDtoList = listNearByPeopleFromZset(strZsetUserKey, listNearByPeopleRequest.getPageIndex(),
                listNearByPeopleRequest.getPageSize());

        return WrapMapper.ok(nearByPeopleDtoList);

    }

    /**
     * 分页从zset中查询指定用户附近的人
     */
    private List<NearByPeopleDto> listNearByPeopleFromZset(String strZsetUserKey, Integer pageIndex, Integer pageSize) {

        Integer startPage = (pageIndex - 1) * pageSize;
        Integer endPage = pageIndex * pageSize - 1;
        List<NearByPeopleDto> nearByPeopleDtoList = new ArrayList<>();
        //对应Redis原生命令:zrange key 0 2 withscores
        Set<ZSetOperations.TypedTuple<Object>> zsetUsers = objRedisTemplate.opsForZSet()
                .rangeWithScores(strZsetUserKey, startPage,
                        endPage);

        for (ZSetOperations.TypedTuple<Object> zsetUser : zsetUsers) {
            NearByPeopleDto nearByPeopleDto = new NearByPeopleDto();
            nearByPeopleDto.setCustomerId(zsetUser.getValue().toString());
            nearByPeopleDto.setDistance(zsetUser.getScore());
            nearByPeopleDtoList.add(nearByPeopleDto);
        }

        return nearByPeopleDtoList;
    }
}

RedisLuaUtil

/*
 * Copyright (c) 2020. [email protected] All Rights Reserved.
 * 项目名称：Spring Boot实战分页查询附近的人: Redis+GeoHash+Lua
 * 类名称：RedisLuaUtil.java
 * 创建人：张晗
 * 联系方式：[email protected]
 * 开源地址: https://github.com/dangnianchuntian/springboot
 * 博客地址: https://zhanghan.blog.csdn.net
 */

package com.zhanghan.zhnearbypeople.util;


import org.springframework.data.redis.core.script.DigestUtils;
import org.springframework.data.redis.core.script.RedisScript;

public class RedisLuaUtil {

    private static final RedisScript<Long> GEO_RADIUS_STOREDIST_SCRIPT;

    public static RedisScript<Long> GEO_RADIUS_STOREDIST_SCRIPT() {
        return GEO_RADIUS_STOREDIST_SCRIPT;
    }

    static {
        StringBuilder sb = new StringBuilder();
        sb.append("return redis.call('georadiusbymember',KEYS[1],KEYS[2],KEYS[3],KEYS[4],KEYS[5],KEYS[6],KEYS[7])");
        GEO_RADIUS_STOREDIST_SCRIPT = new RedisScriptImpl<>(sb.toString(), Long.class);
    }

    private static class RedisScriptImpl<T> implements RedisScript<T> {
        private final String script;
        private final String sha1;
        private final Class<T> resultType;

        public RedisScriptImpl(String script, Class<T> resultType) {
            this.script = script;
            this.sha1 = DigestUtils.sha1DigestAsHex(script);
            this.resultType = resultType;
        }

        @Override
        public String getSha1() {
            return sha1;
        }

        @Override
        public Class<T> getResultType() {
            return resultType;
        }

        @Override
        public String getScriptAsString() {
            return script;
        }
    }
}

测试

模拟用户上传地理位置进行存储
- 进行请求
- 查看效果
模拟用户sunliu查找附近100km的人，按3条一分页进行查询
- 进行请求

总结

亮点：
- 分页实现思路：将geo集合中的数据按距离由近到远筛选好后，通过storedist放入一个新的Zset集合
- redisTemplate没有针对原生命令georadiusbymember的storedist参数实现，灵活运用Lua脚本去实现
geo集合在亿级别以内的数据量没有问题，当超过亿后需要根据产品需要对Redis中的大值进行拆分，比如按照地域进行拆分等
有了地理位置，自己正在研究如何通过经纬度绘制出自己的运动路线，验证出来后与大家共享

Spring Boot实战分页查询附近的人: Redis+GeoHash+Lua

前言

架构设计

Redis原生命令实现

代码实现

测试

总结

猜你喜欢