157 lines
3.6 KiB
Go
157 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
|
|
"contrib.go.opencensus.io/exporter/prometheus"
|
|
"github.com/filecoin-project/go-jsonrpc"
|
|
promclient "github.com/prometheus/client_golang/prometheus"
|
|
"go.opencensus.io/tag"
|
|
|
|
lapi "github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/api/v0api"
|
|
"github.com/filecoin-project/lotus/api/v1api"
|
|
"github.com/filecoin-project/lotus/build"
|
|
lcli "github.com/filecoin-project/lotus/cli"
|
|
"github.com/filecoin-project/lotus/lib/lotuslog"
|
|
"github.com/filecoin-project/lotus/metrics"
|
|
|
|
logging "github.com/ipfs/go-log/v2"
|
|
"go.opencensus.io/stats/view"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var log = logging.Logger("gateway")
|
|
|
|
func main() {
|
|
lotuslog.SetupLogLevels()
|
|
|
|
local := []*cli.Command{
|
|
runCmd,
|
|
}
|
|
|
|
app := &cli.App{
|
|
Name: "lotus-gateway",
|
|
Usage: "Public API server for lotus",
|
|
Version: build.UserVersion(),
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "repo",
|
|
EnvVars: []string{"LOTUS_PATH"},
|
|
Value: "~/.lotus", // TODO: Consider XDG_DATA_HOME
|
|
},
|
|
},
|
|
|
|
Commands: local,
|
|
}
|
|
app.Setup()
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
|
log.Warnf("%+v", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
var runCmd = &cli.Command{
|
|
Name: "run",
|
|
Usage: "Start api server",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "listen",
|
|
Usage: "host address and port the api server will listen on",
|
|
Value: "0.0.0.0:2346",
|
|
},
|
|
&cli.IntFlag{
|
|
Name: "api-max-req-size",
|
|
Usage: "maximum API request size accepted by the JSON RPC server",
|
|
},
|
|
},
|
|
Action: func(cctx *cli.Context) error {
|
|
log.Info("Starting lotus gateway")
|
|
|
|
ctx := lcli.ReqContext(cctx)
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
defer cancel()
|
|
|
|
// Register all metric views
|
|
if err := view.Register(
|
|
metrics.ChainNodeViews...,
|
|
); err != nil {
|
|
log.Fatalf("Cannot register the view: %v", err)
|
|
}
|
|
|
|
api, closer, err := lcli.GetFullNodeAPIV1(cctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer closer()
|
|
|
|
address := cctx.String("listen")
|
|
mux := mux.NewRouter()
|
|
|
|
log.Info("Setting up API endpoint at " + address)
|
|
|
|
serveRpc := func(path string, hnd interface{}) {
|
|
serverOptions := make([]jsonrpc.ServerOption, 0)
|
|
if maxRequestSize := cctx.Int("api-max-req-size"); maxRequestSize != 0 {
|
|
serverOptions = append(serverOptions, jsonrpc.WithMaxRequestSize(int64(maxRequestSize)))
|
|
}
|
|
rpcServer := jsonrpc.NewServer(serverOptions...)
|
|
rpcServer.Register("Filecoin", hnd)
|
|
|
|
mux.Handle(path, rpcServer)
|
|
}
|
|
|
|
ma := metrics.MetricedGatewayAPI(NewGatewayAPI(api))
|
|
|
|
serveRpc("/rpc/v1", ma)
|
|
serveRpc("/rpc/v0", lapi.Wrap(new(v1api.FullNodeStruct), new(v0api.WrapperV1Full), ma))
|
|
|
|
registry := promclient.DefaultRegisterer.(*promclient.Registry)
|
|
exporter, err := prometheus.NewExporter(prometheus.Options{
|
|
Registry: registry,
|
|
Namespace: "lotus_gw",
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mux.Handle("/debug/metrics", exporter)
|
|
|
|
mux.PathPrefix("/").Handler(http.DefaultServeMux)
|
|
|
|
/*ah := &auth.Handler{
|
|
Verify: nodeApi.AuthVerify,
|
|
Next: mux.ServeHTTP,
|
|
}*/
|
|
|
|
srv := &http.Server{
|
|
Handler: mux,
|
|
BaseContext: func(listener net.Listener) context.Context {
|
|
ctx, _ := tag.New(context.Background(), tag.Upsert(metrics.APIInterface, "lotus-gateway"))
|
|
return ctx
|
|
},
|
|
}
|
|
|
|
go func() {
|
|
<-ctx.Done()
|
|
log.Warn("Shutting down...")
|
|
if err := srv.Shutdown(context.TODO()); err != nil {
|
|
log.Errorf("shutting down RPC server failed: %s", err)
|
|
}
|
|
log.Warn("Graceful shutdown successful")
|
|
}()
|
|
|
|
nl, err := net.Listen("tcp", address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return srv.Serve(nl)
|
|
},
|
|
}
|