Merge pull request #37 from h0n9/feature/36_gracefully-shutdown-lake-…

…service

Gracefully shutdown lake service

cli/agent/agent.go

@@ -52,7 +52,10 @@ var Cmd = &cobra.Command{
 		ctx, cancel := context.WithCancel(context.Background())
 		logger.Info().Msg("initalized context")
 
-		var grpcServer *grpc.Server = nil
+		var (
+			lakeService *lake.Service = nil
+			grpcServer  *grpc.Server  = nil
+		)
 
 		sigCh := make(chan os.Signal, 1)
 		signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
@@ -64,14 +67,19 @@ var Cmd = &cobra.Command{
 			cancel()
 			if grpcServer != nil {
 				grpcServer.GracefulStop()
+				logger.Info().Msg("gracefully stopped gRPC server")
+			}
+			if lakeService != nil {
+				lakeService.Close()
+				logger.Info().Msg("closed lake service")
 			}
 			wg.Done()
 		}()
 		logger.Info().Msg("listening os signal: SIGINT, SIGTERM")
 
 		grpcServer = grpc.NewServer()
 		logger.Info().Msg("initalized gRPC server")
-		lakeService, err := lake.NewService(
+		lakeService, err = lake.NewService(
 			ctx,
 			&logger,
 			[]byte(seed),

lake/lake.go

@@ -47,7 +47,6 @@ func (service *Service) Close() {
 	if service.relayer != nil {
 		service.relayer.Close()
 	}
-	service.logger.Info().Msg("closed lake service")
 }
 
 func (service *Service) Publish(ctx context.Context, req *pb.PublishReq) (*pb.PublishRes, error) {