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 %s/log", raftTransporter.scheme+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 %s/vote [%s]", raftTransporter.scheme+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 %s/log/append [%d]", raftTransporter.scheme+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 %s/snapshot/ ", raftTransporter.scheme+raftServer.Name()) if resp := raftServer.RequestSnapshot(aereq); resp != nil { w.WriteHeader(http.StatusOK) json.NewEncoder(w).Encode(resp) return } } warn("[Snapshot] ERROR: %v", err) w.WriteHeader(http.StatusInternalServerError) } // Response to recover from snapshot request func SnapshotRecoveryHttpHandler(w http.ResponseWriter, req *http.Request) { aereq := &raft.SnapshotRecoveryRequest{} err := decodeJsonRequest(req, aereq) if err == nil { debug("[recv] POST %s/snapshotRecovery/ ", raftTransporter.scheme+raftServer.Name()) if resp := raftServer.SnapshotRecoveryRequest(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 %s/client/ ", raftTransporter.scheme+raftServer.Name()) w.WriteHeader(http.StatusOK) client := hostname + ":" + 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 } }