fn-registry agent
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"fn-registry/functions/infra"
|
|
)
|
|
|
|
// persistCopiedCode runs infra.AuditCopiedCode and INSERT OR IGNOREs results
|
|
// into the copied_code table. Returns (newly_inserted, total_detected, error).
|
|
func persistCopiedCode(callDB *DB, registryRoot string) (int, int, error) {
|
|
entries, err := infra.AuditCopiedCode(registryRoot)
|
|
if err != nil {
|
|
return 0, 0, fmt.Errorf("audit: %w", err)
|
|
}
|
|
now := time.Now().UTC().Unix()
|
|
tx, err := callDB.conn.Begin()
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
stmt, err := tx.Prepare(`INSERT OR IGNORE INTO copied_code
|
|
(app_file, app_function, registry_id, body_hash, similarity, kind, detected_at)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)`)
|
|
if err != nil {
|
|
_ = tx.Rollback()
|
|
return 0, 0, err
|
|
}
|
|
defer stmt.Close()
|
|
inserted := 0
|
|
for _, e := range entries {
|
|
res, err := stmt.Exec(e.AppFile, e.AppFunction, e.RegistryID, e.BodyHash, e.Similarity, e.Kind, now)
|
|
if err != nil {
|
|
_ = tx.Rollback()
|
|
return 0, 0, err
|
|
}
|
|
if n, _ := res.RowsAffected(); n > 0 {
|
|
inserted++
|
|
}
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return 0, 0, err
|
|
}
|
|
return inserted, len(entries), nil
|
|
}
|