available-values-inserting-db

intelops · Nov 8, 2023 · 4a0a60a · 4a0a60a
1 parent 5af755c
commit 4a0a60a
Showing 1 changed file with 9 additions and 22 deletions.
diff --git a/client/pkg/clickhouse/db_client.go b/client/pkg/clickhouse/db_client.go
@@ -603,11 +603,12 @@ func (c *DBClient) InsertTrivyImageMetrics(metrics model.TrivyImage) {
 func (c *DBClient) InsertTrivySbomMetrics(metrics model.Sbom) {
 	log.Println("####started inserting value")
 	result := metrics.Report
-	var (
-		tx, _   = c.conn.Begin()
-		stmt, _ = tx.Prepare(InsertTrivySbom)
-	)
+
 	if result.CycloneDX != nil {
+        var (
+            tx, _   = c.conn.Begin()
+            stmt, _ = tx.Prepare(InsertTrivySbom)
+        )
 		if _,err:= stmt.Exec(
 			metrics.ID,
 			result.CycloneDX.Metadata.Component.Name,
@@ -621,29 +622,15 @@ func (c *DBClient) InsertTrivySbomMetrics(metrics model.Sbom) {
 		); err!=nil {
 			log.Fatal(err)
 		}
-
-	}else {
-		if _,err:= stmt.Exec(
-			metrics.ID,
-			"-",
-			"-",
-			"-",
-			"-",
-			"-",
-			"-",
-			"-",
-			"-",
-		); err!=nil {
+		if err:=tx.Commit();err!=nil {
 			log.Fatal(err)
 		}
+		stmt.Close()
+	}else {
+		log.Println("No Data Available")
 
 	}
 
-	if err:=tx.Commit();err!=nil {
-		log.Fatal(err)
-	}
-	stmt.Close()
-
 }
 func (c *DBClient) Close() {
 	_ = c.conn.Close()