在处理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)));
}
}此方法的性能瓶颈主要体现在以下两点：
- N+1查询问题： 对于GEOSEARCH返回的N个地理点位，客户端会发起N次独立的HGETALL请求。每次请求都需要网络往返（Round Trip Time, RTT），当N很大时，累积的RTT会显著增加处理时间。
- 客户端计算开销： 所有的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 weightedSumPHP客户端调用：
// 假设 $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, //
