fix lint

Signed-off-by: Sammy Oina <[email protected]>
ultravioletrs · Jan 9, 2025 · de8c517 · de8c517
1 parent 20c214f
commit de8c517
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 8 deletions.
diff --git a/agent/cvm/api/grpc/client.go b/agent/cvm/api/grpc/client.go
@@ -133,7 +133,7 @@ func (client *CVMClient) executeRun(ctx context.Context, runReq *cvm.Computation
 	client.mu.Lock()
 	defer client.mu.Unlock()
 
-	err := client.sp.Start(agent.AgentConfig{
+	err := client.sp.Start(ctx, agent.AgentConfig{
 		Port:         runReq.AgentConfig.Port,
 		Host:         runReq.AgentConfig.Host,
 		CertFile:     runReq.AgentConfig.CertFile,

diff --git a/agent/cvm/server/cvm.go b/agent/cvm/server/cvm.go
@@ -36,7 +36,7 @@ func NewServerProvider(logger *slog.Logger, svc agent.Service) *AgentServer {
 	}
 }
 
-func (as *AgentServer) Start(cfg agent.AgentConfig, cmp agent.Computation) error {
+func (as *AgentServer) Start(ctx context.Context, cfg agent.AgentConfig, cmp agent.Computation) error {
 	if cfg.Port == "" {
 		cfg.Port = defSvcGRPCPort
 	}
@@ -72,7 +72,7 @@ func (as *AgentServer) Start(cfg agent.AgentConfig, cmp agent.Computation) error
 		return err
 	}
 
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := context.WithCancel(ctx)
 
 	as.gs = grpcserver.New(ctx, cancel, svcName, agentGrpcServerConfig, registerAgentServiceServer, as.logger, qp, authSvc)
 

diff --git a/cmd/agent/main.go b/cmd/agent/main.go
@@ -44,14 +44,14 @@ func main() {
 	ctx, cancel := context.WithCancel(context.Background())
 	g, ctx := errgroup.WithContext(ctx)
 
-	var exitCode int
-	defer mglog.ExitWithError(&exitCode)
-
 	var cfg config
 	if err := env.Parse(&cfg); err != nil {
 		log.Fatalf("failed to load %s configuration : %s", svcName, err)
 	}
 
+	var exitCode int
+	defer mglog.ExitWithError(&exitCode)
+
 	var level slog.Level
 	if err := level.UnmarshalText([]byte(cfg.LogLevel)); err != nil {
 		log.Println(err)

diff --git a/internal/logger/protohandler.go b/internal/logger/protohandler.go
@@ -96,7 +96,6 @@ func (h *handler) Handle(_ context.Context, r slog.Record) error {
 		if err != nil {
 			return err
 		}
-
 	}
 
 	return nil

diff --git a/internal/logger/protohandler_test.go b/internal/logger/protohandler_test.go
@@ -43,7 +43,6 @@ func TestHandleMessageSuccess(t *testing.T) {
 
 // TestHandleMessageFailure tests the caching mechanism when the write fails.
 func TestHandleMessageFailure(t *testing.T) {
-
 	protohandler := NewProtoHandler(&failedWriter{}, nil, make(chan *cvm.ClientStreamMessage, 1))
 	record := slog.Record{
 		Time:    time.Now(),
-Original file line number
+Diff line change
@@ Expand Up @@
     		if err != nil {
     			return err
     		}
     	}
     	return nil
@@ Expand Down @@