Browse Source

extend signal model to include node ids

Abhishek Kondur 1 year ago
parent
commit
66b2516ba7
3 changed files with 9 additions and 5 deletions
  1. 2 0
      models/host.go
  2. 5 0
      models/mqtt.go
  3. 2 5
      pro/controllers/failover.go

+ 2 - 0
models/host.go

@@ -152,6 +152,8 @@ type Signal struct {
 	ToHostPubKey      string       `json:"to_host_pubkey"`
 	FromHostID        string       `json:"from_host_id"`
 	ToHostID          string       `json:"to_host_id"`
+	FromNodeID        string       `json:"from_node_id"`
+	ToNodeID          string       `json:"to_node_id"`
 	Reply             bool         `json:"reply"`
 	Action            SignalAction `json:"action"`
 	IsPro             bool         `json:"is_pro"`

+ 5 - 0
models/mqtt.go

@@ -70,3 +70,8 @@ type FwUpdate struct {
 	IsEgressGw bool                  `json:"is_egress_gw"`
 	EgressInfo map[string]EgressInfo `json:"egress_info"`
 }
+
+// FailOverMeReq - struct for failover req
+type FailOverMeReq struct {
+	NodeID string `json:"node_id"`
+}

+ 2 - 5
pro/controllers/failover.go

@@ -10,15 +10,12 @@ import (
 	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"
 )
 
-type FailOverMeReq struct {
-	NodeID string `json:"node_id"`
-}
-
 // 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)
@@ -121,7 +118,7 @@ func failOverME(w http.ResponseWriter, r *http.Request) {
 		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
 		return
 	}
-	var failOverReq FailOverMeReq
+	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())