raftlib use

raftsample.go

package main

import (
  "fmt"
  "net"
  "io"
  "github.com/hashicorp/raft"
  "time"
  "os"
  "flag"
)

func main() {
  fmt.Println("start")

  myselfptr := flag.String("m", "127.0.0.1:7777", "myself")
  flag.Parse()
  me := *myselfptr

  conf := raft.DefaultConfig()
  conf.LocalID = raft.ServerID(me)
  inmemstore := raft.NewInmemStore()
  snapstore := raft.NewInmemSnapshotStore()


  ip, _ := net.ResolveTCPAddr("tcp", me)
  fmt.Println(ip)
  trans, err := raft.NewTCPTransport(me, ip, 10, 2*time.Second, os.Stdout)
  if err != nil {
    fmt.Println(err)
  }


  c := []raft.Server{
    raft.Server{
      Suffrage: raft.Voter,
      ID: raft.ServerID("127.0.0.1:7777"),
      Address: raft.ServerAddress("127.0.0.1:7777"),
    },
    raft.Server{
      Suffrage: raft.Voter,
      ID: raft.ServerID("127.0.0.1:7778"),
      Address: raft.ServerAddress("127.0.0.1:7778"),
    },
    raft.Server{
      Suffrage: raft.Voter,
      ID: raft.ServerID("127.0.0.1:7779"),
      Address: raft.ServerAddress("127.0.0.1:7779"),
    },
  }
  config := raft.Configuration{Servers: c}
  err = raft.BootstrapCluster(conf, inmemstore, inmemstore, snapstore, trans, config)
  fmt.Println(err)

  raf, err := raft.NewRaft(conf, nil, inmemstore, inmemstore, snapstore, trans)

  t := time.NewTicker(time.Duration(1) * time.Second)
  for {
    select {
    case <-t.C:
      fmt.Println(raf.Leader())
    }
  }
}