diff options
| author | Jonathan Amsterdam <jba@google.com> | 2020-04-15 16:57:51 -0400 |
|---|---|---|
| committer | Jonathan Amsterdam <jba@google.com> | 2020-04-16 14:10:02 +0000 |
| commit | 7d2a6cfadd9231e0ab26e8e05ce1431837c49dac (patch) | |
| tree | 722eb8e44c646e3958639f95864b2300ba7d01b8 /internal/postgres/versionstate.go | |
| parent | f3d7078125541e8902382be09e4a770414d90a59 (diff) | |
| download | go-x-pkgsite-7d2a6cfadd9231e0ab26e8e05ce1431837c49dac.tar.xz | |
internal/database: log transaction start and end
This requires adding a context parameter to the Transact method.
Change-Id: I98e8b9cbd8ce724cbed693d7549e82cf8ee75990
Reviewed-on: https://team-review.git.corp.google.com/c/golang/discovery/+/719480
Reviewed-by: Julie Qiu <julieqiu@google.com>
Diffstat (limited to 'internal/postgres/versionstate.go')
| -rw-r--r-- | internal/postgres/versionstate.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/internal/postgres/versionstate.go b/internal/postgres/versionstate.go index 3caa5578..423a425f 100644 --- a/internal/postgres/versionstate.go +++ b/internal/postgres/versionstate.go @@ -38,7 +38,7 @@ func (db *DB) InsertIndexVersions(ctx context.Context, versions []*internal.Inde DO UPDATE SET index_timestamp=excluded.index_timestamp, next_processed_after=CURRENT_TIMESTAMP` - return db.db.Transact(func(tx *database.DB) error { + return db.db.Transact(ctx, func(tx *database.DB) error { return tx.BulkInsert(ctx, "module_version_states", cols, vals, conflictAction) }) } @@ -52,7 +52,7 @@ func (db *DB) UpsertModuleVersionState(ctx context.Context, modulePath, vers, ap ctx, span := trace.StartSpan(ctx, "UpsertModuleVersionState") defer span.End() - return db.db.Transact(func(tx *database.DB) error { + return db.db.Transact(ctx, func(tx *database.DB) error { var sqlErrorMsg string if fetchErr != nil { sqlErrorMsg = fetchErr.Error() |