internal void AddActivity(string id, string activityName, string activityStatus, string data, int order, int compensateTimes, int overtimeCompensateTimes)
{
var targetActivityType = activityFinder.FindType(activityName);
var activityInterface = targetActivityType.GetInterfaces().FirstOrDefault();
if (!activityInterface.IsGenericType)
{
throw new ActivityNotFoundWhenCompensateRetryException(activityName);
}
var dataType = activityInterface.GetGenericArguments()[0];
var dataParameter = serializer.Deserialize(data, dataType);
ActivityWapper activityWapper = new ActivityWapper
{
Name = activityName,
ActivityDataType = dataType,
ActivityStatus = (ActivityStatus)Enum.Parse(typeof(ActivityStatus), activityStatus),
ActivityType = targetActivityType,
DataObj = dataParameter,
Order = order,
ServiceProvider = serviceProvider,
Id = id,
CompensateTimes = compensateTimes,
OvertimeCompensateTimes = overtimeCompensateTimes
};
activities.Add(activityWapper);
}
public void AddActivity(string activityName, object data, int timeOutSeconds = 2)
{
var targetActivityType = activityFinder.FindType(activityName);
var activityInterface = targetActivityType.GetInterfaces().FirstOrDefault();
if (!activityInterface.IsGenericType)
{
throw new ActivityImplementIrregularException(activityName);
}
var dataType = activityInterface.GetGenericArguments()[0];
ActivityWapper activityWapper = new ActivityWapper
{
Name = activityName,
ActivityDataType = dataType,
ActivityStatus = ActivityStatus.NotStarted,
ActivityType = targetActivityType,
DataObj = data,
Order = CurrentMaxOrder + 1,
ServiceProvider = serviceProvider,
TimeOutSeconds = timeOutSeconds,
};
activities.Add(activityWapper);
}
private async Task RecursiveCompensateActivity(ActivityWapper activityWapper)
{
var activityId = activityWapper.Id;
if (activityWapper.ActivityStatus == ActivityStatus.ExecutingOvertime)
{
activityWapper.OvertimeCompensateTimes++;
}
else
{
activityWapper.CompensateTimes++;
}
try
{
if (activityWapper.ActivityStatus != ActivityStatus.ExecutingOvertime)
{
activityWapper.ActivityStatus = ActivityStatus.Compensating;
}
await activityWapper.InvokeCompensate();
if (activityWapper.ActivityStatus == ActivityStatus.ExecutingOvertime)
{
// 超时 补偿次数已到
var isCompensated = activityWapper.OvertimeCompensateTimes >= poleSagasOption.MaxOvertimeCompensateTimes;
if (isCompensated)
{
activityWapper.ActivityStatus = ActivityStatus.Compensated;
}
await eventSender.ActivityOvertimeCompensated(activityId, isCompensated);
}
else
{
activityWapper.ActivityStatus = ActivityStatus.Compensated;
await eventSender.ActivityCompensated(activityId);
}
var compensateActivity = GetNextCompensateActivity();
if (compensateActivity == null)
{
return;
}
await RecursiveCompensateActivity(compensateActivity);
}
catch (Exception exception)
{
activityWapper.ActivityStatus = ActivityStatus.CompensateAborted;
// todo: 超时操作 如果出错可能 减少 补偿次数 这里 先不做处理
if (activityWapper.CompensateTimes >= poleSagasOption.MaxCompensateTimes)
{
// 此时 结束 saga 并且设置状态 为 Error
await eventSender.ActivityCompensateAborted(activityId, Id, exception.InnerException != null?exception.InnerException.Message + exception.StackTrace : exception.Message + exception.StackTrace);
}
else
{
await eventSender.ActivityCompensateAborted(activityId, string.Empty, exception.InnerException != null?exception.InnerException.Message + exception.StackTrace : exception.Message + exception.StackTrace);
}
}
}
private async Task <ActivityExecuteResult> RecursiveExecuteActivity(ActivityWapper activityWapper)
{
var activityId = snowflakeIdGenerator.NextId();
activityWapper.Id = activityId;
activityWapper.CancellationTokenSource = new System.Threading.CancellationTokenSource(activityWapper.TimeOutSeconds * 1000);
try
{
var content = serializer.Serialize(activityWapper.DataObj, activityWapper.ActivityDataType);
activityWapper.ActivityStatus = ActivityStatus.Executing;
await eventSender.ActivityExecuting(activityId, activityWapper.Name, Id, content, activityWapper.Order, DateTime.UtcNow);
var result = await activityWapper.InvokeExecute();
if (!result.IsSuccess)
{
activityWapper.ActivityStatus = ActivityStatus.Revoked;
await eventSender.ActivityRevoked(activityId);
await CompensateActivity(result, currentExecuteOrder);
var expiresAt = DateTime.UtcNow.AddSeconds(poleSagasOption.CompeletedSagaExpiredAfterSeconds);
await eventSender.SagaEnded(Id, expiresAt);
return(result);
}
activityWapper.ActivityStatus = ActivityStatus.Executed;
var executeActivity = GetNextExecuteActivity();
if (executeActivity == null)
{
var expiresAt = DateTime.UtcNow.AddSeconds(poleSagasOption.CompeletedSagaExpiredAfterSeconds);
await eventSender.SagaEnded(Id, expiresAt);
return(result);
}
else
{
return(await RecursiveExecuteActivity(executeActivity));
}
}
catch (Exception exception)
{
if (activityWapper.CancellationTokenSource.Token.IsCancellationRequested)
{
var errors = exception.InnerException != null ? exception.InnerException.Message + exception.StackTrace : exception.Message + exception.StackTrace;
var result = new ActivityExecuteResult
{
IsSuccess = false,
Errors = errors
};
var bytesContent = serializer.SerializeToUtf8Bytes(activityWapper.DataObj, activityWapper.ActivityDataType);
activityWapper.ActivityStatus = ActivityStatus.ExecutingOvertime;
await eventSender.ActivityExecuteOvertime(activityId);
// 超时的时候 需要首先补偿这个超时的操作
return(await CompensateActivity(result, currentExecuteOrder + 1));
}
else
{
var errors = exception.InnerException != null ? exception.InnerException.Message + exception.StackTrace : exception.Message + exception.StackTrace;
var result = new ActivityExecuteResult
{
IsSuccess = false,
Errors = errors
};
activityWapper.ActivityStatus = ActivityStatus.ExecuteAborted;
await eventSender.ActivityExecuteAborted(activityId);
// 出错的时候 需要首先补偿这个出错的操作
var executeResult = await CompensateActivity(result, currentExecuteOrder + 1);
var expiresAt = DateTime.UtcNow.AddSeconds(poleSagasOption.CompeletedSagaExpiredAfterSeconds);
if (IsCompensated)
{
await eventSender.SagaEnded(Id, expiresAt);
}
return(executeResult);
}
}
}