forked from finejian/go-connect-eth-contract
/
connecter.go
271 lines (247 loc) · 6.78 KB
/
connecter.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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
package coin
import (
"context"
"fmt"
"log"
"math/big"
"strings"
"github.com/ethereum/go-ethereum/accounts/abi/bind"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/ethclient"
)
var (
// CoinAddr superCoin 合约地址
CoinAddr = common.HexToAddress("0x84F70FEa5Ba54323C0EF85c58A47c98E1a2fe2Db")
// CoinHash superCoin 合约地址Hash
CoinHash = CoinAddr.Hash()
)
// Connecter SuperCoin连接者
type Connecter struct {
ctx context.Context
conn *ethclient.Client
coin *SuperCoin
}
// NewConnecter 生成一个SuperCoin连接者
func NewConnecter() *Connecter {
// Dial这里支持传入 ws、http、ipc的多种链接
// 如果是经常需要调用最好还是使用 ws 方式保持通讯状态
conn, err := ethclient.Dial("ws://127.0.0.1:8546")
if err != nil {
panic(err)
}
coin, err := NewSuperCoin(CoinAddr, conn)
if err != nil {
panic(err)
}
return &Connecter{
ctx: context.Background(),
conn: conn,
coin: coin,
}
}
// NewConnecterWithDeploy 部署合约,并创建一个connecter
func NewConnecterWithDeploy(ownerAuth *bind.TransactOpts) *Connecter {
conn, err := ethclient.Dial("ws://127.0.0.1:8546")
if err != nil {
panic(err)
}
_, tx, coin, err := DeploySuperCoin(ownerAuth, conn)
if err != nil {
panic(err)
}
ctx := context.Background()
CoinAddr, err = bind.WaitDeployed(ctx, conn, tx)
if err != nil {
panic(err)
}
CoinHash = CoinAddr.Hash()
return &Connecter{
ctx: ctx,
conn: conn,
coin: coin,
}
}
// BlockNumber 当前块高度
func (c *Connecter) BlockNumber() *big.Int {
blockNumber, err := c.conn.BlockByNumber(c.ctx, nil)
if err != nil {
log.Fatal("Get block number error", err)
return big.NewInt(0)
}
return blockNumber.Number()
}
// ContractName SuperCoin名称
func (c *Connecter) ContractName() string {
name, err := c.coin.Name(nil)
if err != nil {
log.Fatal("Get contract name error", err)
return ""
}
return name
}
// BalanceOfEth 以太币余额
func (c *Connecter) BalanceOfEth(addr common.Address) *big.Int {
balance, err := c.conn.BalanceAt(c.ctx, addr, nil)
if err != nil {
log.Fatal("Get address eth balance error", err)
return big.NewInt(0)
}
return balance
}
// BalanceOfCoin SuperCoin余额
func (c *Connecter) BalanceOfCoin(addr common.Address) *big.Int {
balance, err := c.coin.BalanceOf(nil, addr)
if err != nil {
log.Fatal("Get address SuperCoin balance error", err)
return big.NewInt(0)
}
return balance
}
// TotalSupply SuperCoin已铸币量
func (c *Connecter) TotalSupply() *big.Int {
totalSupply, err := c.coin.TotalSupply(nil)
if err != nil {
log.Fatal("Get SuperCoin totalSuply error", err)
return big.NewInt(0)
}
return totalSupply
}
// AuthAccount 解锁账户
// 正式使用时候,此处应该限制GasPrice和GasLimit
func AuthAccount(key, passphrase string) *bind.TransactOpts {
auth, err := bind.NewTransactor(strings.NewReader(key), passphrase)
if err != nil {
log.Fatalf("Auth account error: %v", err)
return nil
}
log.Printf("Gas price is: %v, Gas limit is: %v", auth.GasPrice, auth.GasLimit)
return auth
}
// TransferCoin SuperCoin转账
func (c *Connecter) TransferCoin(fromAuth *bind.TransactOpts, to common.Address, amount *big.Int) bool {
tx, err := c.coin.Transfer(fromAuth, to, amount)
if err != nil {
log.Fatalf("Transfer SuperCoin from %s to %s amount %s error: %v", fromAuth.From.String(), to.String(), amount.String(), err)
return false
}
// 等待执行
receipt, err := bind.WaitMined(c.ctx, c.conn, tx)
if err != nil {
log.Fatalf("Wait Transfer Mined error: %v", err)
return false
}
return receipt.Status == 1
}
// TransferLogs SuperCoin转账记录
func (c *Connecter) TransferLogs(froms []common.Address, tos []common.Address) {
iter, err := c.coin.FilterTransfer(&bind.FilterOpts{}, froms, tos)
defer iter.Close()
if err != nil {
panic(err)
}
for {
if !iter.Next() {
break
}
log.Printf("Transfer log: %s", iter.Event.String())
}
}
func (s *SuperCoinTransfer) String() string {
return fmt.Sprintf("From: %s, To: %s, Amount: %s", s.From.Hex(), s.To.Hex(), s.Value)
}
// ExistsWhiteList 地址是否在白名单
func (c *Connecter) ExistsWhiteList(addr common.Address) bool {
exists, err := c.coin.Whitelist(&bind.CallOpts{}, addr)
if err != nil {
log.Fatalf("Query address on the white list error: %v", err)
return false
}
return exists
}
// AddToWhitelist 地址是否在白名单
func (c *Connecter) AddToWhitelist(ownerAuth *bind.TransactOpts, addr common.Address) bool {
tx, err := c.coin.AddToWhitelist(ownerAuth, addr)
if err != nil {
log.Fatalf("Add address to the white list error: %v", err)
return false
}
// 等待执行
receipt, err := bind.WaitMined(c.ctx, c.conn, tx)
if err != nil {
log.Fatalf("Wait AddToWhitelist Mined error: %v", err)
return false
}
return receipt.Status == 1
}
// RemoveFromWhitelist 地址是否在白名单
func (c *Connecter) RemoveFromWhitelist(ownerAuth *bind.TransactOpts, addr common.Address) bool {
tx, err := c.coin.RemoveFromWhitelist(ownerAuth, addr)
if err != nil {
log.Fatalf("Remove address from the white list error: %v", err)
return false
}
// 等待执行
receipt, err := bind.WaitMined(c.ctx, c.conn, tx)
if err != nil {
log.Fatalf("Wait RemoveFromWhitelist Mined error: %v", err)
return false
}
return receipt.Status == 1
}
// Mint 给白名单地址铸币
func (c *Connecter) Mint(ownerAuth *bind.TransactOpts, addr common.Address, amount *big.Int) bool {
tx, err := c.coin.Mint(ownerAuth, addr, amount)
if err != nil {
log.Fatalf("Mint to address SC error: %v", err)
return false
}
// 等待执行
receipt, err := bind.WaitMined(c.ctx, c.conn, tx)
if err != nil {
log.Fatalf("Wait Mint SC Mined error: %v", err)
return false
}
return receipt.Status == 1
}
// MintLogs 铸币记录
func (c *Connecter) MintLogs() {
iter, err := c.coin.FilterMint(&bind.FilterOpts{}, nil)
defer iter.Close()
if err != nil {
panic(err)
}
for {
if !iter.Next() {
break
}
log.Printf("Mint log: %s", iter.Event.String())
}
}
func (s *SuperCoinMint) String() string {
return fmt.Sprintf("Address: %s, Amount: %s", s.To.String(), s.Amount.String())
}
// WatchTransferAndMint 监听transfer和mint事件
func (c *Connecter) WatchTransferAndMint() {
transferCh := make(chan *SuperCoinTransfer)
transferSub, err := c.coin.WatchTransfer(&bind.WatchOpts{}, transferCh, nil, nil)
if err != nil {
panic(err)
}
mintCh := make(chan *SuperCoinMint)
mintSub, err := c.coin.WatchMint(&bind.WatchOpts{}, mintCh, nil)
if err != nil {
panic(err)
}
for {
select {
case t := <-transferCh:
log.Printf("Transfer From: %s, To: %s, Amount: %d", t.From.String(), t.To.String(), t.Value)
case e := <-transferSub.Err():
panic(e)
case m := <-mintCh:
log.Printf("Mint To: %s, Amount: %d", m.To.String(), m.Amount)
case e := <-mintSub.Err():
panic(e)
}
}
}