优化Redis地理空间数据计算：告别客户端循环低效

// 假设 $redis 已经连接
$geoPoints = $redis->executeRaw(["GEOSEARCH", $tableName, $type, $lon, $lat, "BYRADIUS", $radius, $metric, "WITHDIST"]);

$weightedSum = 0;

// 客户端循环处理
for ($i = 0; $i < count($geoPoints); $i++) {
    $memberId = $geoPoints[$i][0];
    $distance = (float)$geoPoints[$i][1];

    // 为每个成员执行 HGETALL，导致大量网络往返
    $memberData = $redis->hgetall($memberId);

    if ($memberData != NULL) {
        $objArray = (object)$memberData;
        $cc = (float)$objArray->cc;
        // 客户端执行计算
        $weightedSum += ($cc * ($radius - ($distance / $radius)));
    }
}
// 最终得到 $weightedSum

-- KEYS: 不使用 KEYS，所有数据通过 ARGV 传递
-- ARGV: [tableName, type, lon, lat, radius, metric, searchRadius, searchMetric]
--       [1] tableName: GEOSET的键名
--       [2] type: BYLONLAT 或 BYMEMBER
--       [3] lon: 经度 (如果 type 是 BYLONLAT)
--       [4] lat: 纬度 (如果 type 是 BYLONLAT)
--       [5] searchRadius: 搜索半径
--       [6] metric: 距离单位 (m, km, ft, mi)
--       [7] computationRadius: 用于计算的原始半径 (即 PHP 代码中的 $radius)

local tableName = ARGV[1]
local searchType = ARGV[2]
local lon = ARGV[3]
local lat = ARGV[4]
local searchRadius = ARGV[5]
local metric = ARGV[6]
local computationRadius = tonumber(ARGV[7]) -- 将字符串转换为数字

local geoPoints

-- 根据 searchType 构建 GEOSEARCH 命令
if searchType == 'BYLONLAT' then
    geoPoints = redis.call('GEOSEARCH', tableName, searchType, lon, lat, 'BYRADIUS', searchRadius, metric, 'WITHDIST')
else
    -- 如果是 BYMEMBER，则 ARGV 结构需要调整，此处简化为 BYLONLAT
    -- 实际应用中需要更灵活的 ARGV 处理
    return redis.error_reply("Unsupported searchType: " .. searchType)
end


local weightedSum = 0.0

-- 遍历 GEOSEARCH 结果
for i = 1, #geoPoints do
    local memberId = geoPoints[i][1]
    local distance = tonumber(geoPoints[i][2]) -- 将距离字符串转换为数字

    -- 获取成员的 HSET 数据
    local memberData = redis.call('HGETALL', memberId)
    local cc = 0.0

    -- 解析 HGETALL 结果，查找 'cc' 字段
    if #memberData > 0 then
        for j = 1, #memberData, 2 do
            if memberData[j] == 'cc' then
                cc = tonumber(memberData[j+1])
                break
            end
        end
    end

    -- 执行加权求和计算
    if cc ~= 0 then -- 确保 cc 值有效
        weightedSum = weightedSum + (cc * (computationRadius - (distance / computationRadius)))
    end
end

return weightedSum

// 假设 $redis 已经连接
$tableName = 'myGeoSet'; // 替换为你的 GEOSET 键名
$lon = -84.769;
$lat = 39.909;
$searchRadius = 20; // GEOSEARCH 的半径
$metric = 'km'; // 距离单位
$computationRadius = 20.0; // 用于计算的原始半径，与 $searchRadius 可能相同或不同

// Lua 脚本内容
$luaScript = <<<LUA
-- ... 上述 Lua 脚本内容 ...
LUA;

// 执行 Lua 脚本
// 注意：KEYS 参数为空，所有参数通过 ARGV 传递
try {
    $result = $redis->eval($luaScript, [
        $tableName, 'BYLONLAT', $lon, $lat, $searchRadius, $metric, $computationRadius
    ], 0); // 0 表示没有 KEYS 参数

    echo "计算得到的加权和: " . $result . PHP_EOL;
} catch (RedisException $e) {
    echo "执行 Lua 脚本失败: " . $e->getMessage() . PHP_EOL;
}