Skip to content

Commit 318eadb

Browse files
an1l4an1l4
committed
error-handle-changes-added
1 parent 6815d55 commit 318eadb

File tree

1 file changed

+30
-60
lines changed

1 file changed

+30
-60
lines changed

‎client/pkg/clickhouse/db_client.go‎

Lines changed: 30 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -136,14 +136,12 @@ func (c *DBClient) InsertContainerEventAzure(pushEvent model.AzureContainerPushE
136136

137137
tx, err := c.conn.Begin()
138138
if err != nil {
139-
log.Printf("error beginning transaction: %v", err)
140-
return
139+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
141140
}
142141

143142
stmt, err := tx.Prepare(string(InsertAzureContainerPushEvent))
144143
if err != nil {
145-
log.Printf("error preparing statement: %v", err)
146-
return
144+
log.Fatalf("error preparing statement: %v", err)
147145
}
148146

149147
defer stmt.Close()
@@ -187,14 +185,12 @@ func (c *DBClient) InsertContainerEventAzure(pushEvent model.AzureContainerPushE
187185
func (c *DBClient) InsertContainerEventQuay(pushEvent model.QuayImagePushPayload) {
188186
tx, err := c.conn.Begin()
189187
if err != nil {
190-
log.Printf("error beginning transaction: %v", err)
191-
return
188+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
192189
}
193190

194191
stmt, err := tx.Prepare(string(InsertQuayContainerPushEvent))
195192
if err != nil {
196-
log.Printf("error preparing statement: %v", err)
197-
return
193+
log.Fatalf("error preparing statement: %v", err)
198194
}
199195

200196
defer stmt.Close()
@@ -240,14 +236,12 @@ func (c *DBClient) InsertContainerEventQuay(pushEvent model.QuayImagePushPayload
240236
func (c *DBClient) InsertContainerEventJfrog(pushEvent model.JfrogContainerPushEventPayload) {
241237
tx, err := c.conn.Begin()
242238
if err != nil {
243-
log.Printf("error beginning transaction: %v", err)
244-
return
239+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
245240
}
246241

247242
stmt, err := tx.Prepare(string(InsertJfrogContainerPushEvent))
248243
if err != nil {
249-
log.Printf("error preparing statement: %v", err)
250-
return
244+
log.Fatalf("error preparing statement: %v", err)
251245
}
252246

253247
defer stmt.Close()
@@ -293,13 +287,11 @@ func (c *DBClient) InsertContainerEventJfrog(pushEvent model.JfrogContainerPushE
293287
func (c *DBClient) InsertRakeesMetrics(metrics model.RakeesMetrics) {
294288
tx, err := c.conn.Begin()
295289
if err != nil {
296-
log.Printf("error beginning transaction: %v", err)
297-
return
290+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
298291
}
299292
stmt, err := tx.Prepare(string(InsertRakees))
300293
if err != nil {
301-
log.Printf("error preparing statement: %v", err)
302-
return
294+
log.Fatalf("error preparing statement: %v", err)
303295
}
304296

305297
defer stmt.Close()
@@ -325,13 +317,11 @@ func (c *DBClient) InsertRakeesMetrics(metrics model.RakeesMetrics) {
325317
func (c *DBClient) InsertKetallEvent(metrics model.Resource) {
326318
tx, err := c.conn.Begin()
327319
if err != nil {
328-
log.Printf("error beginning transaction: %v", err)
329-
return
320+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
330321
}
331322
stmt, err := tx.Prepare(string(InsertKetall))
332323
if err != nil {
333-
log.Printf("error preparing statement: %v", err)
334-
return
324+
log.Fatalf("error preparing statement: %v", err)
335325
}
336326

337327
defer stmt.Close()
@@ -356,13 +346,11 @@ func (c *DBClient) InsertKetallEvent(metrics model.Resource) {
356346
func (c *DBClient) InsertOutdatedEvent(metrics model.CheckResultfinal) {
357347
tx, err := c.conn.Begin()
358348
if err != nil {
359-
log.Printf("error beginning transaction: %v", err)
360-
return
349+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
361350
}
362351
stmt, err := tx.Prepare(string(InsertOutdated))
363352
if err != nil {
364-
log.Printf("error preparing statement: %v", err)
365-
return
353+
log.Fatalf("error preparing statement: %v", err)
366354
}
367355

368356
defer stmt.Close()
@@ -389,13 +377,11 @@ func (c *DBClient) InsertOutdatedEvent(metrics model.CheckResultfinal) {
389377
func (c *DBClient) InsertDeprecatedAPI(deprecatedAPI model.DeprecatedAPI) {
390378
tx, err := c.conn.Begin()
391379
if err != nil {
392-
log.Printf("error beginning transaction: %v", err)
393-
return
380+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
394381
}
395382
stmt, err := tx.Prepare(string(InsertDepricatedApi))
396383
if err != nil {
397-
log.Printf("error preparing statement: %v", err)
398-
return
384+
log.Fatalf("error preparing statement: %v", err)
399385
}
400386

401387
defer stmt.Close()
@@ -429,13 +415,11 @@ func (c *DBClient) InsertDeprecatedAPI(deprecatedAPI model.DeprecatedAPI) {
429415
func (c *DBClient) InsertDeletedAPI(deletedAPI model.DeletedAPI) {
430416
tx, err := c.conn.Begin()
431417
if err != nil {
432-
log.Printf("error beginning transaction: %v", err)
433-
return
418+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
434419
}
435420
stmt, err := tx.Prepare(string(InsertDeletedApi))
436421
if err != nil {
437-
log.Printf("error preparing statement: %v", err)
438-
return
422+
log.Fatalf("error preparing statement: %v", err)
439423
}
440424

441425
defer stmt.Close()
@@ -470,13 +454,11 @@ func (c *DBClient) InsertDeletedAPI(deletedAPI model.DeletedAPI) {
470454
func (c *DBClient) InsertKubvizEvent(metrics model.Metrics) {
471455
tx, err := c.conn.Begin()
472456
if err != nil {
473-
log.Printf("error beginning transaction: %v", err)
474-
return
457+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
475458
}
476459
stmt, err := tx.Prepare(string(InsertKubvizEvent))
477460
if err != nil {
478-
log.Printf("error preparing statement: %v", err)
479-
return
461+
log.Fatalf("error preparing statement: %v", err)
480462
}
481463

482464
defer stmt.Close()
@@ -539,13 +521,11 @@ func (c *DBClient) InsertContainerEvent(event string) {
539521
func (c *DBClient) InsertKubeScoreMetrics(metrics model.KubeScoreRecommendations) {
540522
tx, err := c.conn.Begin()
541523
if err != nil {
542-
log.Printf("error beginning transaction: %v", err)
543-
return
524+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
544525
}
545526
stmt, err := tx.Prepare(InsertKubeScore)
546527
if err != nil {
547-
log.Printf("error preparing statement: %v", err)
548-
return
528+
log.Fatalf("error preparing statement: %v", err)
549529
}
550530
defer stmt.Close()
551531

@@ -571,13 +551,11 @@ func (c *DBClient) InsertTrivyMetrics(metrics model.Trivy) {
571551
for _, vulnerability := range result.Vulnerabilities {
572552
tx, err := c.conn.Begin()
573553
if err != nil {
574-
log.Printf("error beginning transaction: %v", err)
575-
return
554+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
576555
}
577556
stmt, err := tx.Prepare(InsertTrivyVul)
578557
if err != nil {
579-
log.Printf("error preparing statement: %v", err)
580-
return
558+
log.Fatalf("error preparing statement: %v", err)
581559
}
582560
if _, err := stmt.Exec(
583561
metrics.ID,
@@ -608,13 +586,11 @@ func (c *DBClient) InsertTrivyMetrics(metrics model.Trivy) {
608586
for _, misconfiguration := range result.Misconfigurations {
609587
tx, err := c.conn.Begin()
610588
if err != nil {
611-
log.Printf("error beginning transaction: %v", err)
612-
return
589+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
613590
}
614591
stmt, err := tx.Prepare(InsertTrivyMisconfig)
615592
if err != nil {
616-
log.Printf("error preparing statement: %v", err)
617-
return
593+
log.Fatalf("error preparing statement: %v", err)
618594
}
619595

620596
defer stmt.Close()
@@ -654,13 +630,11 @@ func (c *DBClient) InsertTrivyImageMetrics(metrics model.TrivyImage) {
654630
for _, vulnerability := range result.Vulnerabilities {
655631
tx, err := c.conn.Begin()
656632
if err != nil {
657-
log.Printf("error beginning transaction: %v", err)
658-
return
633+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
659634
}
660635
stmt, err := tx.Prepare(InsertTrivyImage)
661636
if err != nil {
662-
log.Printf("error preparing statement: %v", err)
663-
return
637+
log.Fatalf("error preparing statement: %v", err)
664638
}
665639

666640
if _, err := stmt.Exec(
@@ -700,13 +674,11 @@ func (c *DBClient) InsertTrivySbomMetrics(metrics model.Sbom) {
700674
if result.CycloneDX != nil {
701675
tx, err := c.conn.Begin()
702676
if err != nil {
703-
log.Printf("error beginning transaction: %v", err)
704-
return
677+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
705678
}
706679
stmt, err := tx.Prepare(InsertTrivySbom)
707680
if err != nil {
708-
log.Printf("error preparing statement: %v", err)
709-
return
681+
log.Fatalf("error preparing statement: %v", err)
710682
}
711683

712684
if _, err := stmt.Exec(
@@ -834,14 +806,12 @@ func (c *DBClient) RetrieveKubvizEvent() ([]model.DbEvent, error) {
834806
func (c *DBClient) InsertContainerEventDockerHub(build model.DockerHubBuild) {
835807
tx, err := c.conn.Begin()
836808
if err != nil {
837-
log.Printf("error beginning transaction: %v", err)
838-
return
809+
log.Fatalf("error beginning transaction, clickhouse connection not available: %v", err)
839810
}
840811
stmt, err := tx.Prepare(string(InsertDockerHubBuild))
841812

842813
if err != nil {
843-
log.Printf("error preparing statement: %v", err)
844-
return
814+
log.Fatalf("error preparing statement: %v", err)
845815
}
846816
defer stmt.Close()
847817

0 commit comments

Comments
 (0)