namespace Dpz.Core.Service.RepositoryServiceImpl;

public class WaitExecutionService(IRepository<WaitExecution> repository, IMapper mapper)
    : IWaitExecutionService
{
    public async Task<IList<VmWaitExecution>> GetNotExecuteTaskAsync()
    {
        var list = await repository
            .SearchFor(x => x.Status != ExecuteStatus.ExecuteComplete)
            .OrderBy(x => x.Id)
            .Take(1000)
            .ToListAsync();
        return mapper.Map<List<VmWaitExecution>>(list);
    }

    public async Task UpdateStatusAsync(string id, ExecuteStatus status)
    {
        if (ObjectId.TryParse(id, out var oid))
        {
            var update = Builders<WaitExecution>
                .Update.Set(x => x.Status, status)
                .Set(x => x.LastUpdateTime, DateTime.Now);
            await repository.UpdateAsync(x => x.Id == oid, update);
        }
    }

    public async Task ClearExpiredAsync()
    {
        var ids = await repository
            .SearchFor(x => x.Status == ExecuteStatus.ExecuteComplete)
            .OrderByDescending(x => x.Id)
            .ThenByDescending(x => x.LastUpdateTime)
            .Take(1000)
            .Select(x => x.Id)
            .ToListAsync();

        var filter = Builders<WaitExecution>.Filter.In(x => x.Id, ids);
        await repository.DeleteAsync(filter);
    }
}