zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-05 16:57:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

relay fix

Browse Source
This commit is contained in:
Abhishek Kondur
2023-01-10 10:18:03 +05:30
parent d373e90597
commit 80d9032d47
2 changed files with 72 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
controllers/hosts.go
View File

@@ -214,7 +214,7 @@ func createHostRelay(w http.ResponseWriter, r *http.Request) {
return
}
relay.HostID = params["hostid"]
relayHost, relayedHosts, err := logic.CreateHostRelay(relay)
relayHost, _, err := logic.CreateHostRelay(relay)
if err != nil {
logger.Log(0, r.Header.Get("user"),
fmt.Sprintf("failed to create relay on host [%s]: %v", relay.HostID, err))
@@ -223,18 +223,24 @@ func createHostRelay(w http.ResponseWriter, r *http.Request) {
}
logger.Log(1, r.Header.Get("user"), "created relay on host", relay.HostID)
for _, relayedHost := range relayedHosts {
// for _, relayedHost := range relayedHosts {
err = mq.PublishSingleHostUpdate(&relayedHost)
if err != nil {
logger.Log(1, "error sending update to relayed host ", relayedHost.ID.String(), ": ", err.Error())
// err = mq.PublishSingleHostUpdate(&relayedHost)
// if err != nil {
// logger.Log(1, "error sending update to relayed host ", relayedHost.ID.String(), ": ", err.Error())
// }
// }
// // publish host update for relayhost
// err = mq.PublishSingleHostUpdate(relayHost)
// if err != nil {
// logger.Log(1, "error sending update to relay host ", relayHost.ID.String(), ": ", err.Error())
// }
go func() {
if err := mq.PublishPeerUpdate("", false); err != nil {
logger.Log(0, "fail to publish peer update: ", err.Error())
}
}
// publish host update for relayhost
err = mq.PublishSingleHostUpdate(relayHost)
if err != nil {
logger.Log(1, "error sending update to relay host ", relayHost.ID.String(), ": ", err.Error())
}
}()
apiHostData := relayHost.ConvertNMHostToAPI()
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(apiHostData)
@@ -255,23 +261,28 @@ func deleteHostRelay(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
hostid := params["hostid"]
relayHost, relayedHosts, err := logic.DeleteHostRelay(hostid)
relayHost, _, err := logic.DeleteHostRelay(hostid)
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
}
logger.Log(1, r.Header.Get("user"), "deleted relay host", hostid)
for _, relayedHost := range relayedHosts {
err = mq.PublishSingleHostUpdate(&relayedHost)
if err != nil {
logger.Log(1, "error sending update to relayed host ", relayedHost.ID.String(), ": ", err.Error())
// for _, relayedHost := range relayedHosts {
// err = mq.PublishSingleHostUpdate(&relayedHost)
// if err != nil {
// logger.Log(1, "error sending update to relayed host ", relayedHost.ID.String(), ": ", err.Error())
// }
// }
// err = mq.PublishSingleHostUpdate(relayHost)
// if err != nil {
// logger.Log(1, "error sending update to relayed host ", relayHost.ID.String(), ": ", err.Error())
// }
go func() {
if err := mq.PublishPeerUpdate("", false); err != nil {
logger.Log(0, "fail to publish peer update: ", err.Error())
}
}
err = mq.PublishSingleHostUpdate(relayHost)
if err != nil {
logger.Log(1, "error sending update to relayed host ", relayHost.ID.String(), ": ", err.Error())
}
}()
apiHostData := relayHost.ConvertNMHostToAPI()
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(apiHostData)