Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 24 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
Created .gitignore file for flag type go.
# If you prefer the allow list template instead of the deny list, see community template:
# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
#
# Binaries for programs and plugins
*.exe
*.exe~
*.dll
*.so
*.dylib

# Test binary, built with `go test -c`
*.test

# Output of the go coverage tool, specifically when used with LiteIDE
*.out

# Dependency directories (remove the comment below to include it)
# vendor/

# Go workspace file
go.work
bin
*.log
29 changes: 29 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
.PHONY: all

all: raft-server client

update-proto:
protoc --proto_path=./proto --go_out=proto/raft --go_opt=paths=source_relative raft.proto
protoc --proto_path=./proto --go_out=proto/server --go_opt=paths=source_relative server.proto

raft-server:
go build -o bin/raft cmd/server/main.go

client:
go build -o bin/client cmd/client/main.go

run-3-server:
echo "running on port 8000" && ./bin/raft -port=8000 -log_path=server1.log &
echo "running on port 8001" && ./bin/raft -port=8001 -log_path=server2.log &
echo "running on port 8002" && ./bin/raft -port=8002 -log_path=server3.log &

run-5-server:
echo "running on port 8000" && ./bin/raft -port=8000 -log_path=server1.log &
echo "running on port 8001" && ./bin/raft -port=8001 -log_path=server2.log &
echo "running on port 8002" && ./bin/raft -port=8002 -log_path=server3.log &
echo "running on port 8003" && ./bin/raft -port=8003 -log_path=server4.log &
echo "running on port 8004" && ./bin/raft -port=8004 -log_path=server5.log &

clean:
rm -rf bin
rm *.log
44 changes: 44 additions & 0 deletions cmd/client/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
package main

import (
"flag"
"fmt"

raftPb "github.com/adylanrff/raft-algorithm/proto/raft"
serverPb "github.com/adylanrff/raft-algorithm/proto/server"
"github.com/adylanrff/raft-algorithm/raft"
"github.com/adylanrff/raft-algorithm/rpc"
"github.com/adylanrff/raft-algorithm/util"
)

var port int
var logPath string

func init() {
flag.IntVar(&port, "port", 8000, "server target port")
flag.StringVar(&logPath, "log_path", "server.log", "log path")

flag.Parse()
}

func main() {
util.InitLogger(logPath)

raftMsg := rpc.ServerMessageDTO{
ServerMessage: &serverPb.ServerMessage{
Method: raft.RaftMethodName_RequestVotes,
Payload: &serverPb.ServerMessage_ServerRequest{
ServerRequest: &serverPb.ServerRequest{
Request: &serverPb.ServerRequest_RequestVoteRequest{
RequestVoteRequest: &raftPb.RequestVoteRequest{},
},
},
},
},
}

rpcClient := rpc.NewRPCClient()
resp, err := rpcClient.Call(fmt.Sprintf("0.0.0.0:%d", port), &raftMsg)

fmt.Printf("resp=%+v,err=%v", resp, err)
}
67 changes: 67 additions & 0 deletions cmd/server/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
package main

import (
"flag"
"fmt"
"math/rand"
"sync"
"time"

"github.com/adylanrff/raft-algorithm/raft"
"github.com/adylanrff/raft-algorithm/rpc"
"github.com/adylanrff/raft-algorithm/util"
log "github.com/sirupsen/logrus"
)

var port int
var logPath string

func init() {
flag.IntVar(&port, "port", 8000, "server port")
flag.StringVar(&logPath, "log_path", "server.log", "log path")

flag.Parse()
rand.Seed(time.Now().UnixNano())
}

func main() {
util.InitLogger(logPath)

address := fmt.Sprintf("127.0.0.1:%d", port)
serverID := raft.GetRaftServerIDFromAddress(address)

raftHandler := raft.NewRaft(serverID, address, raft.RaftConfig{
ElectionTimeout: 1000 * time.Millisecond,
IdleTimeout: 500 * time.Millisecond,
ClusterMemberAddreses: []string{
"127.0.0.1:8000",
"127.0.0.1:8001",
"127.0.0.1:8002",
"127.0.0.1:8003",
"127.0.0.1:8004",
},
})

raftServerHandler := raft.NewRaftServerhandler(raftHandler)

server := rpc.NewServer(address)
server.AddHandler(raft.RaftMethodName_RequestVotes, raftServerHandler.RequestVoteHandler)
server.AddHandler(raft.RaftMethodName_AppendEntries, raftServerHandler.AppendEntriesHandler)

var wg sync.WaitGroup

wg.Add(2)
go func() {
defer wg.Done()
server.Run()
}()

go func() {
defer wg.Done()
raftHandler.Run()
}()

wg.Wait()

log.Debug("server terminated")
}
9 changes: 9 additions & 0 deletions go.mod
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
module github.com/adylanrff/raft-algorithm

go 1.19

require (
github.com/sirupsen/logrus v1.9.0 // indirect
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 // indirect
google.golang.org/protobuf v1.28.1 // indirect
)
17 changes: 17 additions & 0 deletions go.sum
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/sirupsen/logrus v1.9.0 h1:trlNQbNUG3OdDrDil03MCb1H2o9nJ1x4/5LYw7byDE0=
github.com/sirupsen/logrus v1.9.0/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 h1:0A+M6Uqn+Eje4kHMK80dtF3JCXC4ykBgQG4Fe06QRhQ=
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w=
google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
26 changes: 26 additions & 0 deletions proto/raft.proto
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
syntax = "proto3";
option go_package = "github.com/adylanrff/raft-algorithm/proto/raft";

// RequestVote Method
message RequestVoteRequest {
int32 term = 1;
string candidate_id = 2;
int32 last_log_index = 3;
int32 last_log_term = 4;
}

message RequestVoteResponse {
int32 term = 1;
bool vote_granted = 2;
}

// AppendEntries Method
message AppendEntriesRequest {
int32 term = 1;
string leader_id = 2;
}

message AppendEntriesResponse {
int32 term = 1;
bool success = 2;
}
Loading