exit with error code (#225)

Signed-off-by: Sammy Oina <sammyoina@gmail.com>

cmd/agent/main.go

@@ -11,6 +11,7 @@ import (
 	"log/slog"
 	"time"
 
+	mglog "github.com/absmach/magistrala/logger"
 	"github.com/absmach/magistrala/pkg/prometheus"
 	"github.com/cenkalti/backoff/v4"
 	"github.com/google/go-sev-guest/client"
@@ -52,9 +53,13 @@ func main() {
 	}
 	defer conn.Close()
 
+	var exitCode int
+	defer mglog.ExitWithError(&exitCode)
+
 	var level slog.Level
 	if err := level.UnmarshalText([]byte(cfg.AgentConfig.LogLevel)); err != nil {
 		log.Println(err)
+		exitCode = 1
 		return
 	}
 	handler := agentlogger.NewProtoHandler(conn, &slog.HandlerOptions{Level: level}, cfg.ID)
@@ -63,6 +68,7 @@ func main() {
 	eventSvc, err := events.New(svcName, cfg.ID, manager.ManagerVsockPort)
 	if err != nil {
 		logger.Error(fmt.Sprintf("failed to create events service %s", err.Error()))
+		exitCode = 1
 		return
 	}
 	defer eventSvc.Close()
@@ -70,6 +76,7 @@ func main() {
 	qp, err := quoteprovider.GetQuoteProvider()
 	if err != nil {
 		logger.Error(fmt.Sprintf("failed to create quote provider %s", err.Error()))
+		exitCode = 1
 		return
 	}
 
@@ -93,6 +100,7 @@ func main() {
 	authSvc, err := auth.New(cfg)
 	if err != nil {
 		logger.Error(fmt.Sprintf("failed to create auth service %s", err.Error()))
+		exitCode = 1
 		return
 	}
 

cmd/manager/main.go

@@ -57,9 +57,13 @@ func main() {
 		log.Fatalf(err.Error())
 	}
 
+	var exitCode int
+	defer mglog.ExitWithError(&exitCode)
+
 	if cfg.InstanceID == "" {
 		if cfg.InstanceID, err = uuid.New().ID(); err != nil {
 			logger.Error(fmt.Sprintf("Failed to generate instance ID: %s", err))
+			exitCode = 1
 			return
 		}
 	}
@@ -78,6 +82,7 @@ func main() {
 	qemuCfg := qemu.Config{}
 	if err := env.ParseWithOptions(&qemuCfg, env.Options{Prefix: envPrefixQemu}); err != nil {
 		logger.Error(fmt.Sprintf("failed to load QEMU configuration: %s", err))
+		exitCode = 1
 		return
 	}
 	args := qemuCfg.ConstructQemuArgs()
@@ -86,12 +91,14 @@ func main() {
 	managerGRPCConfig := grpc.Config{}
 	if err := env.ParseWithOptions(&managerGRPCConfig, env.Options{Prefix: envPrefixGRPC}); err != nil {
 		logger.Error(fmt.Sprintf("failed to load %s gRPC client configuration : %s", svcName, err))
+		exitCode = 1
 		return
 	}
 
 	managerGRPCClient, managerClient, err := managergrpc.NewManagerClient(managerGRPCConfig)
 	if err != nil {
 		logger.Error(err.Error())
+		exitCode = 1
 		return
 	}
 	defer managerGRPCClient.Close()
@@ -99,6 +106,7 @@ func main() {
 	pc, err := managerClient.Process(ctx)
 	if err != nil {
 		logger.Error(err.Error())
+		exitCode = 1
 		return
 	}
 
@@ -106,6 +114,7 @@ func main() {
 	svc, err := newService(logger, tracer, qemuCfg, eventsChan, cfg.BackendMeasurementBinary)
 	if err != nil {
 		logger.Error(err.Error())
+		exitCode = 1
 		return
 	}