gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2024-09-21 07:46:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

listenport logic

Browse source
This commit is contained in:
afeiszli 2022-02-22 17:50:41 -05:00
parent 4c976b49a2
commit 04d15d27af
1 changed files with 33 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
netclient/functions/join.go
View file

@ -162,12 +162,6 @@ func JoinNetwork(cfg config.ClientConfig, privateKey string, iscomms bool) error
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
// get free port based on returned default listen port
node.ListenPort, err = ncutils.GetFreePort(node.ListenPort)
if err != nil {
fmt.Printf("Error retrieving port: %v", err)
}
// safety check. If returned node from server is local, but not currently configured as local, set to local addr
if cfg.Node.IsLocal != "yes" && node.IsLocal == "yes" && node.LocalRange != "" {
node.LocalAddress, err = ncutils.GetLocalIP(node.LocalRange)
@ -209,12 +203,18 @@ func JoinNetwork(cfg config.ClientConfig, privateKey string, iscomms bool) error
}
ncutils.PrintLog("node created on remote server...updating configs", 1)
// keep track of the old listenport value
oldListenPort := node.ListenPort
nodeData := res.Data
if err = json.Unmarshal([]byte(nodeData), &node); err != nil {
return err
}
cfg.Node = node
setListenPort(oldListenPort, &cfg)
err = config.ModConfig(&node)
if err != nil {
return err
@ -281,3 +281,30 @@ func formatName(node models.Node) string {
}
return node.Name
}
func setListenPort(oldListenPort int32, cfg *config.ClientConfig) {
log.Println("DELETE ME: set listenport")
log.Println("DELETE ME: oldport " + string(oldListenPort))
log.Println("DELETE ME: newport " + string(cfg.Node.ListenPort))
// keep track of the returned listenport value
newListenPort := cfg.Node.ListenPort
if newListenPort != oldListenPort {
var errN error
// get free port based on returned default listen port
cfg.Node.ListenPort, errN = ncutils.GetFreePort(cfg.Node.ListenPort)
if errN != nil {
cfg.Node.ListenPort = newListenPort
ncutils.PrintLog("Error retrieving port: "+errN.Error(), 1)
}
// if newListenPort has been modified to find an available port, publish to server
if cfg.Node.ListenPort != newListenPort {
var currentCommsCfg = getCommsCfgByNode(&cfg.Node)
log.Println("DELETE ME: publishing update of port: " + string(cfg.Node.ListenPort))
PublishNodeUpdate(&currentCommsCfg, cfg)
}
}
}