示例#1
0
文件: list.go 项目: abronan/proton
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))
	}
}
示例#2
0
文件: members.go 项目: abronan/proton
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)
	}
}
示例#3
0
文件: put.go 项目: abronan/proton
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")
	}
}
示例#4
0
文件: join.go 项目: abronan/proton
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 {}
}