lotus/cmd/lotus-storage-miner/run.go

127 lines
3.0 KiB
Go
Raw Normal View History

2019-07-18 23:16:23 +00:00
package main
import (
2019-09-17 14:23:08 +00:00
"context"
2019-07-19 09:24:11 +00:00
"net/http"
"os"
2019-09-17 14:23:08 +00:00
"os/signal"
"syscall"
2019-07-19 09:24:11 +00:00
2019-10-03 18:12:30 +00:00
"github.com/filecoin-project/go-lotus/build"
2019-07-19 09:24:11 +00:00
"github.com/multiformats/go-multiaddr"
"golang.org/x/xerrors"
2019-07-18 23:16:23 +00:00
"gopkg.in/urfave/cli.v2"
2019-07-24 00:58:31 +00:00
"github.com/filecoin-project/go-lotus/api"
2019-07-18 23:16:23 +00:00
lcli "github.com/filecoin-project/go-lotus/cli"
2019-07-24 17:09:00 +00:00
"github.com/filecoin-project/go-lotus/lib/auth"
2019-07-19 09:24:11 +00:00
"github.com/filecoin-project/go-lotus/lib/jsonrpc"
2019-07-27 01:54:03 +00:00
"github.com/filecoin-project/go-lotus/lib/sectorbuilder"
2019-07-19 09:24:11 +00:00
"github.com/filecoin-project/go-lotus/node"
2019-07-27 01:54:03 +00:00
"github.com/filecoin-project/go-lotus/node/modules"
2019-07-19 09:24:11 +00:00
"github.com/filecoin-project/go-lotus/node/repo"
2019-07-18 23:16:23 +00:00
)
var runCmd = &cli.Command{
2019-07-18 23:16:23 +00:00
Name: "run",
Usage: "Start a lotus storage miner process",
Flags: []cli.Flag{
2019-07-19 09:24:11 +00:00
&cli.StringFlag{
Name: "api",
Value: "2345",
},
},
2019-07-18 23:16:23 +00:00
Action: func(cctx *cli.Context) error {
2019-10-02 17:20:30 +00:00
if err := build.GetParams(true); err != nil {
return xerrors.Errorf("fetching proof parameters: %w", err)
}
2019-10-03 18:12:30 +00:00
nodeApi, ncloser, err := lcli.GetFullNodeAPI(cctx)
2019-07-18 23:16:23 +00:00
if err != nil {
return err
}
2019-10-03 18:12:30 +00:00
defer ncloser()
2019-09-17 18:36:06 +00:00
ctx := lcli.DaemonContext(cctx)
2019-07-18 23:16:23 +00:00
2019-07-24 00:58:31 +00:00
v, err := nodeApi.Version(ctx)
2019-07-24 11:20:00 +00:00
if err != nil {
return err
}
2019-07-18 23:16:23 +00:00
2019-07-27 01:54:03 +00:00
storageRepoPath := cctx.String(FlagStorageRepo)
r, err := repo.NewFS(storageRepoPath)
2019-07-19 09:24:11 +00:00
if err != nil {
return err
}
2019-07-23 21:54:54 +00:00
ok, err := r.Exists()
if err != nil {
return err
}
if !ok {
2019-07-27 01:54:03 +00:00
return xerrors.Errorf("repo at '%s' is not initialized, run 'lotus-storage-miner init' to set it up", storageRepoPath)
2019-07-19 09:24:11 +00:00
}
2019-07-24 00:58:31 +00:00
var minerapi api.StorageMiner
2019-09-17 14:23:08 +00:00
stop, err := node.New(ctx,
2019-07-24 00:58:31 +00:00
node.StorageMiner(&minerapi),
2019-07-19 09:24:11 +00:00
node.Online(),
node.Repo(r),
node.Override(node.SetApiEndpointKey, func(lr repo.LockedRepo) error {
apima, err := multiaddr.NewMultiaddr("/ip4/127.0.0.1/tcp/" + cctx.String("api"))
if err != nil {
return err
}
return lr.SetAPIEndpoint(apima)
}),
2019-07-27 01:54:03 +00:00
node.Override(new(*sectorbuilder.SectorBuilderConfig), modules.SectorBuilderConfig(storageRepoPath)),
node.Override(new(api.FullNode), nodeApi),
2019-07-19 09:24:11 +00:00
)
if err != nil {
return err
}
2019-08-02 16:37:42 +00:00
// Bootstrap with full node
remoteAddrs, err := nodeApi.NetAddrsListen(ctx)
if err != nil {
return err
}
if err := minerapi.NetConnect(ctx, remoteAddrs); err != nil {
return err
}
2019-07-18 23:16:23 +00:00
log.Infof("Remote version %s", v)
2019-07-19 09:24:11 +00:00
rpcServer := jsonrpc.NewServer()
2019-07-24 01:10:26 +00:00
rpcServer.Register("Filecoin", api.PermissionedStorMinerAPI(minerapi))
2019-07-24 17:09:00 +00:00
ah := &auth.Handler{
Verify: minerapi.AuthVerify,
2019-07-25 12:54:19 +00:00
Next: rpcServer.ServeHTTP,
2019-07-24 17:09:00 +00:00
}
http.Handle("/rpc/v0", ah)
2019-09-17 14:23:08 +00:00
srv := &http.Server{Addr: "127.0.0.1:" + cctx.String("api"), Handler: http.DefaultServeMux}
sigChan := make(chan os.Signal, 2)
go func() {
<-sigChan
log.Warn("Shutting down..")
if err := stop(context.TODO()); err != nil {
log.Errorf("graceful shutting down failed: %s", err)
}
if err := srv.Shutdown(context.TODO()); err != nil {
log.Errorf("shutting down RPC server failed: %s", err)
}
log.Warn("Graceful shutdown successful")
}()
signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT)
return srv.ListenAndServe()
2019-07-18 23:16:23 +00:00
},
}