main.go

package main

import (
	"fmt"
	"log"
	"os"
	"os/signal"
	"raft/client"
	"raft/fsm"
	r "raft/node"
	pb "raft/protos"
	"syscall"
	"time"
)

func main() {
	// There are four servers but first 3 don't know about the fourth one, should be added with addServer RPC
	config := &pb.Configuration{
		Members: map[string]string{
			"1": "localhost:8000",
			"2": "localhost:8005",
			"3": "localhost:8021",
		},
		LogIndex: -1,
	}
	raft1, err := r.InitRaftNode("1", "localhost:8000", config, fsm.NewFSMManager("1"))
	if err != nil {
		panic(err)
	}
	raft2, err := r.InitRaftNode("2", "localhost:8005", config, fsm.NewFSMManager("2"))
	if err != nil {
		panic(err)
	}
	raft3, err := r.InitRaftNode("3", "localhost:8021", config, fsm.NewFSMManager("3"))
	if err != nil {
		panic(err)
	}

	configNew := &pb.Configuration{
		Members: map[string]string{
			"1": "localhost:8000",
			"2": "localhost:8005",
			"3": "localhost:8021",
			"4": "localhost:8023",
			"5": "localhost:8024",
		},
		LogIndex: -1,
	}

	raft4, err := r.InitRaftNode("4", "localhost:8023", configNew, fsm.NewFSMManager("4"))
	if err != nil {
		panic(err)
	}

	raft5, err := r.InitRaftNode("5", "localhost:8024", configNew, fsm.NewFSMManager("5"))
	if err != nil {
		panic(err)
	}

	err = raft1.Start()
	err = raft2.Start()
	err = raft3.Start()
	err = raft4.Start()
	err = raft5.Start()

	// Client 1
	cl, err := client.NewRaftClient("client1", map[string]string{
		"1": "localhost:8000",
		"2": "localhost:8005",
		"3": "localhost:8021",
		"4": "localhost:8023",
		"5": "localhost:8024",
	})

	if err != nil {
		fmt.Println(err)
	}

	go func() {
		client.SimulateClientCommands(cl)
		<-time.After(5 * time.Second)
		log.Println("Shutting down the leader")
		raftNodes := map[string]*r.RaftNode{
			"1": raft1,
			"2": raft2,
			"3": raft3,
			"4": raft4,
			"5": raft5,
		}

		for id := range configNew.GetMembers() {
			if node, exists := raftNodes[id]; exists {
				if node.GetState().String() == "Leader" {
					node.Shutdown()
					break
				}
			}
		}
		<-time.After(10 * time.Second)
		client.SimulateClientCommands(cl)
	}()

	quitCh := make(chan os.Signal, 1)
	signal.Notify(quitCh, syscall.SIGINT, syscall.SIGTERM)
	<-quitCh
}