-
Notifications
You must be signed in to change notification settings - Fork 10
/
main.go
118 lines (103 loc) · 3.48 KB
/
main.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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package main
import (
"context"
"os"
"os/signal"
"sync"
"syscall"
"time"
"github.com/ethereum/go-ethereum/common"
"github.com/rs/zerolog/log"
"github.com/textileio/go-tableland/buildinfo"
"github.com/textileio/go-tableland/cmd/healthbot/counterprobe"
"github.com/textileio/go-tableland/pkg/client"
clientV1 "github.com/textileio/go-tableland/pkg/client/v1"
"github.com/textileio/go-tableland/pkg/logging"
"github.com/textileio/go-tableland/pkg/metrics"
"github.com/textileio/go-tableland/pkg/wallet"
)
func main() {
cfg := setupConfig()
logging.SetupLogger(buildinfo.GitCommit, cfg.Log.Debug, cfg.Log.Human)
if err := metrics.SetupInstrumentation(":"+cfg.Metrics.Port, "tableland:healthbot"); err != nil {
log.Fatal().Err(err).Str("port", cfg.Metrics.Port).Msg("could not setup instrumentation")
}
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM, syscall.SIGQUIT)
defer cancel()
var wg sync.WaitGroup
for _, chainCfg := range cfg.Chains {
checkInterval, err := time.ParseDuration(chainCfg.Probe.CheckInterval)
if err != nil {
log.Fatal().Err(err).Msgf("check interval has invalid format: %s", chainCfg.Probe.CheckInterval)
}
receiptTimeout, err := time.ParseDuration(chainCfg.Probe.ReceiptTimeout)
if err != nil {
log.Fatal().Err(err).Msgf("receipt timeout has invalid format: %s", chainCfg.Probe.ReceiptTimeout)
}
wallet, err := wallet.NewWallet(chainCfg.WalletPrivateKey)
if err != nil {
log.Fatal().Err(err).Msg("unable to create wallet from private key string")
}
chain, ok := client.Chains[client.ChainID(chainCfg.ChainID)]
if !ok {
log.Fatal().Int("chain_id", chainCfg.ChainID).Msg("the chain id isn't supported in the Tableland client")
}
if chainCfg.OverrideClient.GatewayEndpoint != "" {
chain.Endpoint = chainCfg.OverrideClient.GatewayEndpoint
}
if chainCfg.OverrideClient.ContractAddr != "" {
chain.ContractAddr = common.HexToAddress(chainCfg.OverrideClient.ContractAddr)
}
// For Filecoin Calibration, we use Glif endpoint
opts := []clientV1.NewClientOption{clientV1.NewClientChain(chain)}
if chain.ID == 314159 {
// Glif API key is empty string because currently we are using the free tier (public node)
opts = append(opts, clientV1.NewClientGlifAPIKey(chainCfg.GlifAPIKey))
} else {
opts = append(opts, clientV1.NewClientAlchemyAPIKey(chainCfg.AlchemyAPIKey))
}
client, err := clientV1.NewClient(ctx, wallet, opts...)
if err != nil {
log.Fatal().Err(err).Msg("error creating tbl client")
}
suggestedGasPriceMultiplier := 1.0
if chainCfg.OverrideClient.SuggestedGasPriceMultiplier > 0 {
suggestedGasPriceMultiplier = chainCfg.OverrideClient.SuggestedGasPriceMultiplier
}
estimatedGasLimitMultiplier := 1.0
if chainCfg.OverrideClient.EstimatedGasLimitMultiplier > 0 {
estimatedGasLimitMultiplier = chainCfg.OverrideClient.EstimatedGasLimitMultiplier
}
cp, err := counterprobe.New(
chain.Name,
client,
chainCfg.Probe.Tablename,
checkInterval,
receiptTimeout,
suggestedGasPriceMultiplier,
estimatedGasLimitMultiplier,
)
if err != nil {
log.Fatal().Err(err).Msg("initializing counter-probe")
}
balanceTracker, err := NewBalanceTracker(
chainCfg,
wallet,
15*time.Second,
)
if err != nil {
log.Fatal().Err(err).Msg("initializing balance tracker")
}
wg.Add(2)
go func() {
defer wg.Done()
cp.Run(ctx)
}()
go func() {
defer wg.Done()
balanceTracker.Run(ctx)
}()
}
wg.Wait()
log.Info().Msg("daemon closed")
}