main.go 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. package main
  2. import (
  3. "flag"
  4. "log"
  5. "net/http"
  6. "strconv"
  7. "time"
  8. "github.com/coreos/etcd/etcdserver"
  9. "github.com/coreos/etcd/etcdserver/etcdhttp"
  10. "github.com/coreos/etcd/raft"
  11. "github.com/coreos/etcd/raft/raftpb"
  12. "github.com/coreos/etcd/store"
  13. )
  14. var (
  15. fid = flag.String("id", "0xBEEF", "Id of this server")
  16. timeout = flag.Duration("timeout", 10*time.Second, "Request Timeout")
  17. laddr = flag.String("l", ":8080", "HTTP service address (e.g., ':8080')")
  18. peers = etcdhttp.Peers{}
  19. )
  20. func init() {
  21. peers.Set("0xBEEF=localhost:8080")
  22. flag.Var(peers, "peers", "your peers")
  23. }
  24. func main() {
  25. flag.Parse()
  26. id, err := strconv.ParseInt(*fid, 0, 64)
  27. if err != nil {
  28. log.Fatal(err)
  29. }
  30. if peers.Pick(id) == "" {
  31. log.Fatalf("%#x=<addr> must be specified in peers", id)
  32. }
  33. n := raft.Start(id, peers.Ids(), 10, 1)
  34. tk := time.NewTicker(100 * time.Millisecond)
  35. s := &etcdserver.Server{
  36. Store: store.New(),
  37. Node: n,
  38. Save: func(st raftpb.State, ents []raftpb.Entry) {}, // TODO: use wal
  39. Send: etcdhttp.Sender(peers),
  40. Ticker: tk.C,
  41. }
  42. etcdserver.Start(s)
  43. h := &etcdhttp.Handler{
  44. Timeout: *timeout,
  45. Server: s,
  46. }
  47. http.Handle("/", h)
  48. log.Fatal(http.ListenAndServe(*laddr, nil))
  49. }