Browse Source

go routines now run forever like they are supposed to

Matthew R Kasun 3 years ago
parent
commit
8768a2c390
1 changed files with 90 additions and 86 deletions
  1. 90 86
      netclient/functions/daemon.go

+ 90 - 86
netclient/functions/daemon.go

@@ -30,6 +30,7 @@ func Daemon() error {
 	signal.Notify(quit, syscall.SIGTERM, os.Interrupt)
 	<-quit
 	cancel()
+	ncutils.Log("all done")
 	return nil
 }
 
@@ -48,29 +49,28 @@ func SetupMQTT(cfg config.ClientConfig) mqtt.Client {
 
 //Netclient sets up Message Queue and subsribes/publishes updates to/from server
 func Netclient(ctx context.Context, network string) {
-	select {
-	case <-ctx.Done():
-		ncutils.Log("shutting down daemon")
-		return
-	default:
-		var cfg config.ClientConfig
-		cfg.Network = network
-		cfg.ReadConfig()
-		//fix NodeID to remove ### so NodeID can be used as message topic
-		//remove with GRA-73
-		cfg.Node.ID = strings.ReplaceAll(cfg.Node.ID, "###", "-")
-		ncutils.Log("daemon started for network:" + network)
-		client := SetupMQTT(cfg)
-		if token := client.Subscribe("#", 0, nil); token.Wait() && token.Error() != nil {
-			log.Fatal(token.Error())
-		}
-		client.AddRoute("update/"+cfg.Node.ID, NodeUpdate)
-		client.AddRoute("update/peers/"+cfg.Node.ID, UpdatePeers)
-		client.AddRoute("update/keys/"+cfg.Node.ID, UpdateKeys)
-		defer client.Disconnect(250)
-		go Checkin(ctx, cfg, network)
-		go Metrics(ctx, cfg, network)
+	ncutils.Log("netclient go routine started for " + network)
+	var cfg config.ClientConfig
+	cfg.Network = network
+	cfg.ReadConfig()
+	//fix NodeID to remove ### so NodeID can be used as message topic
+	//remove with GRA-73
+	cfg.Node.ID = strings.ReplaceAll(cfg.Node.ID, "###", "-")
+	ncutils.Log("daemon started for network:" + network)
+	client := SetupMQTT(cfg)
+	if token := client.Subscribe("#", 0, nil); token.Wait() && token.Error() != nil {
+		log.Fatal(token.Error())
 	}
+	client.AddRoute("update/"+cfg.Node.ID, NodeUpdate)
+	client.AddRoute("update/peers/"+cfg.Node.ID, UpdatePeers)
+	client.AddRoute("update/keys/"+cfg.Node.ID, UpdateKeys)
+	defer client.Disconnect(250)
+	go Checkin(ctx, cfg, network)
+	go Metrics(ctx, cfg, network)
+	<-ctx.Done()
+	ncutils.Log("shutting down daemon")
+	return
+	ncutils.Log("netclient go routine ended for " + network)
 }
 
 //All -- mqtt message hander for all ('#') topics
@@ -97,42 +97,44 @@ var UpdateKeys mqtt.MessageHandler = func(client mqtt.Client, msg mqtt.Message)
 //Checkin  -- go routine that checks for public or local ip changes, publishes changes
 //   if there are no updates, simply "pings" the server as a checkin
 func Checkin(ctx context.Context, cfg config.ClientConfig, network string) {
-	select {
-	case <-ctx.Done():
-		ncutils.Log("Checkin cancelled")
-		return
-		//delay should be configuraable -> use cfg.Node.NetworkSettings.DefaultCheckInInterval ??
-	case <-time.After(time.Second * 10):
-		ncutils.Log("Checkin running")
-		if cfg.Node.Roaming == "yes" && cfg.Node.IsStatic != "yes" {
-			extIP, err := ncutils.GetPublicIP()
-			if err != nil {
-				ncutils.PrintLog("error encountered checking ip addresses: "+err.Error(), 1)
-			}
-			if cfg.Node.Endpoint != extIP && extIP != "" {
-				ncutils.PrintLog("endpoint has changed from "+cfg.Node.Endpoint+" to "+extIP, 1)
-				UpdateEndpoint(cfg, network, extIP)
-			}
-			intIP, err := getPrivateAddr()
-			if err != nil {
-				ncutils.PrintLog("error encountered checking ip addresses: "+err.Error(), 1)
-			}
-			if cfg.Node.LocalAddress != intIP && intIP != "" {
-				ncutils.PrintLog("local Address has changed from "+cfg.Node.LocalAddress+" to "+intIP, 1)
-				UpdateLocalAddress(cfg, network, intIP)
-			}
-		} else {
-			localIP, err := ncutils.GetLocalIP(cfg.Node.LocalRange)
-			if err != nil {
-				ncutils.PrintLog("error encountered checking ip addresses: "+err.Error(), 1)
-			}
-			if cfg.Node.Endpoint != localIP && localIP != "" {
-				ncutils.PrintLog("endpoint has changed from "+cfg.Node.Endpoint+" to "+localIP, 1)
-				UpdateEndpoint(cfg, network, localIP)
+	for {
+		select {
+		case <-ctx.Done():
+			ncutils.Log("Checkin cancelled")
+			return
+			//delay should be configuraable -> use cfg.Node.NetworkSettings.DefaultCheckInInterval ??
+		case <-time.After(time.Second * 10):
+			ncutils.Log("Checkin running")
+			if cfg.Node.Roaming == "yes" && cfg.Node.IsStatic != "yes" {
+				extIP, err := ncutils.GetPublicIP()
+				if err != nil {
+					ncutils.PrintLog("error encountered checking ip addresses: "+err.Error(), 1)
+				}
+				if cfg.Node.Endpoint != extIP && extIP != "" {
+					ncutils.PrintLog("endpoint has changed from "+cfg.Node.Endpoint+" to "+extIP, 1)
+					UpdateEndpoint(cfg, network, extIP)
+				}
+				intIP, err := getPrivateAddr()
+				if err != nil {
+					ncutils.PrintLog("error encountered checking ip addresses: "+err.Error(), 1)
+				}
+				if cfg.Node.LocalAddress != intIP && intIP != "" {
+					ncutils.PrintLog("local Address has changed from "+cfg.Node.LocalAddress+" to "+intIP, 1)
+					UpdateLocalAddress(cfg, network, intIP)
+				}
+			} else {
+				localIP, err := ncutils.GetLocalIP(cfg.Node.LocalRange)
+				if err != nil {
+					ncutils.PrintLog("error encountered checking ip addresses: "+err.Error(), 1)
+				}
+				if cfg.Node.Endpoint != localIP && localIP != "" {
+					ncutils.PrintLog("endpoint has changed from "+cfg.Node.Endpoint+" to "+localIP, 1)
+					UpdateEndpoint(cfg, network, localIP)
+				}
 			}
+			Hello(cfg, network)
+			ncutils.Log("Checkin complete")
 		}
-		Hello(cfg, network)
-		ncutils.Log("Checkin complete")
 	}
 }
 
@@ -159,7 +161,7 @@ func UpdateLocalAddress(cfg config.ClientConfig, network, ip string) {
 //Hello -- ping the broker to let server know node is alive and doing fine
 func Hello(cfg config.ClientConfig, network string) {
 	client := SetupMQTT(cfg)
-	if token := client.Publish("ping/"+network+"/"+cfg.Node.ID, 0, false, "hello world!"); token.Wait() && token.Error() != nil {
+	if token := client.Publish("ping/"+cfg.Node.ID, 0, false, "hello world!"); token.Wait() && token.Error() != nil {
 		ncutils.Log("error publishing ping " + token.Error().Error())
 	}
 	client.Disconnect(250)
@@ -167,35 +169,37 @@ func Hello(cfg config.ClientConfig, network string) {
 
 //Metics --  go routine that collects wireguard metrics and publishes to broker
 func Metrics(ctx context.Context, cfg config.ClientConfig, network string) {
-	select {
-	case <-ctx.Done():
-		ncutils.Log("Metrics collection cancelled")
-		return
-		//delay should be configuraable -> use cfg.Node.NetworkSettings.DefaultCheckInInterval ??
-	case <-time.After(time.Second * 60):
-		ncutils.Log("Metrics collection running")
-		ncutils.Log("Metrics running")
-		wg, err := wgctrl.New()
-		if err != nil {
-			ncutils.Log("error getting devices " + err.Error())
-			break
-		}
-		device, err := wg.Device(cfg.Node.Interface)
-		if err != nil {
-			ncutils.Log("error readind wg device " + err.Error())
-			break
-		}
-		bytes, err := json.Marshal(device.Peers)
-		if err != nil {
-			ncutils.Log("error marshaling peers " + err.Error())
-			break
-		}
-		client := SetupMQTT(cfg)
-		if token := client.Publish("metrics/"+cfg.Node.ID, 1, false, bytes); token.Wait() && token.Error() != nil {
-			ncutils.Log("error publishing metrics " + token.Error().Error())
+	for {
+		select {
+		case <-ctx.Done():
+			ncutils.Log("Metrics collection cancelled")
+			return
+			//delay should be configuraable -> use cfg.Node.NetworkSettings.DefaultCheckInInterval ??
+		case <-time.After(time.Second * 60):
+			ncutils.Log("Metrics collection running")
+			ncutils.Log("Metrics running")
+			wg, err := wgctrl.New()
+			if err != nil {
+				ncutils.Log("error getting devices " + err.Error())
+				break
+			}
+			device, err := wg.Device(cfg.Node.Interface)
+			if err != nil {
+				ncutils.Log("error readind wg device " + err.Error())
+				break
+			}
+			bytes, err := json.Marshal(device.Peers)
+			if err != nil {
+				ncutils.Log("error marshaling peers " + err.Error())
+				break
+			}
+			client := SetupMQTT(cfg)
+			if token := client.Publish("metrics/"+cfg.Node.ID, 1, false, bytes); token.Wait() && token.Error() != nil {
+				ncutils.Log("error publishing metrics " + token.Error().Error())
+			}
+			wg.Close()
+			client.Disconnect(250)
+			ncutils.Log("metrics collection complete")
 		}
-		wg.Close()
-		client.Disconnect(250)
-		ncutils.Log("metrics collection complete")
 	}
 }