45 lines
919 B
Go
45 lines
919 B
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"syscall"
|
|
|
|
"github.com/lbryio/lbry.go/crypto"
|
|
"github.com/lbryio/reflector.go/cluster"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func init() {
|
|
var cmd = &cobra.Command{
|
|
Use: "cluster [start|join]",
|
|
Short: "Connect to cluster",
|
|
ValidArgs: []string{"start", "join"},
|
|
Args: argFuncs(cobra.ExactArgs(1), cobra.OnlyValidArgs),
|
|
Run: clusterCmd,
|
|
}
|
|
RootCmd.AddCommand(cmd)
|
|
}
|
|
|
|
func clusterCmd(cmd *cobra.Command, args []string) {
|
|
port := 17946
|
|
var c *cluster.Cluster
|
|
if args[0] == "start" {
|
|
c = cluster.New(port, "")
|
|
} else {
|
|
c = cluster.New(port+1+int(crypto.RandInt64(1000)), "127.0.0.1:"+strconv.Itoa(port))
|
|
}
|
|
|
|
err := c.Connect()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
interruptChan := make(chan os.Signal, 1)
|
|
signal.Notify(interruptChan, os.Interrupt, syscall.SIGTERM)
|
|
<-interruptChan
|
|
c.Shutdown()
|
|
}
|