public void ChangePeriod_TimerStartedPeriodChangedAfterFirstCall_PeriodChangedBeforeNextCall()
{
var entity = new Entity {
Id = "entity0"
};
Entity lastInvokedEntity = null;
var timerFactory = new TestTimerFactory();
using var scheduler = new EntityActionScheduler <Entity>(e =>
{
lastInvokedEntity = e;
return(Task.CompletedTask);
}, autoStart: true, runOnce: false, timerFactory);
scheduler.ScheduleEntity(entity, TimeSpan.FromMilliseconds(Period0));
timerFactory.VerifyTimer(0, Period0);
timerFactory.FireTimer(0);
var newPeriod = TimeSpan.FromMilliseconds(Period1);
scheduler.ChangePeriod(entity, newPeriod);
timerFactory.VerifyTimer(0, Period1);
Assert.Same(entity, lastInvokedEntity);
}
private void VerifyEntities(EntityActionScheduler <Entity> scheduler, params Entity[] entities)
{
var actualCount = 0;
foreach (var entity in entities)
{
Assert.True(scheduler.IsScheduled(entity));
actualCount++;
}
Assert.Equal(entities.Length, actualCount);
}
public ActiveHealthCheckMonitor(
IOptions <ActiveHealthCheckMonitorOptions> monitorOptions,
IEnumerable <IActiveHealthCheckPolicy> policies,
IProbingRequestFactory probingRequestFactory,
ITimerFactory timerFactory,
ILogger <ActiveHealthCheckMonitor> logger)
{
_monitorOptions = monitorOptions?.Value ?? throw new ArgumentNullException(nameof(monitorOptions));
_policies = policies?.ToDictionaryByUniqueId(p => p.Name) ?? throw new ArgumentNullException(nameof(policies));
_probingRequestFactory = probingRequestFactory ?? throw new ArgumentNullException(nameof(probingRequestFactory));
_logger = logger ?? throw new ArgumentNullException(nameof(logger));
_scheduler = new EntityActionScheduler <ClusterInfo>(cluster => ProbeCluster(cluster), autoStart: false, runOnce: false, timerFactory);
}
public void Schedule_AutoStartDisabledRunOnceEnabled_StartsManuallyAndRunsEachRegistrationOnlyOnce()
{
var entity0 = new Entity {
Id = "entity0"
};
var entity1 = new Entity {
Id = "entity1"
};
Entity lastInvokedEntity = null;
var timerFactory = new TestTimerFactory();
using var scheduler = new EntityActionScheduler <Entity>(e =>
{
lastInvokedEntity = e;
return(Task.CompletedTask);
}, autoStart: false, runOnce: true, timerFactory);
scheduler.ScheduleEntity(entity0, TimeSpan.FromMilliseconds(Period0));
scheduler.ScheduleEntity(entity1, TimeSpan.FromMilliseconds(Period1));
Assert.Equal(2, timerFactory.Count);
timerFactory.VerifyTimer(0, Timeout.Infinite);
timerFactory.VerifyTimer(1, Timeout.Infinite);
scheduler.Start();
VerifyEntities(scheduler, entity0, entity1);
Assert.Equal(2, timerFactory.Count);
timerFactory.VerifyTimer(0, Period0);
timerFactory.VerifyTimer(1, Period1);
timerFactory.FireTimer(1);
Assert.Same(entity1, lastInvokedEntity);
VerifyEntities(scheduler, entity0);
timerFactory.FireTimer(0);
Assert.Same(entity0, lastInvokedEntity);
Assert.False(scheduler.IsScheduled(entity0));
Assert.False(scheduler.IsScheduled(entity1));
timerFactory.AssertTimerDisposed(0);
timerFactory.AssertTimerDisposed(1);
}
public void Schedule_AutoStartEnabledRunOnceDisabled_StartsAutomaticallyAndRunsIndefinitely()
{
var entity0 = new Entity {
Id = "entity0"
};
var entity1 = new Entity {
Id = "entity1"
};
var timerFactory = new TestTimerFactory();
Entity lastInvokedEntity = null;
using var scheduler = new EntityActionScheduler <Entity>(e =>
{
lastInvokedEntity = e;
return(Task.CompletedTask);
}, autoStart: true, runOnce: false, timerFactory);
scheduler.ScheduleEntity(entity0, TimeSpan.FromMilliseconds(20000));
scheduler.ScheduleEntity(entity1, TimeSpan.FromMilliseconds(10000));
VerifyEntities(scheduler, entity0, entity1);
Assert.Equal(2, timerFactory.Count);
timerFactory.VerifyTimer(0, Period0);
timerFactory.VerifyTimer(1, Period1);
timerFactory.FireTimer(1);
Assert.Same(entity1, lastInvokedEntity);
timerFactory.FireTimer(0);
Assert.Same(entity0, lastInvokedEntity);
timerFactory.FireTimer(1);
Assert.Same(entity1, lastInvokedEntity);
timerFactory.FireTimer(0);
Assert.Same(entity0, lastInvokedEntity);
VerifyEntities(scheduler, entity0, entity1);
Assert.Equal(2, timerFactory.Count);
timerFactory.VerifyTimer(0, Period0);
timerFactory.VerifyTimer(1, Period1);
}
public void Unschedule_EntityUnscheduledAfterFirstCall_CallbackInvokedOnlyOnce()
{
var entity0 = new Entity {
Id = "entity0"
};
var entity1 = new Entity {
Id = "entity1"
};
Entity lastInvokedEntity = null;
var timerFactory = new TestTimerFactory();
using var scheduler = new EntityActionScheduler <Entity>(e =>
{
lastInvokedEntity = e;
return(Task.CompletedTask);
}, autoStart: true, runOnce: false, timerFactory);
scheduler.ScheduleEntity(entity0, TimeSpan.FromMilliseconds(Period0));
scheduler.ScheduleEntity(entity1, TimeSpan.FromMilliseconds(Period1));
VerifyEntities(scheduler, entity0, entity1);
timerFactory.FireTimer(1);
Assert.Same(entity1, lastInvokedEntity);
timerFactory.FireTimer(0);
Assert.Same(entity0, lastInvokedEntity);
scheduler.UnscheduleEntity(entity1);
VerifyEntities(scheduler, entity0);
timerFactory.AssertTimerDisposed(1);
timerFactory.FireTimer(0);
Assert.Same(entity0, lastInvokedEntity);
VerifyEntities(scheduler, entity0);
}
public ReactivationScheduler(ITimerFactory timerFactory, ILogger <ReactivationScheduler> logger)
{
_scheduler = new EntityActionScheduler <DestinationInfo>(d => Reactivate(d), autoStart: true, runOnce: true, timerFactory);
_logger = logger ?? throw new ArgumentNullException(nameof(logger));
}
