func list(c *cli.Context) {
var (
err error
)
hosts := c.StringSlice("host")
if c.IsSet("host") || c.IsSet("H") {
hosts = hosts[1:]
}
client, err := proton.GetRaftClient(hosts[0], 2*time.Second)
if err != nil {
log.Fatal("couldn't initialize client connection")
}
resp, err := client.ListObjects(context.TODO(), &proton.ListObjectsRequest{})
if err != nil {
log.Fatal("Can't list objects in the cluster")
}
fmt.Println("Keys:")
for _, obj := range resp.Objects {
fmt.Println(":", obj.Key, ":", string(obj.Value))
}
}
func members(c *cli.Context) {
var (
err error
)
hosts := c.StringSlice("host")
if c.IsSet("host") || c.IsSet("H") {
hosts = hosts[1:]
}
client, err := proton.GetRaftClient(hosts[0], 2*time.Second)
if err != nil {
log.Fatal("couldn't initialize client connection")
}
resp, err := client.ListMembers(context.TODO(), &proton.ListMembersRequest{})
if err != nil {
log.Fatal("Can't list members in the cluster")
}
fmt.Println("Nodes:")
for _, node := range resp.Members {
fmt.Println(":", node.ID)
}
}
func put(c *cli.Context) {
var (
err error
)
hosts := c.StringSlice("host")
if c.IsSet("host") || c.IsSet("H") {
hosts = hosts[1:]
}
key := c.String("key")
if key == "" {
log.Fatal("key flag must be set")
}
value := []byte(c.String("value"))
if c.String("value") == "" {
log.Fatal("value flag must be set")
}
client, err := proton.GetRaftClient(hosts[0], 2*time.Second)
if err != nil {
log.Fatal("couldn't initialize client connection")
}
resp, err := client.PutObject(context.TODO(), &proton.PutObjectRequest{Object: &proton.Pair{Key: key, Value: value}})
if resp == nil || err != nil {
log.Fatal("Can't put object in the cluster")
}
}
func join(c *cli.Context) {
var (
err error
)
hosts := c.StringSlice("host")
if c.IsSet("host") || c.IsSet("H") {
hosts = hosts[1:]
}
lis, err := net.Listen("tcp", hosts[0])
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
server := grpc.NewServer()
joinAddr := c.String("join")
hostname := c.String("hostname")
if c.Bool("withRaftLogs") {
raftLogger = &raft.DefaultLogger{Logger: log.New(ioutil.Discard, "", 0)}
}
id := proton.GenID(hostname)
cfg := proton.DefaultNodeConfig()
cfg.Logger = raftLogger
node, err := proton.NewNode(id, hosts[0], cfg, handler)
if err != nil {
log.Fatal("Can't initialize raft node")
}
proton.Register(server, node)
client, err := proton.GetRaftClient(joinAddr, 2*time.Second)
if err != nil {
log.Fatal("couldn't initialize client connection")
}
// Start raft
go node.Start()
go server.Serve(lis)
info := &proton.NodeInfo{
ID: id,
Addr: hosts[0],
}
resp, err := client.JoinRaft(context.Background(), info)
if err != nil {
log.Fatalf("could not join: %v", err)
}
err = node.RegisterNodes(resp.GetNodes())
if err != nil {
log.Fatal(err)
}
ticker := time.NewTicker(time.Second * 5)
go func() {
for _ = range ticker.C {
fmt.Println("----- Leader is: ", node.Leader())
}
}()
select {}
}