提交代码

This commit is contained in:
ChenYi 2025-04-17 14:12:02 +08:00
parent 0584515df9
commit 871ed615a4
3 changed files with 61 additions and 119 deletions

View File

@ -22,6 +22,7 @@ using Volo.Abp.BackgroundWorkers;
using Volo.Abp.BackgroundWorkers.Hangfire; using Volo.Abp.BackgroundWorkers.Hangfire;
using Volo.Abp.EventBus; using Volo.Abp.EventBus;
using Volo.Abp.Modularity; using Volo.Abp.Modularity;
using Microsoft.Extensions.Options;
namespace JiShe.CollectBus; namespace JiShe.CollectBus;
@ -69,13 +70,14 @@ public class CollectBusApplicationModule : AbpModule
//初始化主题信息 //初始化主题信息
var kafkaAdminClient = context.ServiceProvider.GetRequiredService<IAdminClientService>(); var kafkaAdminClient = context.ServiceProvider.GetRequiredService<IAdminClientService>();
var configuration = context.ServiceProvider.GetRequiredService<IConfiguration>(); var configuration = context.ServiceProvider.GetRequiredService<IConfiguration>();
var kafkaOptions = context.ServiceProvider.GetRequiredService<IOptions<KafkaOptionConfig>>();
List<string> topics = ProtocolConstExtensions.GetAllTopicNamesByIssued(); List<string> topics = ProtocolConstExtensions.GetAllTopicNamesByIssued();
topics.AddRange(ProtocolConstExtensions.GetAllTopicNamesByReceived()); topics.AddRange(ProtocolConstExtensions.GetAllTopicNamesByReceived());
foreach (var item in topics) foreach (var item in topics)
{ {
await kafkaAdminClient.CreateTopicAsync(item, configuration.GetValue<int>(CommonConst.NumPartitions), configuration.GetValue<short>(CommonConst.KafkaReplicationFactor)); await kafkaAdminClient.CreateTopicAsync(item, kafkaOptions.Value.NumPartitions, kafkaOptions.Value.KafkaReplicationFactor);
} }
} }

View File

@ -396,7 +396,7 @@ namespace JiShe.CollectBus.RedisDataCache
local last_score = ARGV[5] local last_score = ARGV[5]
local last_member = ARGV[6] local last_member = ARGV[6]
-- 3 -- 5
local members local members
if command == 'ZRANGEBYSCORE' then if command == 'ZRANGEBYSCORE' then
members = redis.call('ZRANGEBYSCORE', KEYS[1], range_start, range_end, members = redis.call('ZRANGEBYSCORE', KEYS[1], range_start, range_end,
@ -541,66 +541,6 @@ namespace JiShe.CollectBus.RedisDataCache
} }
} }
/// <summary>
/// 并行分页导出方法(百万级数据支持)
/// </summary>
public async Task<List<T>> FullExportParallel<T>(
string hashKey,
string zsetKey,
int parallelDegree = 10,
int pageSize = 5000) where T : DeviceCacheBasicModel
{
var result = new ConcurrentBag<T>();
var totalCount = await GetTotalCount(zsetKey);
var totalPages = (int)Math.Ceiling(totalCount / (double)pageSize);
var semaphore = new SemaphoreSlim(parallelDegree);
var tasks = new List<Task>();
decimal? lastScore = null;
string lastMember = null;
var isDescending = true;
for (int page = 0; page < totalPages; page++)
{
await semaphore.WaitAsync();
tasks.Add(Task.Run(async () =>
{
try
{
var pageResult = await GetAllPagedData<T>(
hashKey,
zsetKey,
pageSize,
lastScore,
lastMember,
isDescending);
foreach (var item in pageResult.Items)
{
result.Add(item);
}
// 更新分页锚点
if (pageResult.HasNext)
{
lastScore = pageResult.NextScore;
lastMember = pageResult.NextMember;
}
}
finally
{
semaphore.Release();
}
}));
}
await Task.WhenAll(tasks);
return result.ToList();
}
/// <summary> /// <summary>
/// 通过ZSET索引获取数据支持10万级别数据处理控制在13秒以内。 /// 通过ZSET索引获取数据支持10万级别数据处理控制在13秒以内。
/// </summary> /// </summary>

View File

@ -22,7 +22,7 @@ namespace JiShe.CollectBus.Kafka
/// <summary> /// <summary>
/// kafka主题副本数量 /// kafka主题副本数量
/// </summary> /// </summary>
public int KafkaReplicationFactor { get; set; } public short KafkaReplicationFactor { get; set; }
/// <summary> /// <summary>
/// kafka主题分区数量 /// kafka主题分区数量