rpi/rcond
1
0
Fork 0
mirror of https://github.com/0x1d/rcond.git synced 2025-12-14 18:25:21 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add cluster join and leave handlers

Browse Source
This commit is contained in:
0x1d
2025-05-17 16:51:25 +02:00
parent 49014c951f
commit de8af61ba7
5 changed files with 115 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
pkg/http/handlers.go
View File

@@ -5,6 +5,7 @@ import (
"encoding/json"
"log"
"net/http"
"strings"
"github.com/0x1d/rcond/pkg/cluster"
network "github.com/0x1d/rcond/pkg/network"
@@ -289,12 +290,49 @@ func HandleShutdown(w http.ResponseWriter, r *http.Request) {
json.NewEncoder(w).Encode(map[string]string{"status": "success"})
}
func ClusterAgentWrapper(agent *cluster.Agent) func(http.ResponseWriter, *http.Request) {
func ClusterAgentHandler(agent *cluster.Agent, handler func(http.ResponseWriter, *http.Request, *cluster.Agent)) func(http.ResponseWriter, *http.Request) {
return func(w http.ResponseWriter, r *http.Request) {
HandleClusterMembers(w, r, agent)
handler(w, r, agent)
}
}
func HandleClusterJoin(w http.ResponseWriter, r *http.Request, agent *cluster.Agent) {
var joinRequest struct {
Join []string `json:"join"`
}
err := json.NewDecoder(r.Body).Decode(&joinRequest)
if err != nil {
writeError(w, "Invalid request body", http.StatusBadRequest)
return
}
joinAddrs := strings.Join(joinRequest.Join, ",")
if joinAddrs == "" {
writeError(w, "No join addresses provided", http.StatusBadRequest)
return
}
addrs := strings.Split(joinAddrs, ",")
n, err := agent.Join(addrs, true)
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(map[string]int{"joined": n})
}
func HandleClusterLeave(w http.ResponseWriter, r *http.Request, agent *cluster.Agent) {
err := agent.Leave()
if err != nil {
writeError(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(map[string]string{"status": "success"})
}
func HandleClusterMembers(w http.ResponseWriter, r *http.Request, agent *cluster.Agent) {
if agent == nil {
writeError(w, "cluster agent is not initialized", http.StatusInternalServerError)

4
pkg/http/server.go
View File

@@ -76,7 +76,9 @@ func (s *Server) RegisterRoutes() {
s.router.HandleFunc("/users/{user}/keys/{fingerprint}", s.verifyToken(HandleRemoveAuthorizedKey)).Methods(http.MethodDelete)
s.router.HandleFunc("/system/restart", s.verifyToken(HandleReboot)).Methods(http.MethodPost)
s.router.HandleFunc("/system/shutdown", s.verifyToken(HandleShutdown)).Methods(http.MethodPost)
s.router.HandleFunc("/cluster/members", s.verifyToken(ClusterAgentWrapper(s.clusterAgent))).Methods(http.MethodGet)
s.router.HandleFunc("/cluster/members", s.verifyToken(ClusterAgentHandler(s.clusterAgent, HandleClusterMembers))).Methods(http.MethodGet)
s.router.HandleFunc("/cluster/join", s.verifyToken(ClusterAgentHandler(s.clusterAgent, HandleClusterJoin))).Methods(http.MethodPost)
s.router.HandleFunc("/cluster/leave", s.verifyToken(ClusterAgentHandler(s.clusterAgent, HandleClusterLeave))).Methods(http.MethodPost)
}
func (s *Server) healthHandler(w http.ResponseWriter, r *http.Request) {