public IMethodStub <TInput, TOutput> Create <TInput, TOutput>(Method <TInput, TOutput> method)
where TInput : IMessage <TInput>, new() where TOutput : IMessage <TOutput>, new()
{
Task <IExecutionResult <TOutput> > SendAsync(TInput input)
{
throw new NotSupportedException();
}
async Task <TOutput> CallAsync(TInput input)
{
var chainContext = _chainContext;
var transaction = new Transaction()
{
From = FromAddress,
To = ConsensusContractAddress,
MethodName = method.Name,
Params = ByteString.CopyFrom(method.RequestMarshaller.Serializer(input))
};
var trace =
await _transactionReadOnlyExecutionService.ExecuteAsync(chainContext, transaction,
_blockTimeProvider.GetBlockTime());
return(trace.IsSuccessful()
? method.ResponseMarshaller.Deserializer(trace.ReturnValue.ToByteArray())
: default);
public async Task <ContractReaderContext> GetContractReaderContextAsync(IChainContext chainContext)
{
var timestamp = _blockTimeProvider.GetBlockTime();
var sender = Address.FromPublicKey(await _accountService.GetPublicKeyAsync());
var consensusContractAddress = await _smartContractAddressService.GetAddressByContractNameAsync(
chainContext, ConsensusSmartContractAddressNameProvider.StringName);
return(new ContractReaderContext
{
BlockHash = chainContext.BlockHash,
BlockHeight = chainContext.BlockHeight,
ContractAddress = consensusContractAddress,
Sender = sender,
Timestamp = timestamp
});
}
public Timestamp GetBlockTime()
{
return(_blockTimeProvider.GetBlockTime());
}
public IMethodStub <TInput, TOutput> Create <TInput, TOutput>(Method <TInput, TOutput> method)
where TInput : IMessage <TInput>, new() where TOutput : IMessage <TOutput>, new()
{
Transaction GetTransaction(TInput input)
{
var refBlockInfo = _refBlockInfoProvider.GetRefBlockInfo();
var transaction = GetTransactionWithoutSignature(input, method);
transaction.RefBlockNumber = refBlockInfo.Height;
transaction.RefBlockPrefix = refBlockInfo.Prefix;
var signature = CryptoHelper.SignWithPrivateKey(
KeyPair.PrivateKey, transaction.GetHash().Value.ToByteArray());
transaction.Signature = ByteString.CopyFrom(signature);
return(transaction);
}
async Task <IExecutionResult <TOutput> > SendAsync(TInput input)
{
var transaction = GetTransaction(input);
var transactionResult = await _testTransactionExecutor.ExecuteAsync(transaction);
if (transactionResult == null)
{
return(new ExecutionResult <TOutput> {
Transaction = transaction
});
}
ResetBlockTime();
return(new ExecutionResult <TOutput>
{
Transaction = transaction, TransactionResult = transactionResult,
Output = method.ResponseMarshaller.Deserializer(transactionResult.ReturnValue.ToByteArray())
});
}
async Task <IExecutionResult <TOutput> > SendWithExceptionAsync(TInput input)
{
var transaction = GetTransaction(input);
var transactionResult = await _testTransactionExecutor.ExecuteWithExceptionAsync(transaction);
if (transactionResult == null)
{
return(new ExecutionResult <TOutput> {
Transaction = transaction
});
}
ResetBlockTime();
return(new ExecutionResult <TOutput>
{
Transaction = transaction, TransactionResult = transactionResult,
Output = method.ResponseMarshaller.Deserializer(transactionResult.ReturnValue.ToByteArray())
});
}
void ResetBlockTime()
{
if (!_resetBlockTimeProvider.Enabled)
{
return;
}
var currentBlockTime = _blockTimeProvider.GetBlockTime();
currentBlockTime.AddMilliseconds(_resetBlockTimeProvider.StepMilliseconds);
_blockTimeProvider.SetBlockTime(currentBlockTime);
}
async Task <TOutput> CallAsync(TInput input)
{
var transaction = GetTransactionWithoutSignature(input, method);
var returnValue = await _testTransactionExecutor.ReadAsync(transaction);
return(method.ResponseMarshaller.Deserializer(returnValue.ToByteArray()));
}
async Task <StringValue> CallWithExceptionAsync(TInput input)
{
var transaction = GetTransactionWithoutSignature(input, method);
var returnValue = await _testTransactionExecutor.ReadWithExceptionAsync(transaction);
return(new StringValue {
Value = returnValue.Value
});
}
return(new MethodStub <TInput, TOutput>(method, SendAsync, CallAsync, GetTransaction, SendWithExceptionAsync,
CallWithExceptionAsync));
}