return leak id in the insert

palavrapasse · Jan 23, 2023 · 5e2e1b3 · 5e2e1b3
1 parent e805ef3
commit 5e2e1b3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/database/database.go b/pkg/database/database.go
@@ -55,7 +55,7 @@ func (ctx DatabaseContext[R]) NewTransactionContext() (TransactionContext[R], er
 	return TransactionContext[R]{Tx: tx}, err
 }
 
-func (ctx DatabaseContext[Record]) Insert(i Import) error {
+func (ctx DatabaseContext[Record]) Insert(i Import) (AutoGenKey, error) {
 
 	var tx *sql.Tx
 
@@ -75,6 +75,8 @@ func (ctx DatabaseContext[Record]) Insert(i Import) error {
 		}
 	}()
 
+	var leakId AutoGenKey
+
 	func() {
 		us := make([]User, len(i.AffectedUsers))
 		cr := make([]Credentials, len(i.AffectedUsers))
@@ -125,6 +127,7 @@ func (ctx DatabaseContext[Record]) Insert(i Import) error {
 		ps := pts[4].(PrimaryTable[Platform]).Records
 
 		l := ls[0]
+		leakId = l.LeakId
 		afu := map[User]Credentials{}
 
 		for k := range us {
@@ -166,7 +169,7 @@ func (ctx DatabaseContext[Record]) Insert(i Import) error {
 		err = tx.Commit()
 	}
 
-	return err
+	return leakId, err
 }
 
 func (ctx DatabaseContext[Record]) InsertSubscription(s Subscription) error {