mirror of https://github.com/status-im/go-waku.git
91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
package rpc
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/rpc/v2"
|
|
logging "github.com/ipfs/go-log"
|
|
"github.com/status-im/go-waku/waku/v2/node"
|
|
)
|
|
|
|
var log = logging.Logger("wakurpc")
|
|
|
|
type WakuRpc struct {
|
|
node *node.WakuNode
|
|
server *http.Server
|
|
|
|
relayService *RelayService
|
|
}
|
|
|
|
func NewWakuRpc(node *node.WakuNode, address string, port int) *WakuRpc {
|
|
s := rpc.NewServer()
|
|
s.RegisterCodec(NewSnakeCaseCodec(), "application/json")
|
|
s.RegisterCodec(NewSnakeCaseCodec(), "application/json;charset=UTF-8")
|
|
|
|
err := s.RegisterService(&DebugService{node}, "Debug")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
relayService := NewRelayService(node)
|
|
|
|
err = s.RegisterService(relayService, "Relay")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
err = s.RegisterService(&StoreService{node}, "Store")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
err = s.RegisterService(&AdminService{node}, "Admin")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
err = s.RegisterService(&FilterService{node}, "Filter")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
err = s.RegisterService(&PrivateService{node}, "Private")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/jsonrpc", func(w http.ResponseWriter, r *http.Request) {
|
|
t := time.Now()
|
|
s.ServeHTTP(w, r)
|
|
log.Infof("RPC request at %s took %s", r.URL.Path, time.Since(t))
|
|
})
|
|
|
|
listenAddr := fmt.Sprintf("%s:%d", address, port)
|
|
|
|
server := &http.Server{
|
|
Addr: listenAddr,
|
|
Handler: mux,
|
|
}
|
|
|
|
return &WakuRpc{
|
|
node: node,
|
|
server: server,
|
|
relayService: relayService,
|
|
}
|
|
}
|
|
|
|
func (r *WakuRpc) Start() {
|
|
go r.relayService.Start()
|
|
defer r.relayService.Stop()
|
|
log.Info("Rpc server started at ", r.server.Addr)
|
|
log.Info("server stopped ", r.server.ListenAndServe())
|
|
}
|
|
|
|
func (r *WakuRpc) Stop(ctx context.Context) error {
|
|
return r.server.Shutdown(ctx)
|
|
}
|