forked from SciSharp/BotSharp
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCrontabController.cs
More file actions
96 lines (84 loc) · 3.06 KB
/
CrontabController.cs
File metadata and controls
96 lines (84 loc) · 3.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
using BotSharp.Abstraction.Crontab;
using BotSharp.Abstraction.Crontab.Models;
using BotSharp.Core.Crontab.Services;
using BotSharp.OpenAPI.Controllers.Crontab.Models;
namespace BotSharp.OpenAPI.Controllers;
[Authorize]
[ApiController]
public class CrontabController : ControllerBase
{
private readonly IServiceProvider _services;
private readonly IServiceScopeFactory _scopeFactory;
private readonly ILogger<CrontabController> _logger;
public CrontabController(
IServiceProvider services,
IServiceScopeFactory scopeFactory,
ILogger<CrontabController> logger)
{
_services = services;
_scopeFactory = scopeFactory;
_logger = logger;
}
[HttpPost("/crontab/{name}")]
public async Task<bool> RunCrontab(string name)
{
var cron = _services.GetRequiredService<ICrontabService>();
var crons = await cron.GetCrontable();
var found = crons.FirstOrDefault(x => x.Title.IsEqualTo(name));
if (found == null)
{
_logger.LogWarning($"Cannnot find crontab {name}");
return false;
}
try
{
_logger.LogWarning($"Start running crontab {name}");
await cron.ScheduledTimeArrived(found);
_logger.LogWarning($"Complete running crontab {name}");
return true;
}
catch (Exception ex)
{
_logger.LogError(ex, $"Error when running crontab {name}");
return false;
}
}
/// <summary>
/// As the Dkron job trigger API, run every 1 minutes
/// </summary>
/// <returns></returns>
[HttpPost("/crontab/scheduling-per-minute")]
public async Task<CrontabSchedulingResult> SchedulingCrontab()
{
var result = new CrontabSchedulingResult();
var allowedCrons = await GetCrontabItems();
foreach (var item in allowedCrons)
{
if (item.CheckNextOccurrenceEveryOneMinute())
{
_logger.LogInformation($"Crontab: {item.Title}, One occurrence was matched, attempting to execute...");
Task.Run(() => ExecuteTimeArrivedItem(item));
result.OccurrenceMatchedItems.Add(item.Title);
}
}
await Task.Delay(1000);
return result;
}
private async Task<List<CrontabItem>> GetCrontabItems(string? title = null)
{
var crontabService = _services.GetRequiredService<ICrontabService>();
var crons = await crontabService.GetCrontable();
var allowedCrons = crons.Where(cron => cron.TriggerType == CronTabItemTriggerType.OpenAPI).ToList();
if (title is null)
{
return allowedCrons;
}
return allowedCrons.Where(cron => cron.Title.IsEqualTo(title)).ToList();
}
private async Task ExecuteTimeArrivedItem(CrontabItem item)
{
using var scope = _scopeFactory.CreateScope();
var crontabService = scope.ServiceProvider.GetRequiredService<ICrontabService>();
await crontabService.ExecuteTimeArrivedItemWithReentryProtection(item);
}
}