Pulled most of gaiad into server, with rough implementation

This commit is contained in:
Ethan Frey
2018-03-01 02:30:13 +00:00
committed by rigelrozanski
parent a99d913982
commit 438d18e059
5 changed files with 155 additions and 58 deletions
+31
View File
@@ -0,0 +1,31 @@
package server
import (
"github.com/spf13/cobra"
tcmd "github.com/tendermint/tendermint/cmd/tendermint/commands"
"github.com/tendermint/tmlibs/log"
)
// UnsafeResetAllCmd - extension of the tendermint command, resets initialization
func UnsafeResetAllCmd(logger log.Logger) *cobra.Command {
cmd := resetAll{logger}
return &cobra.Command{
Use: "unsafe_reset_all",
Short: "Reset all blockchain data",
RunE: cmd.run,
}
}
type resetAll struct {
logger log.Logger
}
func (r resetAll) run(cmd *cobra.Command, args []string) error {
cfg, err := tcmd.ParseConfig()
if err != nil {
return err
}
tcmd.ResetAll(cfg.DBDir(), cfg.PrivValidatorFile(), r.logger)
return nil
}
+107
View File
@@ -0,0 +1,107 @@
package server
import (
"github.com/pkg/errors"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"github.com/tendermint/abci/server"
tcmd "github.com/tendermint/tendermint/cmd/tendermint/commands"
"github.com/tendermint/tendermint/node"
"github.com/tendermint/tendermint/proxy"
"github.com/tendermint/tendermint/types"
cmn "github.com/tendermint/tmlibs/common"
"github.com/cosmos/cosmos-sdk/baseapp"
)
const (
flagWithTendermint = "with-tendermint"
flagAddress = "address"
)
// StartNodeCmd runs the service passed in, either
// stand-alone, or in-process with tendermint
func StartNodeCmd(app *baseapp.BaseApp) *cobra.Command {
start := startNodeCmd{
app: app,
}
cmd := &cobra.Command{
Use: "start",
Short: "Run the full node",
RunE: start.run,
}
// basic flags for abci app
cmd.Flags().Bool(flagWithTendermint, true, "run abci app embedded in-process with tendermint")
cmd.Flags().String(flagAddress, "tcp://0.0.0.0:46658", "Listen address")
// AddNodeFlags adds support for all
// tendermint-specific command line options
tcmd.AddNodeFlags(cmd)
return cmd
}
type startNodeCmd struct {
// do this in main:
// rootDir := viper.GetString(cli.HomeFlag)
// node.Logger = ....
app *baseapp.BaseApp
}
func (s startNodeCmd) run(cmd *cobra.Command, args []string) error {
logger := s.app.Logger
if !viper.GetBool(flagWithTendermint) {
logger.Info("Starting ABCI without Tendermint")
return s.startStandAlone()
}
logger.Info("Starting ABCI with Tendermint")
return s.startInProcess()
}
func (s startNodeCmd) startStandAlone() error {
logger := s.app.Logger
// Start the ABCI listener
addr := viper.GetString(flagAddress)
svr, err := server.NewServer(addr, "socket", s.app)
if err != nil {
return errors.Errorf("Error creating listener: %v\n", err)
}
svr.SetLogger(logger.With("module", "abci-server"))
svr.Start()
// Wait forever
cmn.TrapSignal(func() {
// Cleanup
svr.Stop()
})
return nil
}
func (s startNodeCmd) startInProcess() error {
logger := s.app.Logger
cfg, err := tcmd.ParseConfig()
if err != nil {
return err
}
// Create & start tendermint node
n, err := node.NewNode(cfg,
types.LoadOrGenPrivValidatorFS(cfg.PrivValidatorFile()),
proxy.NewLocalClientCreator(s.app),
node.DefaultGenesisDocProviderFunc(cfg),
node.DefaultDBProvider,
logger.With("module", "node"))
if err != nil {
return err
}
err = n.Start()
if err != nil {
return err
}
// Trap signal, run forever.
n.RunForever()
return nil
}