Redis地理计算优化：告别客户端循环，提升服务器效率

服务器端效率 " />

在处理Redis中的地理空间数据时，常见的场景是先通过GEOSEARCH命令获取附近点位及其距离，然后针对每个点位，从其他数据结构（如HSET）中获取额外信息，并进行复杂的数学计算。原始方法中，这种计算通常在客户端通过循环逐一执行HGETALL并计算加权和，当地理点位数量庞大时，这种模式会产生严重的性能瓶颈。

理解性能瓶颈

原始代码片段展示了这种低效模式：

$geoPoints = $redis->executeRaw(["GEOSEARCH", $tableName, $type, $lon, $lat, "BYRADIUS", $radius, $metric, "WITHDIST"]);
$weightedSum = 0;
for ($i = 0; $i < count($geoPoints); $i++) {
    // $geoPoints[$i][0] 是成员名，例如 '2819483906'
    if ($redis->hgetall($geoPoints[$i][0]) != NULL) {
        $objArray = (object)$redis->hgetall($geoPoints[$i][0]);
        $cc = (float)$objArray->cc;
        $weightedSum += ($cc * ($radius - ((float)$geoPoints[$i][1] / $radius)));
    }
}

此方法的性能瓶颈主要体现在以下两点：

1. N+1查询问题： 对于GEOSEARCH返回的N个地理点位，客户端会发起N次独立的HGETALL请求。每次请求都需要网络往返（Round Trip Time, RTT），当N很大时，累积的RTT会显著增加处理时间。
2. 客户端计算开销： 所有的HGETALL数据获取和后续的数学计算都在客户端完成，增加了客户端的CPU和内存负担，并且无法利用Redis服务器的原子性操作优势。

为了解决这些问题，我们可以采用以下几种优化策略。

优化策略一：利用Redis Lua脚本实现服务器端计算

Redis支持通过Lua脚本在服务器端执行复杂逻辑。将数据获取和计算逻辑封装在一个Lua脚本中，可以实现原子性操作，并且将所有计算在Redis服务器内部完成，极大减少网络往返次数。这是解决此类问题的最有效方法之一。

优势：

• 减少网络延迟： 客户端只需发送一个脚本执行请求，Redis服务器处理所有逻辑并返回最终结果。
• 原子性： 脚本作为一个整体执行，不会被其他命令中断，保证数据一致性。
• 提升性能： 避免了客户端与服务器之间频繁的数据传输和协议解析开销。

Lua脚本示例：

-- 参数：
-- KEYS[1]: GEOSET的键名 (tableName)
-- ARGV[1]: 搜索类型 (FROMLONLAT 或 FROMMEMBER)
-- ARGV[2]: 经度 或 成员名
-- ARGV[3]: 纬度 (如果 ARGV[1] 是 FROMLONLAT)
-- ARGV[4]: 搜索半径
-- ARGV[5]: 距离单位
-- ARGV[6]: 搜索半径的原始值 (用于计算)

local tableName = KEYS[1]
local searchType = ARGV[1]
local lonOrMember = ARGV[2]
local lat = ARGV[3]
local radius = tonumber(ARGV[4])
local metric = ARGV[5]
local originalRadius = tonumber(ARGV[6]) -- 原始radius，用于权重计算

local geoPoints = {}
if searchType == "FROMLONLAT" then
    geoPoints = redis.call("GEOSEARCH", tableName, "FROMLONLAT", lonOrMember, lat, "BYRADIUS", radius, metric, "WITHDIST")
elseif searchType == "FROMMEMBER" then
    geoPoints = redis.call("GEOSEARCH", tableName, "FROMMEMBER", lonOrMember, "BYRADIUS", radius, metric, "WITHDIST")
else
    return redis.error_reply("Invalid search type")
end

local weightedSum = 0
local member_names = {}

-- 收集所有需要HGETALL的成员名
for i=1, #geoPoints do
    local memberName = geoPoints[i][1]
    table.insert(member_names, memberName)
end

-- 批量获取HSET数据：在Lua脚本内部循环调用HGETALL，虽然是循环，但操作在服务器内部，无网络开销
local hset_data = {}
for i=1, #member_names do
    local memberName = member_names[i]
    local data = redis.call("HGETALL", memberName)
    if #data > 0 then
        -- 将HGETALL返回的键值对数组转换为Lua table
        local obj = {}
        for j=1, #data, 2 do
            obj[data[j]] = data[j+1]
        end
        hset_data[memberName] = obj
    end
end

-- 执行加权和计算
for i=1, #geoPoints do
    local memberName = geoPoints[i][1]
    local distance = tonumber(geoPoints[i][2])

    local obj = hset_data[memberName]
    if obj and obj.cc then
        local cc = tonumber(obj.cc)
        weightedSum = weightedSum + (cc * (originalRadius - (distance / originalRadius)))
    end
end

return weightedSum

PHP客户端调用：

// 假设 $redis 已经连接到 Redis 服务器
$tableName = "myGeoSet"; // 你的GEOSET键名
$lon = -84.7691;
$lat = 39.9091;
$radius = 20; // 搜索半径
$metric = "km"; // 距离单位
$searchType = "FROMLONLAT"; // 或 "FROMMEMBER"

// Lua脚本内容，可以从文件加载或直接定义
$luaScript = <<script('load', $luaScript);

// 调用脚本
// KEYS数组：包含所有Redis键名，Lua脚本中通过KEYS[i]访问
// ARGV数组：包含所有参数值，Lua脚本中通过ARGV[i]访问
try {
    $result = $redis->evalsha(
        $scriptSha,
        [$tableName], // KEYS
        [
            $searchType,
            (string)$lon, //