/
common.go
86 lines (73 loc) · 2.77 KB
/
common.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
package test
import (
"context"
"fmt"
"github.com/0xPolygonHermez/zkevm-node/db"
"github.com/0xPolygonHermez/zkevm-node/event"
"github.com/0xPolygonHermez/zkevm-node/event/nileventstorage"
"github.com/0xPolygonHermez/zkevm-node/l1infotree"
"github.com/0xPolygonHermez/zkevm-node/log"
"github.com/0xPolygonHermez/zkevm-node/merkletree"
"github.com/0xPolygonHermez/zkevm-node/merkletree/hashdb"
"github.com/0xPolygonHermez/zkevm-node/state"
"github.com/0xPolygonHermez/zkevm-node/state/pgstatestorage"
"github.com/0xPolygonHermez/zkevm-node/state/runtime/executor"
"github.com/0xPolygonHermez/zkevm-node/test/dbutils"
"github.com/0xPolygonHermez/zkevm-node/test/testutils"
"github.com/jackc/pgx/v4/pgxpool"
"google.golang.org/grpc"
)
const (
Ether155V = 27
)
var (
stateTree *merkletree.StateTree
stateDb *pgxpool.Pool
err error
StateDBCfg = dbutils.NewStateConfigFromEnv()
ctx = context.Background()
ExecutorClient executor.ExecutorServiceClient
mtDBServiceClient hashdb.HashDBServiceClient
executorClientConn, mtDBClientConn *grpc.ClientConn
executorCancel, mtDBCancel context.CancelFunc
Genesis = state.Genesis{}
)
func CloseTestState() {
stateDb.Close()
executorCancel()
executorClientConn.Close()
mtDBCancel()
mtDBClientConn.Close()
}
func InitTestState(stateCfg state.Config) *state.State {
InitOrResetDB(StateDBCfg)
stateDb, err = db.NewSQLDB(StateDBCfg)
if err != nil {
panic(err)
}
zkProverURI := testutils.GetEnv("ZKPROVER_URI", "zkevm-prover")
executorServerConfig := executor.Config{URI: fmt.Sprintf("%s:50071", zkProverURI), MaxGRPCMessageSize: 100000000}
ExecutorClient, executorClientConn, executorCancel = executor.NewExecutorClient(ctx, executorServerConfig)
s := executorClientConn.GetState()
log.Infof("executorClientConn state: %s", s.String())
mtDBServerConfig := merkletree.Config{URI: fmt.Sprintf("%s:50061", zkProverURI)}
mtDBServiceClient, mtDBClientConn, mtDBCancel = merkletree.NewMTDBServiceClient(ctx, mtDBServerConfig)
s = mtDBClientConn.GetState()
log.Infof("stateDbClientConn state: %s", s.String())
stateTree = merkletree.NewStateTree(mtDBServiceClient)
eventStorage, err := nileventstorage.NewNilEventStorage()
if err != nil {
panic(err)
}
eventLog := event.NewEventLog(event.Config{}, eventStorage)
mt, err := l1infotree.NewL1InfoTree(32, [][32]byte{})
if err != nil {
panic(err)
}
return state.NewState(stateCfg, pgstatestorage.NewPostgresStorage(stateCfg, stateDb), ExecutorClient, stateTree, eventLog, mt)
}
func InitOrResetDB(cfg db.Config) {
if err := dbutils.InitOrResetState(cfg); err != nil {
panic(err)
}
}