public static void WhenStartedAsLeader <T>(
this ServiceConfigurator <T> configurator, Action <LeaderConfigurationBuilder <T> > builder) where T : class
{
if (configurator == null)
{
throw new ArgumentNullException(nameof(configurator));
}
if (builder == null)
{
throw new ArgumentNullException(nameof(builder));
}
LeaderConfiguration <T> leaderConfiguration = null;
configurator.BeforeStoppingService(async() =>
{
try
{
leaderConfiguration?.ServiceIsStopping.Cancel();
}
catch (TaskCanceledException) { }
if (leaderConfiguration != null)
{
await leaderConfiguration.LeaseManager.ReleaseLease(leaderConfiguration.NodeId);
}
});
configurator.WhenStarted(async service =>
{
if (leaderConfiguration == null)
{
var configurationBuilder = new LeaderConfigurationBuilder <T>();
builder(configurationBuilder);
if (!configurationBuilder.ServiceStoppingTokenIsSet)
{
var serviceStoppingTokenSource = new CancellationTokenSource();
configurationBuilder.WhenStopping(serviceStoppingTokenSource);
}
leaderConfiguration = configurationBuilder.Build();
}
try
{
var worker = new Runner <T>(service, leaderConfiguration);
await worker.Start();
}
catch (TaskCanceledException)
{
// TaskCanceledException bubbles up if the service stopping cancellation token is set
// so we swallow the exception so that the Topshelf OnException handler doesn't see it
}
});
}
public Runner(T service, LeaderConfiguration <T> config)
{
this.service = service;
this.config = config ?? throw new ArgumentNullException(nameof(config));
}