forked from ava-labs/avalanchego
/
main.go
97 lines (80 loc) · 2.33 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
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package main
import (
"fmt"
"os"
"path"
"runtime"
"runtime/pprof"
"github.com/ava-labs/salticidae-go"
"github.com/ava-labs/gecko/genesis"
"github.com/ava-labs/gecko/ids"
"github.com/ava-labs/gecko/utils/crypto"
"github.com/ava-labs/gecko/utils/logging"
"github.com/ava-labs/gecko/vms/avm"
"github.com/ava-labs/gecko/vms/spchainvm"
"github.com/ava-labs/gecko/vms/spdagvm"
)
func main() {
if err != nil {
fmt.Printf("Failed to parse arguments: %s\n", err)
}
// set up logging
config.LoggingConfig.Directory = path.Join(config.LoggingConfig.Directory, "client")
log, err := logging.New(config.LoggingConfig)
if err != nil {
fmt.Printf("Failed to start the logger: %s\n", err)
return
}
defer log.Stop()
// initialize state based on CLI args
net.log = log
crypto.EnableCrypto = config.EnableCrypto
net.decided = make(chan ids.ID, config.MaxOutstandingTxs)
// Init the network
log.AssertNoError(net.Initialize())
net.net.Start()
defer net.net.Stop()
// connect to the node
serr := salticidae.NewError()
remoteIP := salticidae.NewNetAddrFromIPPortString(config.RemoteIP.String(), true, &serr)
if code := serr.GetCode(); code != 0 {
log.Fatal("Sync error %s", salticidae.StrError(serr.GetCode()))
return
}
net.conn = net.net.ConnectSync(remoteIP, true, &serr)
if serr.GetCode() != 0 {
log.Fatal("Sync error %s", salticidae.StrError(serr.GetCode()))
return
}
// start a cpu profile
file, gErr := os.Create("cpu_client.profile")
log.AssertNoError(gErr)
gErr = pprof.StartCPUProfile(file)
log.AssertNoError(gErr)
runtime.SetMutexProfileFraction(1)
defer file.Close()
defer pprof.StopCPUProfile()
net.networkID = config.NetworkID
// start the benchmark we want to run
switch config.Chain {
case spChain:
tx, err := genesis.VMGenesis(config.NetworkID, spchainvm.ID)
log.AssertNoError(err)
net.benchmarkSPChain(tx)
case spDAG:
tx, err := genesis.VMGenesis(config.NetworkID, spdagvm.ID)
log.AssertNoError(err)
net.benchmarkSPChain(tx)
case avmDAG:
tx, err := genesis.VMGenesis(config.NetworkID, avm.ID)
log.AssertNoError(err)
net.benchmarkSPChain(tx)
default:
log.Fatal("did not specify whether to test dag or chain. Exiting")
return
}
// start processing network messages
net.ec.Dispatch()
}