mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 10:05:18 +00:00
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/0glabs/0g-chain/helper/da/service"
|
|
"github.com/0glabs/0g-chain/helper/da/types"
|
|
|
|
"github.com/lesismal/nbio/nbhttp"
|
|
"github.com/lesismal/nbio/nbhttp/websocket"
|
|
)
|
|
|
|
const (
|
|
subscribeMsg = "{\"jsonrpc\":\"2.0\",\"method\":\"subscribe\",\"id\":1,\"params\":{\"query\":\"tm.event='Tx'\"}}"
|
|
)
|
|
|
|
var (
|
|
rpcAddress = flag.String("rpc-address", "34.214.2.28:32001", "address of da-light rpc server")
|
|
wsAddress = flag.String("ws-address", "127.0.0.1:26657", "address of emvos ws server")
|
|
relativePath = flag.String("relative-path", "", "relative path of evmosd")
|
|
account = flag.String("account", "", "account to run evmosd cli")
|
|
keyring = flag.String("keyring", "", "keyring to run evmosd cli")
|
|
homePath = flag.String("home", "", "home path of evmosd node")
|
|
)
|
|
|
|
func newUpgrader() *websocket.Upgrader {
|
|
u := websocket.NewUpgrader()
|
|
u.OnMessage(func(c *websocket.Conn, messageType websocket.MessageType, data []byte) {
|
|
log.Println("onEcho:", string(data))
|
|
ctx := context.WithValue(context.Background(), types.DA_RPC_ADDRESS, *rpcAddress)
|
|
ctx = context.WithValue(ctx, types.NODE_CLI_RELATIVE_PATH, *relativePath)
|
|
ctx = context.WithValue(ctx, types.NODE_CLI_EXEC_ACCOUNT, *account)
|
|
ctx = context.WithValue(ctx, types.NODE_CLI_EXEC_KEYRING, *keyring)
|
|
ctx = context.WithValue(ctx, types.NODE_HOME_PATH, *homePath)
|
|
go func() { service.OnMessage(ctx, c, messageType, data) }()
|
|
})
|
|
|
|
u.OnClose(func(c *websocket.Conn, err error) {
|
|
fmt.Println("OnClose:", c.RemoteAddr().String(), err)
|
|
service.OnClose()
|
|
})
|
|
|
|
return u
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
engine := nbhttp.NewEngine(nbhttp.Config{})
|
|
err := engine.Start()
|
|
if err != nil {
|
|
fmt.Printf("nbio.Start failed: %v\n", err)
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
u := url.URL{Scheme: "ws", Host: *wsAddress, Path: "/websocket"}
|
|
dialer := &websocket.Dialer{
|
|
Engine: engine,
|
|
Upgrader: newUpgrader(),
|
|
DialTimeout: time.Second * 3,
|
|
}
|
|
c, res, err := dialer.Dial(u.String(), nil)
|
|
if err != nil {
|
|
if res != nil && res.Body != nil {
|
|
bReason, _ := io.ReadAll(res.Body)
|
|
fmt.Printf("dial failed: %v, reason: %v\n", err, string(bReason))
|
|
} else {
|
|
fmt.Printf("dial failed: %v\n", err)
|
|
}
|
|
return
|
|
}
|
|
c.WriteMessage(websocket.TextMessage, []byte(subscribeMsg))
|
|
}()
|
|
|
|
interrupt := make(chan os.Signal, 1)
|
|
signal.Notify(interrupt, os.Interrupt)
|
|
<-interrupt
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
engine.Shutdown(ctx)
|
|
}
|