-
Notifications
You must be signed in to change notification settings - Fork 0
/
task.go
61 lines (49 loc) · 1.3 KB
/
task.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package main
import (
"context"
"fmt"
"strings"
// "github.com/davecgh/go-spew/spew"
"github.com/blockchain-tps-test/samples/cosmos/tps"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/pkg/errors"
)
const (
Cosm tps.TaskType = iota
TaskRetryLimit = 10
)
type CosmTask struct {
to sdk.AccAddress
amount int64
tryCount int
}
func (t *CosmTask) Type() tps.TaskType {
return Cosm
}
func (t *CosmTask) TryCount() int {
return t.tryCount
}
func (t *CosmTask) IncrementTryCount() error {
t.tryCount += 1
if t.tryCount >= TaskRetryLimit {
return fmt.Errorf("err task retry limit, tryCount: %d", t.tryCount)
}
return nil
}
func (t *CosmTask) Do(ctx context.Context, client *CosmosClient, priv string, nonce uint64, queue *tps.Queue, logger tps.Logger) error {
_, rootErr := client.SendTx(ctx, priv, nonce, t.to, t.amount)
if rootErr != nil {
if strings.Contains(rootErr.Error(), "invalid sequence") {
logger.Warn(fmt.Sprintf("nonce error, %s", rootErr.Error()))
return tps.ErrWrongNonce
}
logger.Warn(fmt.Sprintf("faild sending, err: %s", rootErr.Error()))
if err := t.IncrementTryCount(); err != nil {
return errors.Wrap(rootErr, err.Error())
}
queue.Push(t)
return nil
}
//logger.Info(fmt.Sprintf("succeed sending tx, hash: %s, nonce: %d", res.Hash, nonce))
return nil
}