task.go

package main

import (
	"context"
	"fmt"
	"strings"

	"github.com/blockchain-tps-test/samples/theta/tps"
)

const (
	Eth tps.TaskType = iota

	TaskRetryLimit = 10
)

type EthTask struct {
	to            string
	amount        int64
	tokenId       int64
	tryCount      int
	transfer_type string
}

func (t *EthTask) Type() tps.TaskType {
	return Eth
}

func (t *EthTask) TryCount() int {
	return t.tryCount
}

func (t *EthTask) IncrementTryCount() error {
	t.tryCount += 1
	if t.tryCount >= TaskRetryLimit {
		return fmt.Errorf("err task retry limit, tryCount: %d", t.tryCount)
	}
	return nil
}

func (t *EthTask) Do(ctx context.Context, client *EthClient, priv string, nonce uint64, queue *tps.Queue, logger tps.Logger, contractAddress string) error {
	//根据不同的model生成不同的发送交易的任务
	var rootErr error
	if t.transfer_type == "CrossChain" {
		_, rootErr = client.CrossChainTNT20Transfer(ctx, priv, nonce, t.to, t.amount, contractAddress, 1) //链间交易
	} else if t.transfer_type == "InChain" {
		//_, rootErr = client.CrossSubChainTNT20Transfer(ctx, priv, nonce, t.to, t.amount, contractAddress, 1) //链内TNT20
		_, rootErr = client.Erc20TransferFrom(ctx, priv, nonce, t.to, t.amount, contractAddress, 1)
	} else {
		logger.Fatal("err model")
	}

	if rootErr != nil { //根据错误捕捉，并返回错误类型
		if strings.Contains(rootErr.Error(), "Invalid Transaction") {
			//logger.Warn(fmt.Sprintf("nonce error, %s", rootErr.Error()))
			logger.Fatal("err model")
			return tps.ErrWrongNonce
		}
		if strings.Contains(rootErr.Error(), "Transaction already seen") {
			//logger.Warn(fmt.Sprintf("nonce error, %s", rootErr.Error()))
			logger.Fatal("err model")
			return tps.ErrTaskRetry
		}
		if strings.Contains(rootErr.Error(), "ValidateInputAdvanced: Got") {
			logger.Warn(fmt.Sprintf("nonce error, %s", rootErr.Error()))
			logger.Fatal("err model")
			return rootErr
		}
		logger.Warn(fmt.Sprintf("faild sending, err: %s", rootErr.Error()))
		if err := t.IncrementTryCount(); err != nil {
			logger.Fatal("err model")
			return tps.ErrWrongNonce
		}
		// if strings.Contains(rootErr.Error(), "ValidateInputAdvanced:"){
		// 	return tps.NonceWrong
		// }
		return tps.ErrWrongNonce
		queue.Push(t)
		return nil
	}

	return nil
}