123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182 |
- package controllers
- import (
- "encoding/json"
- "errors"
- "fmt"
- "net/http"
- "github.com/google/uuid"
- "github.com/gorilla/mux"
- controller "github.com/gravitl/netmaker/controllers"
- "github.com/gravitl/netmaker/logger"
- "github.com/gravitl/netmaker/logic"
- "github.com/gravitl/netmaker/models"
- "github.com/gravitl/netmaker/mq"
- proLogic "github.com/gravitl/netmaker/pro/logic"
- "golang.org/x/exp/slog"
- )
- // FailOverHandlers - handlers for FailOver
- func FailOverHandlers(r *mux.Router) {
- r.HandleFunc("/api/v1/node/{nodeid}/failover", logic.SecurityCheck(true, http.HandlerFunc(createfailOver))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/node/{nodeid}/failover", logic.SecurityCheck(true, http.HandlerFunc(deletefailOver))).Methods(http.MethodDelete)
- r.HandleFunc("/api/v1/node/{network}/failover/reset", logic.SecurityCheck(true, http.HandlerFunc(resetFailOver))).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/node/{nodeid}/failover_me", controller.Authorize(true, false, "host", http.HandlerFunc(failOverME))).Methods(http.MethodPost)
- }
- // swagger:route POST /api/v1/node/failover node createfailOver
- //
- // Create a relay.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: nodeResponse
- func createfailOver(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- nodeid := params["nodeid"]
- // confirm host exists
- node, err := logic.GetNodeByID(nodeid)
- if err != nil {
- slog.Error("failed to get node:", "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- err = proLogic.CreateFailOver(node)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- go mq.PublishPeerUpdate(false)
- w.Header().Set("Content-Type", "application/json")
- logic.ReturnSuccessResponseWithJson(w, r, node, "created failover successfully")
- }
- func resetFailOver(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- net := params["network"]
- nodes, err := logic.GetNetworkNodes(net)
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- for _, node := range nodes {
- if node.FailedOverBy != uuid.Nil {
- node.FailedOverBy = uuid.Nil
- node.FailOverPeers = make(map[string]struct{})
- logic.UpsertNode(&node)
- }
- }
- go mq.PublishPeerUpdate(false)
- w.Header().Set("Content-Type", "application/json")
- logic.ReturnSuccessResponse(w, r, "failover has been reset successfully")
- }
- // swagger:route DELETE /api/v1/node/failover node deletefailOver
- //
- // Create a relay.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: nodeResponse
- func deletefailOver(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- nodeid := params["nodeid"]
- // confirm host exists
- node, err := logic.GetNodeByID(nodeid)
- if err != nil {
- slog.Error("failed to get node:", "error", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- node.IsFailOver = false
- // Reset FailOvered Peers
- err = logic.UpsertNode(&node)
- if err != nil {
- slog.Error("failed to upsert node", "node", node.ID.String(), "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- go func() {
- proLogic.ResetFailOver(&node)
- mq.PublishPeerUpdate(false)
- }()
- w.Header().Set("Content-Type", "application/json")
- logic.ReturnSuccessResponseWithJson(w, r, node, "deleted failover successfully")
- }
- // swagger:route POST /api/node/{nodeid}/failOverME node failOver_me
- //
- // Create a relay.
- //
- // Schemes: https
- //
- // Security:
- // oauth
- //
- // Responses:
- // 200: nodeResponse
- func failOverME(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- nodeid := params["nodeid"]
- // confirm host exists
- node, err := logic.GetNodeByID(nodeid)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to get node:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- failOverNode, exists := proLogic.FailOverExists(node.Network)
- if !exists {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failover node doesn't exist in the network"), "badrequest"))
- return
- }
- var failOverReq models.FailOverMeReq
- err = json.NewDecoder(r.Body).Decode(&failOverReq)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- var sendPeerUpdate bool
- peerNode, err := logic.GetNodeByID(failOverReq.NodeID)
- if err != nil {
- slog.Error("peer not found: ", "nodeid", failOverReq.NodeID, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("peer not found"), "badrequest"))
- return
- }
- if node.IsRelayed || node.IsFailOver {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("node is relayed or acting as failover"), "badrequest"))
- return
- }
- if peerNode.IsRelayed || peerNode.IsFailOver {
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("peer node is relayed or acting as failover"), "badrequest"))
- return
- }
- err = proLogic.SetFailOverCtx(failOverNode, node, peerNode)
- if err != nil {
- slog.Error("failed to create failover", "id", node.ID.String(),
- "network", node.Network, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to create failover: %v", err), "internal"))
- return
- }
- slog.Info("[auto-relay] created relay on node", "node", node.ID.String(), "network", node.Network)
- sendPeerUpdate = true
- if sendPeerUpdate {
- go mq.PublishPeerUpdate(false)
- }
- w.Header().Set("Content-Type", "application/json")
- logic.ReturnSuccessResponse(w, r, "relayed successfully")
- }
|