forked from cerc-io/laconicd-deprecated
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package websocket
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
"github.com/gorilla/mux"
|
|
|
|
"github.com/cosmos/ethermint/rpc/websockets"
|
|
"github.com/cosmos/ethermint/server/config"
|
|
)
|
|
|
|
type Service struct {
|
|
websocketServer *websockets.Server
|
|
}
|
|
|
|
// NewService creates a new gRPC server instance with a defined listener address.
|
|
func NewService(clientCtx client.Context) *Service {
|
|
return &Service{
|
|
websocketServer: websockets.NewServer(clientCtx),
|
|
}
|
|
}
|
|
|
|
// Name returns the JSON-RPC service name
|
|
func (Service) Name() string {
|
|
return "Ethereum Websocket"
|
|
}
|
|
|
|
// Start runs the websocket server
|
|
func (s Service) Start(cfg config.Config) error {
|
|
s.websocketServer.Address = cfg.JSONRPC.Address
|
|
|
|
u, err := url.Parse(cfg.EthereumWebsocket.Address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ws := mux.NewRouter()
|
|
ws.Handle("/", s.websocketServer)
|
|
|
|
errCh := make(chan error)
|
|
go func() {
|
|
err := http.ListenAndServe(":"+u.Port(), ws)
|
|
if err != nil {
|
|
errCh <- err
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case err := <-errCh:
|
|
return err
|
|
case <-time.After(5 * time.Second): // assume server started successfully
|
|
return nil
|
|
}
|
|
}
|