public async Task <bool> Perform(JobContext jobContext)
{
var store = Ioc.GetRequiredService <ISwarmStore>();
var logger = Ioc.GetRequiredService <ILoggerFactory>().CreateLogger <SignalRPerformer>();
var hubContext = Ioc.GetRequiredService <IHubContext <ClientHub> >();
var clients = (await store.GetClients(jobContext.Group)).Where(c => c.IsConnected).ToList();
if (clients.Any())
{
// TODO: 实现分片算法
var shardingParameters = jobContext.ShardingParameters == null
? new string[0]
: jobContext.ShardingParameters?.Split(new[] { ";" }, StringSplitOptions.RemoveEmptyEntries);
var dic = new Dictionary <int, int>();
for (int s = 0; s < jobContext.Sharding; ++s)
{
dic.Add(s, 0);
}
for (int i = 0, j = 0; j < clients.Count && i < jobContext.Sharding;)
{
if (dic[i] >= _retryTimes)
{
return(false);
}
var client = clients[j];
var shardingParameter = i < shardingParameters.Length ? shardingParameters[i] : "";
var jc = jobContext.Clone();
try
{
jc.CurrentSharding = i + 1;
jc.CurrentShardingParameter = shardingParameter;
await store.AddJobState(new JobState
{
JobId = jc.JobId,
TraceId = jc.TraceId,
Client = client.Name,
Sharding = jc.CurrentSharding,
State = State.Performing,
Msg = $"Performing on {client}"
}
);
if (jc.Parameters.ContainsKey(SwarmConts.ArgumentsProperty))
{
var arguments = jc.Parameters[SwarmConts.ArgumentsProperty];
if (!string.IsNullOrWhiteSpace(arguments))
{
arguments = arguments.Replace("%jobid%", jc.JobId);
arguments = arguments.Replace("%traceid%", jc.TraceId);
arguments = arguments.Replace("%sharding%", jc.Sharding.ToString());
arguments = arguments.Replace("%currentsharding%", jc.CurrentSharding.ToString());
arguments = arguments.Replace("%currentshardingparameter%",
jc.CurrentShardingParameter);
arguments = arguments.Replace("%name%", jc.Name);
arguments = arguments.Replace("%group%", jc.Group);
arguments = arguments.Replace("%firetime%",
jc.FireTimeUtc.ToString("yyyy-MM-dd hh:mm:ss"));
jc.Parameters[SwarmConts.ArgumentsProperty] = arguments;
}
}
await hubContext.Clients.Client(client.ConnectionId).SendAsync("Trigger", jc);
await store.UpdateJobState(new JobState
{
TraceId = jc.TraceId,
Client = client.Name,
Sharding = jc.CurrentSharding,
JobId = jc.JobId,
State = State.Performed,
Msg = $"Perform on {client} success"
}
);
logger.LogInformation($"Perform {jobContext.JobId} on {client} success");
i++;
j++;
if (j == clients.Count)
{
j = 0;
}
}
catch (Exception ex)
{
logger.LogError($"Perform {jobContext.JobId} on {client} failed: {ex.Message}");
await store.UpdateJobState(new JobState
{
TraceId = jc.TraceId,
Client = client.Name,
Sharding = jc.CurrentSharding,
JobId = jc.JobId,
State = State.Exit,
Msg = $"Perform on {client} failed"
});
dic[i]++;
// 分片不增加, 尝试其它节点
j++;
if (j == clients.Count)
{
j = 0;
}
}
}
return(true);
}
else
{
logger.LogError($"Unfounded available client for job {jobContext.JobId}");
await store.AddJobState(new JobState
{
JobId = jobContext.JobId,
TraceId = jobContext.TraceId,
Client = "Swarm",
Sharding = 1,
State = State.Exit,
Msg = "Unfounded available client"
}
);
return(false);
}
}
