public void Run()
{
ETTaskCompletionSource tcs = this.Callback;
this.GetParent <TimerComponent>().Remove(this.Id);
tcs.SetResult();
}
public static ETTask FromException(Exception ex)
{
ETTaskCompletionSource tcs = new ETTaskCompletionSource();
tcs.TrySetException(ex);
return(tcs.Task);
}
public static ETTask <T> FromCanceled <T>(CancellationToken token)
{
var tcs = new ETTaskCompletionSource <T>();
tcs.TrySetException(new OperationCanceledException(token));
return(tcs.Task);
}
static CanceledETTaskCache()
{
ETTaskCompletionSource tcs = new ETTaskCompletionSource();
tcs.TrySetCanceled();
Task = tcs.Task;
}
public static ETTask <T> FromException <T>(Exception ex)
{
var tcs = new ETTaskCompletionSource <T>();
tcs.TrySetException(ex);
return(tcs.Task);
}
public ETTask <IResponse> Call(IRequest request)
{
int rpcId = ++RpcId;
var tcs = new ETTaskCompletionSource <IResponse>();
this.requestCallback[rpcId] = (response) =>
{
if (response is ErrorResponse)
{
tcs.SetException(new Exception($"Rpc error: {MongoHelper.ToJson(response)}"));
return;
}
if (ErrorCode.IsRpcNeedThrowException(response.Error))
{
tcs.SetException(new Exception($"Rpc error: {MongoHelper.ToJson(response)}"));
return;
}
tcs.SetResult(response);
};
request.RpcId = rpcId;
this.Send(request);
return(tcs.Task);
}
public ETTask WaitAsync(long time)
{
long tillTime = TimeHelper.Now() + time;
ETTaskCompletionSource tcs = new ETTaskCompletionSource();
OnceWaitTimer timer = EntityFactory.CreateWithParent <OnceWaitTimer, ETTaskCompletionSource>(this, tcs);
this.timers[timer.Id] = timer;
AddToTimeId(tillTime, timer.Id);
return(tcs.Task);
}
public ETTask WaitTillAsync(long tillTime, ETCancellationToken cancellationToken)
{
if (TimeHelper.Now() > tillTime)
{
return(ETTask.CompletedTask);
}
ETTaskCompletionSource tcs = new ETTaskCompletionSource();
OnceWaitTimer timer = EntityFactory.CreateWithParent <OnceWaitTimer, ETTaskCompletionSource>(this, tcs);
this.timers[timer.Id] = timer;
AddToTimeId(tillTime, timer.Id);
cancellationToken.Register(() => { this.Remove(timer.Id); });
return(tcs.Task);
}
public void SetResult()
{
if (moveNext == null)
{
}
else
{
if (this.tcs == null)
{
this.tcs = new ETTaskCompletionSource();
}
this.tcs.TrySetResult();
}
}
public void SetException(Exception exception)
{
if (this.tcs == null)
{
this.tcs = new ETTaskCompletionSource();
}
if (exception is OperationCanceledException ex)
{
this.tcs.TrySetCanceled(ex);
}
else
{
this.tcs.TrySetException(exception);
}
}
public void SetResult(T ret)
{
if (moveNext == null)
{
this.result = ret;
}
else
{
if (this.tcs == null)
{
this.tcs = new ETTaskCompletionSource <T>();
}
this.tcs.TrySetResult(ret);
}
}
public ETTask <CoroutineLock> Wait(CoroutineLockType coroutineLockType, long key)
{
CoroutineLockQueueType coroutineLockQueueType = this.list[(int)coroutineLockType];
if (!coroutineLockQueueType.TryGetValue(key, out CoroutineLockQueue queue))
{
queue = EntityFactory.Create <CoroutineLockQueue>(this.Domain);
coroutineLockQueueType.Add(key, queue);
return(ETTask.FromResult(EntityFactory.CreateWithParent <CoroutineLock, CoroutineLockType, long>(this, coroutineLockType, key)));
}
ETTaskCompletionSource <CoroutineLock> tcs = new ETTaskCompletionSource <CoroutineLock>();
queue.Enqueue(tcs);
return(tcs.Task);
}
public void AwaitOnCompleted <TAwaiter, TStateMachine>(ref TAwaiter awaiter, ref TStateMachine stateMachine)
where TAwaiter : INotifyCompletion
where TStateMachine : IAsyncStateMachine
{
if (moveNext == null)
{
if (this.tcs == null)
{
this.tcs = new ETTaskCompletionSource(); // built future.
}
var runner = new MoveNextRunner <TStateMachine>();
moveNext = runner.Run;
runner.StateMachine = stateMachine; // set after create delegate.
}
awaiter.OnCompleted(moveNext);
}
public void Notify(CoroutineLockType coroutineLockType, long key)
{
CoroutineLockQueueType coroutineLockQueueType = this.list[(int)coroutineLockType];
if (!coroutineLockQueueType.TryGetValue(key, out CoroutineLockQueue queue))
{
throw new Exception($"first work notify not find queue");
}
if (queue.Count == 0)
{
coroutineLockQueueType.Remove(key);
queue.Dispose();
return;
}
ETTaskCompletionSource <CoroutineLock> tcs = queue.Dequeue();
tcs.SetResult(EntityFactory.CreateWithParent <CoroutineLock, CoroutineLockType, long>(this, coroutineLockType, key));
}
public void Enqueue(ETTaskCompletionSource <CoroutineLock> tcs)
{
this.queue.Enqueue(tcs);
}