SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

refine page search

Browse source
This commit is contained in:
Jicheng Lu 2024-02-04 14:20:20 -06:00
parent e7223f1267
commit 041d2df1bc
1 changed files with 22 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/Infrastructure/BotSharp.Core/Repository/FileRepository/FileRepository.AgentTask.cs
View file

@ -12,6 +12,10 @@ public partial class FileRepository
{ {
var tasks = new List<AgentTask>(); var tasks = new List<AgentTask>();
var pager = filter.Pager ?? new Pagination(); var pager = filter.Pager ?? new Pagination();
var skipCount = 0;
var takeCount = 0;
var totalCount = 0;
var matched = true;
var dir = Path.Combine(_dbSettings.FileRepository, _agentSettings.DataDir); var dir = Path.Combine(_dbSettings.FileRepository, _agentSettings.DataDir);
if (!Directory.Exists(dir)) return new PagedItems<AgentTask>(); if (!Directory.Exists(dir)) return new PagedItems<AgentTask>();
@ -22,11 +26,11 @@ public partial class FileRepository
if (!Directory.Exists(taskDir)) continue; if (!Directory.Exists(taskDir)) continue;
var agentId = agentDir.Split(Path.DirectorySeparatorChar).Last(); var agentId = agentDir.Split(Path.DirectorySeparatorChar).Last();
var matched = true;
matched = true;
if (filter?.AgentId != null) if (filter?.AgentId != null)
{ {
matched = matched && agentId == filter.AgentId; matched = agentId == filter.AgentId;
} }
if (!matched) continue; if (!matched) continue;
@ -37,14 +41,26 @@ public partial class FileRepository
var task = ParseAgentTask(taskFile); var task = ParseAgentTask(taskFile);
if (task == null) continue; if (task == null) continue;
matched = true;
if (filter?.Enabled != null) if (filter?.Enabled != null)
{ {
matched = matched && task.Enabled == filter.Enabled; matched = matched && task.Enabled == filter.Enabled;
} }
if (!matched) continue; if (!matched) continue;
totalCount++;
if (takeCount >= pager.Size) continue;
curTasks.Add(task); if (skipCount < pager.Offset)
{
skipCount++;
}
else
{
curTasks.Add(task);
takeCount++;
}
} }
if (curTasks.IsNullOrEmpty()) continue; if (curTasks.IsNullOrEmpty()) continue;
@ -60,8 +76,8 @@ public partial class FileRepository
return new PagedItems<AgentTask> return new PagedItems<AgentTask>
{ {
Items = tasks.Skip(pager.Offset).Take(pager.Size), Items = tasks,
Count = tasks.Count Count = totalCount
}; };
} }