failover.go 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "net/http"
  7. "github.com/google/uuid"
  8. "github.com/gorilla/mux"
  9. controller "github.com/gravitl/netmaker/controllers"
  10. "github.com/gravitl/netmaker/logger"
  11. "github.com/gravitl/netmaker/logic"
  12. "github.com/gravitl/netmaker/models"
  13. "github.com/gravitl/netmaker/mq"
  14. proLogic "github.com/gravitl/netmaker/pro/logic"
  15. "golang.org/x/exp/slog"
  16. )
  17. // FailOverHandlers - handlers for FailOver
  18. func FailOverHandlers(r *mux.Router) {
  19. r.HandleFunc("/api/v1/node/{nodeid}/failover", logic.SecurityCheck(true, http.HandlerFunc(createfailOver))).Methods(http.MethodPost)
  20. r.HandleFunc("/api/v1/node/{nodeid}/failover", logic.SecurityCheck(true, http.HandlerFunc(deletefailOver))).Methods(http.MethodDelete)
  21. r.HandleFunc("/api/v1/node/{network}/failover/reset", logic.SecurityCheck(true, http.HandlerFunc(resetFailOver))).Methods(http.MethodPost)
  22. r.HandleFunc("/api/v1/node/{nodeid}/failover_me", controller.Authorize(true, false, "host", http.HandlerFunc(failOverME))).Methods(http.MethodPost)
  23. }
  24. // swagger:route POST /api/v1/node/failover node createfailOver
  25. //
  26. // Create a relay.
  27. //
  28. // Schemes: https
  29. //
  30. // Security:
  31. // oauth
  32. //
  33. // Responses:
  34. // 200: nodeResponse
  35. func createfailOver(w http.ResponseWriter, r *http.Request) {
  36. var params = mux.Vars(r)
  37. nodeid := params["nodeid"]
  38. // confirm host exists
  39. node, err := logic.GetNodeByID(nodeid)
  40. if err != nil {
  41. slog.Error("failed to get node:", "error", err.Error())
  42. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  43. return
  44. }
  45. err = proLogic.CreateFailOver(node)
  46. if err != nil {
  47. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  48. return
  49. }
  50. go mq.PublishPeerUpdate(false)
  51. w.Header().Set("Content-Type", "application/json")
  52. logic.ReturnSuccessResponseWithJson(w, r, node, "created failover successfully")
  53. }
  54. func resetFailOver(w http.ResponseWriter, r *http.Request) {
  55. var params = mux.Vars(r)
  56. net := params["network"]
  57. nodes, err := logic.GetNetworkNodes(net)
  58. if err != nil {
  59. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  60. return
  61. }
  62. for _, node := range nodes {
  63. if node.FailedOverBy != uuid.Nil {
  64. node.FailedOverBy = uuid.Nil
  65. node.FailOverPeers = make(map[string]struct{})
  66. logic.UpsertNode(&node)
  67. }
  68. }
  69. go mq.PublishPeerUpdate(false)
  70. w.Header().Set("Content-Type", "application/json")
  71. logic.ReturnSuccessResponse(w, r, "failover has been reset successfully")
  72. }
  73. // swagger:route DELETE /api/v1/node/failover node deletefailOver
  74. //
  75. // Create a relay.
  76. //
  77. // Schemes: https
  78. //
  79. // Security:
  80. // oauth
  81. //
  82. // Responses:
  83. // 200: nodeResponse
  84. func deletefailOver(w http.ResponseWriter, r *http.Request) {
  85. var params = mux.Vars(r)
  86. nodeid := params["nodeid"]
  87. // confirm host exists
  88. node, err := logic.GetNodeByID(nodeid)
  89. if err != nil {
  90. slog.Error("failed to get node:", "error", err.Error())
  91. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  92. return
  93. }
  94. node.IsFailOver = false
  95. // Reset FailOvered Peers
  96. err = logic.UpsertNode(&node)
  97. if err != nil {
  98. slog.Error("failed to upsert node", "node", node.ID.String(), "error", err)
  99. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  100. return
  101. }
  102. go func() {
  103. proLogic.ResetFailOver(&node)
  104. mq.PublishPeerUpdate(false)
  105. }()
  106. w.Header().Set("Content-Type", "application/json")
  107. logic.ReturnSuccessResponseWithJson(w, r, node, "deleted failover successfully")
  108. }
  109. // swagger:route POST /api/node/{nodeid}/failOverME node failOver_me
  110. //
  111. // Create a relay.
  112. //
  113. // Schemes: https
  114. //
  115. // Security:
  116. // oauth
  117. //
  118. // Responses:
  119. // 200: nodeResponse
  120. func failOverME(w http.ResponseWriter, r *http.Request) {
  121. var params = mux.Vars(r)
  122. nodeid := params["nodeid"]
  123. // confirm host exists
  124. node, err := logic.GetNodeByID(nodeid)
  125. if err != nil {
  126. logger.Log(0, r.Header.Get("user"), "failed to get node:", err.Error())
  127. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  128. return
  129. }
  130. failOverNode, exists := proLogic.FailOverExists(node.Network)
  131. if !exists {
  132. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("failover node doesn't exist in the network"), "badrequest"))
  133. return
  134. }
  135. var failOverReq models.FailOverMeReq
  136. err = json.NewDecoder(r.Body).Decode(&failOverReq)
  137. if err != nil {
  138. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  139. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  140. return
  141. }
  142. var sendPeerUpdate bool
  143. peerNode, err := logic.GetNodeByID(failOverReq.NodeID)
  144. if err != nil {
  145. slog.Error("peer not found: ", "nodeid", failOverReq.NodeID, "error", err)
  146. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("peer not found"), "badrequest"))
  147. return
  148. }
  149. if node.IsRelayed || node.IsFailOver {
  150. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("node is relayed or acting as failover"), "badrequest"))
  151. return
  152. }
  153. if peerNode.IsRelayed || peerNode.IsFailOver {
  154. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("peer node is relayed or acting as failover"), "badrequest"))
  155. return
  156. }
  157. err = proLogic.SetFailOverCtx(failOverNode, node, peerNode)
  158. if err != nil {
  159. slog.Error("failed to create failover", "id", node.ID.String(),
  160. "network", node.Network, "error", err)
  161. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to create failover: %v", err), "internal"))
  162. return
  163. }
  164. slog.Info("[auto-relay] created relay on node", "node", node.ID.String(), "network", node.Network)
  165. sendPeerUpdate = true
  166. if sendPeerUpdate {
  167. go mq.PublishPeerUpdate(false)
  168. }
  169. w.Header().Set("Content-Type", "application/json")
  170. logic.ReturnSuccessResponse(w, r, "relayed successfully")
  171. }