main.go

package main

import (
	"fmt"
	"sync"
	"time"

	"github.com/josephjang/single-decree-paxos-go/paxos"

	p "github.com/josephjang/single-decree-paxos-go/paxos"
	log "github.com/sirupsen/logrus"
	prefixed "github.com/x-cray/logrus-prefixed-formatter"
)

const channelBufferSize = 1024

func init() {
	log.SetFormatter(&prefixed.TextFormatter{})
	log.SetLevel(log.InfoLevel)
}

// send prepare requests to acceptors
func proposer(name string, num int64, val int64,
	prepareReqChans []chan p.PrepareRequest,
	acceptReqChans []chan p.AcceptRequest,
	wg *sync.WaitGroup) {
	p := paxos.NewProposer(name, num, val, prepareReqChans, acceptReqChans, wg)
	p.Run()
}

var shutdown = false

// process prepare/accept requests
func acceptor(name string, prepareReqChan chan p.PrepareRequest, acceptReqChan chan p.AcceptRequest, wg *sync.WaitGroup) {
	a := paxos.NewAcceptor(name, prepareReqChan, acceptReqChan, wg)
	a.Run()
}

func main() {
	// channels for acceptors
	var prepareReqChans = []chan p.PrepareRequest{
		make(chan p.PrepareRequest, channelBufferSize),
		make(chan p.PrepareRequest, channelBufferSize),
		make(chan p.PrepareRequest, channelBufferSize),
	}

	var acceptReqChans = []chan p.AcceptRequest{
		make(chan p.AcceptRequest, channelBufferSize),
		make(chan p.AcceptRequest, channelBufferSize),
		make(chan p.AcceptRequest, channelBufferSize),
	}

	// channels for proposers

	var proposerWaitGroup sync.WaitGroup
	var acceptorWaitGroup sync.WaitGroup

	for i := range prepareReqChans {
		acceptorWaitGroup.Add(1)
		go acceptor(fmt.Sprintf("A%d", i+1), prepareReqChans[i], acceptReqChans[i], &acceptorWaitGroup)
	}

	proposerWaitGroup.Add(3)
	go proposer("P1", 1, 41, prepareReqChans, acceptReqChans, &proposerWaitGroup)
	go proposer("P2", 2, 42, prepareReqChans, acceptReqChans, &proposerWaitGroup)

	time.Sleep(1 * time.Second)
	go proposer("P3", 3, 10, prepareReqChans, acceptReqChans, &proposerWaitGroup)

	proposerWaitGroup.Wait()
	log.Infof("System: All proposers are finished")
	acceptorWaitGroup.Wait()
	log.Infof("System: All acceptors are finished")
}