Files
tbd-station-14/Content.Server/AI/LoadBalancer/AiActionSystem.cs
2022-05-13 17:59:03 +10:00

28 lines
883 B
C#

using System.Threading;
using Content.Server.CPUJob.JobQueues.Queues;
namespace Content.Server.AI.LoadBalancer
{
/// <summary>
/// This will queue up an AI's request for an action and give it one when possible
/// </summary>
public sealed class AiActionSystem : EntitySystem
{
private readonly AiActionJobQueue _aiRequestQueue = new();
public AiActionRequestJob RequestAction(AiActionRequest request, CancellationTokenSource cancellationToken)
{
var job = new AiActionRequestJob(0.002, request, cancellationToken.Token);
// AI should already know if it shouldn't request again
_aiRequestQueue.EnqueueJob(job);
return job;
}
public override void Update(float frameTime)
{
base.Update(frameTime);
_aiRequestQueue.Process();
}
}
}