优化Redis地理空间数据计算：告别客户端循环，拥抱服务器端效率

$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)));
    }
}

-- 参数：
-- 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

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

// Lua脚本内容，可以从文件加载或直接定义
$luaScript = <<<LUA
-- ... 上面的Lua脚本内容 ...
LUA;

// 预加载脚本到Redis，获取SHA1值
// 生产环境中，通常会在应用启动时加载一次脚本，然后复用SHA1值
$scriptSha = $redis->script('load', $luaScript);

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