[CLOUDGA-20737] Add Feature flag for DB audit (#227)

Co-authored-by: aquatiko <[email protected]>
yugabyte · Apr 22, 2024 · f257362 · f257362
1 parent 90c548a
commit f257362
Show file tree

Hide file tree

Showing 9 changed files with 3 additions and 204 deletions.
diff --git a/cmd/db_audit_logs_exporter_test.go b/cmd/db_audit_logs_exporter_test.go
@@ -49,6 +49,7 @@ var _ = Describe("Db Audit", func() {
 		Expect(err).ToNot(HaveOccurred())
 		os.Setenv("YBM_HOST", fmt.Sprintf("http://%s", server.Addr()))
 		os.Setenv("YBM_APIKEY", "test-token")
+		os.Setenv("YBM_FF_DB_AUDIT_LOGS", "true")
 	})
 
 	Context("When associating DB Audit config", func() {

diff --git a/cmd/root.go b/cmd/root.go
@@ -138,7 +138,7 @@ func init() {
 	rootCmd.AddCommand(user.UserCmd)
 	rootCmd.AddCommand(metrics_exporter.MetricsExporterCmd)
 	rootCmd.AddCommand(integration.IntegrationCmd)
-	rootCmd.AddCommand(db_audit_logs_exporter.DbAuditLogsExporterCmd)
+	util.AddCommandIfFeatureFlag(rootCmd, db_audit_logs_exporter.DbAuditLogsExporterCmd, util.DB_AUDIT_LOGS)
 	util.AddCommandIfFeatureFlag(rootCmd, tools.ToolsCmd, util.TOOLS)
 	util.AddCommandIfFeatureFlag(rootCmd, cdc.CdcCmd, util.CDC)
 

diff --git a/cmd/util/feature_flags.go b/cmd/util/feature_flags.go
@@ -31,6 +31,7 @@ const (
 	TOOLS               FeatureFlag = "TOOLS"
 	AZURE_CIDR_ALLOWED  FeatureFlag = "AZURE_CIDR_ALLOWED"
 	ENTERPRISE_SECURITY FeatureFlag = "ENTERPRISE_SECURITY"
+	DB_AUDIT_LOGS       FeatureFlag = "DB_AUDIT_LOGS"
 )
 
 func (f FeatureFlag) String() string {

diff --git a/docs/ybm.md b/docs/ybm.md
@@ -31,7 +31,6 @@ ybm [flags]
 * [ybm backup](ybm_backup.md)	 - Manage backup operations of a cluster
 * [ybm cluster](ybm_cluster.md)	 - Manage cluster operations
 * [ybm completion](ybm_completion.md)	 - Generate the autocompletion script for the specified shell
-* [ybm db-audit-logs-exporter](ybm_db-audit-logs-exporter.md)	 - Manage DB Audit Logs
 * [ybm integration](ybm_integration.md)	 - Manage Integration
 * [ybm metrics-exporter](ybm_metrics-exporter.md)	 - Manage Metrics Exporter
 * [ybm network-allow-list](ybm_network-allow-list.md)	 - Manage Network Allow Lists

diff --git a/docs/ybm_db-audit-logs-exporter.md b/docs/ybm_db-audit-logs-exporter.md
diff --git a/docs/ybm_db-audit-logs-exporter_assign.md b/docs/ybm_db-audit-logs-exporter_assign.md
diff --git a/docs/ybm_db-audit-logs-exporter_list.md b/docs/ybm_db-audit-logs-exporter_list.md
diff --git a/docs/ybm_db-audit-logs-exporter_unassign.md b/docs/ybm_db-audit-logs-exporter_unassign.md
diff --git a/docs/ybm_db-audit-logs-exporter_update.md b/docs/ybm_db-audit-logs-exporter_update.md