Adjust urn update

weni-ai · Oct 21, 2024 · 24bfdfd · 24bfdfd
1 parent 3ec20d8
commit 24bfdfd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/backends/rapidpro/contact.go b/backends/rapidpro/contact.go
@@ -133,7 +133,7 @@ func contactForURNTeams(ctx context.Context, b *backend, urn urns.URN, org OrgID
 		return contact, err
 	}
 
-	err = updateContactTeamsURN(b.db, contact.URNID_, urn.Identity().String())
+	err = updateContactTeamsURN(ctx, b.db, contact.URNID_, urn.Identity().String())
 	if err != nil {
 		logrus.WithError(err).WithField("urn", urn.Identity()).WithField("org_id", org).Error("error updating contact urn")
 		return contact, err

diff --git a/backends/rapidpro/urn.go b/backends/rapidpro/urn.go
@@ -1,6 +1,7 @@
 package rapidpro
 
 import (
+	"context"
 	"database/sql"
 	"database/sql/driver"
 	"fmt"
@@ -333,8 +334,8 @@ func fullyUpdateContactURN(db *sqlx.Tx, urn *DBContactURN) error {
 	return err
 }
 
-func updateContactTeamsURN(db *sqlx.DB, urnID ContactURNID, newURN string) error {
-	_, err := db.Exec(updateTeamsURN, newURN, urnID)
+func updateContactTeamsURN(ctx context.Context, db *sqlx.DB, urnID ContactURNID, newURN string) error {
+	_, err := db.ExecContext(ctx, updateTeamsURN, newURN, urnID)
 	if err != nil {
 		logrus.WithError(err).WithField("urn_id", urnID).Error("error updating contact urn")
 		return err