Fix cmd init

Makefile

@@ -1,3 +1,12 @@
+# Setting SHELL to bash allows bash commands to be executed by recipes.
+# This is a requirement for 'setup-envtest.sh' in the test target.
+# Options are set to exit when a recipe line exits non-zero or a piped command fails.
+SHELL = /usr/bin/env bash -o pipefail
+.SHELLFLAGS = -ec
+
+# General Project parameters
+PROJECT_DIR := $(shell dirname $(abspath $(lastword $(MAKEFILE_LIST))))
+
 # Image related parameters, used when building image
 IMAGE_REPOSITORY ?= nvidia.com
 IMAGE_NAME ?= multi-networkpolicy-tc
@@ -11,12 +20,6 @@ else
 GOBIN=$(shell go env GOBIN)
 endif
 
-# Setting SHELL to bash allows bash commands to be executed by recipes.
-# This is a requirement for 'setup-envtest.sh' in the test target.
-# Options are set to exit when a recipe line exits non-zero or a piped command fails.
-SHELL = /usr/bin/env bash -o pipefail
-.SHELLFLAGS = -ec
-
 TARGET_OS ?= $(shell go env GOOS)
 TARGET_ARCH ?= $(shell go env GOARCH)
 
@@ -27,7 +30,6 @@ GO_BIN_SUFFIX ?= $(TARGET_OS)-$(TARGET_ARCH)
 
 all: build
 
-
 ##@ General
 
 # The help target prints out all targets with their descriptions organized
@@ -54,7 +56,6 @@ unit-test: ## Run unit tests.
 
 test: lint unit-test ## Run all tests (lint, unit-test).
 
-
 ##@ Build
 .PHONY: build
 build: ## Build multi-networkpolicy-tc binary.
@@ -97,8 +98,13 @@ MOCKERY = $(shell pwd)/bin/mockery
 mockery: ## Download mockery if necessary.
 	$(call go-install-tool,$(MOCKERY),github.com/vektra/mockery/[email protected])
 
+.PHONY: clean
+clean:
+	@rm -rf build
+	@rm -rf bin
+
+
 # go-get-tool will 'go get' any package $2 and install it to $1.
-PROJECT_DIR := $(shell dirname $(abspath $(lastword $(MAKEFILE_LIST))))
 define go-install-tool
 @[ -f $(1) ] || { \
 set -e ;\

cmd/multi-networkpolicy-tc/main.go

@@ -40,15 +40,16 @@ func main() {
 	initLogs(ctx)
 	defer klog.Flush()
 	opts := server.NewOptions()
-	srv, err := server.NewServer(opts)
-	if err != nil {
-		klog.Exit(err)
-	}
 
 	cmd := &cobra.Command{
 		Use:  "multi-networkpolicy-tc",
 		Long: `TBD`,
 		Run: func(cmd *cobra.Command, args []string) {
+			srv, err := server.NewServer(opts)
+			if err != nil {
+				klog.Exit(err)
+			}
+
 			srv.Run(ctx)
 		},
 	}

pkg/server/server.go

@@ -152,6 +152,9 @@ func (s *Server) Run(ctx context.Context) {
 	go s.SyncLoop(ctx)
 
 	s.birthCry()
+
+	// wait on Context
+	<-ctx.Done()
 }
 
 // setInitialized sets s.initialized to 1 if value is true. the set operation is atomic