main.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package main
  2. import (
  3. "log"
  4. "os"
  5. "igit.com/xbase/raft"
  6. "igit.com/xbase/raft/example/basic/common"
  7. )
  8. func main() {
  9. // Configuration
  10. nodeID := "node2"
  11. addr := "127.0.0.1:9002"
  12. dataDir := "../data/node2"
  13. // Initial Cluster configuration (Node 1 + Node 2)
  14. clusterNodes := map[string]string{
  15. "node1": "127.0.0.1:9001",
  16. "node2": "127.0.0.1:9002",
  17. }
  18. config := raft.DefaultConfig()
  19. config.NodeID = nodeID
  20. config.ListenAddr = addr
  21. config.DataDir = dataDir
  22. config.ClusterNodes = clusterNodes
  23. config.LogCompactionEnabled = false
  24. config.Logger = raft.NewConsoleLogger(nodeID, 1)
  25. if err := os.MkdirAll(dataDir, 0755); err != nil {
  26. log.Fatalf("Failed to create data directory: %v", err)
  27. }
  28. server, err := raft.NewKVServer(config)
  29. if err != nil {
  30. log.Fatalf("Failed to create server: %v", err)
  31. }
  32. // Register demo commands
  33. common.RegisterDemoCommands(server.CLI)
  34. if err := server.Start(); err != nil {
  35. log.Fatalf("Failed to start server: %v", err)
  36. }
  37. defer server.Stop()
  38. // Keep main thread running
  39. select {}
  40. }