95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/coreos/go-raft"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
//-------------------------------------------------------------
|
|
// Handlers to handle raft related request via raft server port
|
|
//-------------------------------------------------------------
|
|
|
|
// Get all the current logs
|
|
func GetLogHttpHandler(w http.ResponseWriter, req *http.Request) {
|
|
debug("[recv] GET http://%v/log", raftServer.Name())
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(raftServer.LogEntries())
|
|
}
|
|
|
|
// Response to vote request
|
|
func VoteHttpHandler(w http.ResponseWriter, req *http.Request) {
|
|
rvreq := &raft.RequestVoteRequest{}
|
|
err := decodeJsonRequest(req, rvreq)
|
|
if err == nil {
|
|
debug("[recv] POST http://%v/vote [%s]", raftServer.Name(), rvreq.CandidateName)
|
|
if resp := raftServer.RequestVote(rvreq); resp != nil {
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(resp)
|
|
return
|
|
}
|
|
}
|
|
warn("[vote] ERROR: %v", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}
|
|
|
|
// Response to append entries request
|
|
func AppendEntriesHttpHandler(w http.ResponseWriter, req *http.Request) {
|
|
aereq := &raft.AppendEntriesRequest{}
|
|
err := decodeJsonRequest(req, aereq)
|
|
|
|
if err == nil {
|
|
debug("[recv] POST http://%s/log/append [%d]", raftServer.Name(), len(aereq.Entries))
|
|
if resp := raftServer.AppendEntries(aereq); resp != nil {
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(resp)
|
|
if !resp.Success {
|
|
debug("[Append Entry] Step back")
|
|
}
|
|
return
|
|
}
|
|
}
|
|
warn("[Append Entry] ERROR: %v", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}
|
|
|
|
// Response to recover from snapshot request
|
|
func SnapshotHttpHandler(w http.ResponseWriter, req *http.Request) {
|
|
aereq := &raft.SnapshotRequest{}
|
|
err := decodeJsonRequest(req, aereq)
|
|
if err == nil {
|
|
debug("[recv] POST http://%s/snapshot/ ", raftServer.Name())
|
|
if resp, _ := raftServer.SnapshotRecovery(aereq); resp != nil {
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(resp)
|
|
return
|
|
}
|
|
}
|
|
warn("[Snapshot] ERROR: %v", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}
|
|
|
|
// Get the port that listening for client connecting of the server
|
|
func ClientHttpHandler(w http.ResponseWriter, req *http.Request) {
|
|
debug("[recv] Get http://%v/client/ ", raftServer.Name())
|
|
w.WriteHeader(http.StatusOK)
|
|
client := address + ":" + strconv.Itoa(clientPort)
|
|
w.Write([]byte(client))
|
|
}
|
|
|
|
// Response to the join request
|
|
func JoinHttpHandler(w http.ResponseWriter, req *http.Request) {
|
|
|
|
command := &JoinCommand{}
|
|
|
|
if err := decodeJsonRequest(req, command); err == nil {
|
|
debug("Receive Join Request from %s", command.Name)
|
|
dispatch(command, &w, req, false)
|
|
} else {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|