agent connection error handling w/out exiting

dhamith93 · Dec 14, 2021 · a70c8b8 · a70c8b8
1 parent 449b477
commit a70c8b8
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/agent/main.go b/agent/main.go
@@ -75,6 +75,10 @@ func main() {
 
 func initAgent(config *config.Config) {
 	conn, c, ctx, cancel := createClient(config)
+	if conn == nil {
+		logger.Log("error", "error creating connection")
+		return
+	}
 	defer conn.Close()
 	defer cancel()
 	response, err := c.InitAgent(ctx, &api.ServerInfo{
@@ -90,12 +94,15 @@ func initAgent(config *config.Config) {
 
 func sendMonitorData(monitorData string, config *config.Config) {
 	conn, c, ctx, cancel := createClient(config)
+	if conn == nil {
+		logger.Log("error", "error creating connection")
+		return
+	}
 	defer conn.Close()
 	defer cancel()
 	_, err := c.HandleMonitorData(ctx, &api.MonitorData{MonitorData: monitorData})
 	if err != nil {
 		logger.Log("error", "error sending data: "+err.Error())
-		os.Exit(1)
 	}
 }
 
@@ -113,6 +120,10 @@ func sendCustomMetric(name string, unit string, value string, config *config.Con
 		return
 	}
 	conn, c, ctx, cancel := createClient(config)
+	if conn == nil {
+		logger.Log("error", "error creating connection")
+		return
+	}
 	defer conn.Close()
 	defer cancel()
 	_, err = c.HandleCustomMonitorData(ctx, &api.MonitorData{MonitorData: string(jsonData)})
@@ -135,7 +146,7 @@ func createClient(config *config.Config) (*grpc.ClientConn, api.MonitorDataServi
 	conn, err := grpc.Dial(config.MonitorEndpoint, grpc.WithInsecure())
 	if err != nil {
 		logger.Log("error", "connection error: "+err.Error())
-		os.Exit(1)
+		return nil, nil, nil, nil
 	}
 	c := api.NewMonitorDataServiceClient(conn)
 	token := generateToken()