优化代码

This commit is contained in:
ChenYi 2025-04-17 09:10:01 +08:00
parent cadb69f4b9
commit 4737c5b53d

View File

@ -7,6 +7,7 @@ using JiShe.CollectBus.Common.Models;
using JiShe.CollectBus.FreeRedisProvider; using JiShe.CollectBus.FreeRedisProvider;
using Microsoft.Extensions.Logging; using Microsoft.Extensions.Logging;
using System; using System;
using System.Collections.Concurrent;
using System.Collections.Generic; using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Threading; using System.Threading;
@ -385,79 +386,81 @@ namespace JiShe.CollectBus.RedisDataCache
bool descending = true) bool descending = true)
where T : DeviceCacheBasicModel where T : DeviceCacheBasicModel
{ {
// 参数校验 // 参数校验增强
if (string.IsNullOrWhiteSpace(redisHashCacheKey) || if (string.IsNullOrWhiteSpace(redisHashCacheKey) || string.IsNullOrWhiteSpace(redisZSetScoresIndexCacheKey))
string.IsNullOrWhiteSpace(redisZSetScoresIndexCacheKey))
{ {
_logger.LogError("参数异常: HashKey或ZSetKey为空"); _logger.LogError($"{nameof(GetAllPagedData)} 参数异常,-101");
return null; return null;
} }
if (pageSize < 1 || pageSize > 10000) if (pageSize < 1 || pageSize > 10000)
throw new ArgumentException("分页大小应在1-10000之间"); {
_logger.LogError($"{nameof(GetAllPagedData)} 分页大小应在1-10000之间-102");
return null;
}
var luaScript = @" var luaScript = @"
local command = ARGV[1] local command = ARGV[1]
local range_start = ARGV[2] local range_start = ARGV[2]
local range_end = ARGV[3] local range_end = ARGV[3]
local limit = tonumber(ARGV[4]) local limit = tonumber(ARGV[4])
local last_score = ARGV[5] local last_score = ARGV[5]
local last_member = ARGV[6] local last_member = ARGV[6]
-- --
local members local members
if command == 'ZRANGEBYSCORE' then if command == 'ZRANGEBYSCORE' then
members = redis.call(command, KEYS[1], range_start, range_end, 'WITHSCORES', 'LIMIT', 0, limit * 2) members = redis.call(command, KEYS[1], range_start, range_end, 'WITHSCORES', 'LIMIT', 0, limit * 2)
else else
members = redis.call('ZREVRANGEBYSCORE', KEYS[1], range_start, range_end, 'WITHSCORES', 'LIMIT', 0, limit * 2) members = redis.call('ZREVRANGEBYSCORE', KEYS[1], range_start, range_end, 'WITHSCORES', 'LIMIT', 0, limit * 2)
end end
-- --
local filtered_members = {} local filtered_members = {}
local count = 0 local count = 0
for i = 1, #members, 2 do for i = 1, #members, 2 do
local member = members[i] local member = members[i]
local score = members[i+1] local score = members[i+1]
local include = true local include = true
if last_score ~= '' and last_member ~= '' then if last_score ~= '' and last_member ~= '' then
if command == 'ZRANGEBYSCORE' then if command == 'ZRANGEBYSCORE' then
-- score > last_score (score == last_score member > last_member) -- score > last_score (score == last_score member > last_member)
if score == last_score then if score == last_score then
include = member > last_member include = member > last_member
else else
include = tonumber(score) > tonumber(last_score) include = tonumber(score) > tonumber(last_score)
end end
else else
-- score < last_score (score == last_score member < last_member) -- score < last_score (score == last_score member < last_member)
if score == last_score then if score == last_score then
include = member < last_member include = member < last_member
else else
include = tonumber(score) < tonumber(last_score) include = tonumber(score) < tonumber(last_score)
end end
end end
end end
if include then if include then
table.insert(filtered_members, member) table.insert(filtered_members, member)
table.insert(filtered_members, score) table.insert(filtered_members, score)
count = count + 1 count = count + 1
if count >= limit then if count >= limit then
break break
end end
end end
end end
-- --
local result_members, result_scores = {}, {} local result_members, result_scores = {}, {}
for i=1,#filtered_members,2 do for i=1,#filtered_members,2 do
table.insert(result_members, filtered_members[i]) table.insert(result_members, filtered_members[i])
table.insert(result_scores, filtered_members[i+1]) table.insert(result_scores, filtered_members[i+1])
end end
if #result_members == 0 then return {0,{},{},{}} end if #result_members == 0 then return {0,{},{},{}} end
-- Hash数据 -- Hash数据
local hash_data = redis.call('HMGET', KEYS[2], unpack(result_members)) local hash_data = redis.call('HMGET', KEYS[2], unpack(result_members))
return {#result_members, result_members, result_scores, hash_data}"; return {#result_members, result_members, result_scores, hash_data}";
// 调整范围构造逻辑(移除排他符号) // 调整范围构造逻辑(移除排他符号)
string rangeStart, rangeEnd; string rangeStart, rangeEnd;
@ -504,7 +507,7 @@ return {#result_members, result_members, result_scores, hash_data}";
var hasNext = validItems.Count > pageSize; var hasNext = validItems.Count > pageSize;
var actualItems = hasNext ? validItems.Take(pageSize) : validItems; var actualItems = hasNext ? validItems.Take(pageSize) : validItems;
// 修正分页锚点索引 //分页锚点索引
decimal? nextScore = null; decimal? nextScore = null;
string nextMember = null; string nextMember = null;
if (hasNext && actualItems.Any()) if (hasNext && actualItems.Any())