Skip to content

Commit

Permalink
make vulnerability node_id more unique
Browse files Browse the repository at this point in the history
  • Loading branch information
Harshvardhan Karn committed Mar 28, 2024
1 parent 64f943f commit d5ea76d
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions deepfence_worker/ingesters/vulnerabilites.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,7 @@ func CommitFuncVulnerabilities(ctx context.Context, ns string, data []ingestersU
}

session := driver.NewSession(ctx, neo4j.SessionConfig{AccessMode: neo4j.AccessModeWrite})
if err != nil {
return err
}

defer session.Close(ctx)

tx, err := session.BeginTransaction(ctx, neo4j.WithTxTimeout(30*time.Second))
Expand All @@ -40,6 +38,8 @@ func CommitFuncVulnerabilities(ctx context.Context, ns string, data []ingestersU
return err
}

log.Debug().Msgf("Committing %d vulnerabilities", len(dataMap))

if _, err = tx.Run(ctx, `
UNWIND $batch as row WITH row.rule as rule, row.data as data,
row.scan_id as scan_id, row.node_id as node_id
Expand Down Expand Up @@ -73,7 +73,7 @@ func CVEsToMaps(ms []ingestersUtil.Vulnerability) ([]map[string]interface{}, err
"rule": utils.ToMap(rule),
"data": utils.ToMap(data),
"scan_id": v.ScanID,
"node_id": data.CveCausedByPackage + rule.CveID,
"node_id": data.CveCausedByPackagePath + data.CveCausedByPackage + rule.CveID,
})
}
return res, nil
Expand Down

0 comments on commit d5ea76d

Please sign in to comment.