From 49bed1f7b31bd134773e95cfb9137f2a9bca2735 Mon Sep 17 00:00:00 2001
From: Anlan He <hanlan@vmware.com>
Date: Tue, 13 Jun 2023 13:18:49 -0400
Subject: [PATCH] Fix collector klog flag not supported error (#311)

Add registration of klog flags

Fixes: #306

Signed-off-by: heanlan <hanlan@vmware.com>
---
 build/yamls/collector/templates/ipfix-collector.yaml | 4 +++-
 build/yamls/collector/values.yaml                    | 2 ++
 build/yamls/ipfix-collector.yaml                     | 1 -
 cmd/collector/collector.go                           | 1 +
 4 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/build/yamls/collector/templates/ipfix-collector.yaml b/build/yamls/collector/templates/ipfix-collector.yaml
index 35c84bad..41f8eb37 100644
--- a/build/yamls/collector/templates/ipfix-collector.yaml
+++ b/build/yamls/collector/templates/ipfix-collector.yaml
@@ -44,7 +44,9 @@ spec:
     spec:
       containers:
       - args:
-        - --v=0
+        {{- if .Values.logVerbosity }}
+        - --v={{ .Values.logVerbosity }}
+        {{- end }}
         - --ipfix.port={{ .Values.port }}
         - --ipfix.transport={{ .Values.protocol }}
         image: projects.registry.vmware.com/antrea/ipfix-collector:{{ .Values.image_tag }}
diff --git a/build/yamls/collector/values.yaml b/build/yamls/collector/values.yaml
index 04619e3f..f8f0187e 100644
--- a/build/yamls/collector/values.yaml
+++ b/build/yamls/collector/values.yaml
@@ -1,3 +1,5 @@
 protocol: tcp
 port: 4739
 image_tag: latest
+# -- Log verbosity switch for IPFIX collector.
+logVerbosity: 0
diff --git a/build/yamls/ipfix-collector.yaml b/build/yamls/ipfix-collector.yaml
index 9460e6d5..e926057b 100644
--- a/build/yamls/ipfix-collector.yaml
+++ b/build/yamls/ipfix-collector.yaml
@@ -48,7 +48,6 @@ spec:
     spec:
       containers:
       - args:
-        - --v=0
         - --ipfix.port=4739
         - --ipfix.transport=tcp
         image: projects.registry.vmware.com/antrea/ipfix-collector:latest
diff --git a/cmd/collector/collector.go b/cmd/collector/collector.go
index f566d5b5..55f7b890 100644
--- a/cmd/collector/collector.go
+++ b/cmd/collector/collector.go
@@ -198,6 +198,7 @@ func newCollectorCommand() *cobra.Command {
 	addIPFIXFlags(flags)
 	// Install command line flags
 	flags.AddGoFlagSet(flag.CommandLine)
+	logs.AddFlags(flags)
 	return cmd
 }