use writer on sqlite

This commit is contained in:
Kegan Dougal 2021-07-27 12:23:31 +01:00
parent 0747e3a5f5
commit 749369ae6a

View file

@ -414,7 +414,7 @@ func (d *Database) GetOpenIDTokenAttributes(
func (d *Database) CreateKeyBackup( func (d *Database) CreateKeyBackup(
ctx context.Context, userID, algorithm string, authData json.RawMessage, ctx context.Context, userID, algorithm string, authData json.RawMessage,
) (version string, err error) { ) (version string, err error) {
err = sqlutil.WithTransaction(d.db, func(txn *sql.Tx) error { err = d.writer.Do(d.db, nil, func(txn *sql.Tx) error {
version, err = d.keyBackups.insertKeyBackup(ctx, txn, userID, algorithm, authData) version, err = d.keyBackups.insertKeyBackup(ctx, txn, userID, algorithm, authData)
return err return err
}) })
@ -424,7 +424,7 @@ func (d *Database) CreateKeyBackup(
func (d *Database) UpdateKeyBackupAuthData( func (d *Database) UpdateKeyBackupAuthData(
ctx context.Context, userID, version string, authData json.RawMessage, ctx context.Context, userID, version string, authData json.RawMessage,
) (err error) { ) (err error) {
err = sqlutil.WithTransaction(d.db, func(txn *sql.Tx) error { err = d.writer.Do(d.db, nil, func(txn *sql.Tx) error {
return d.keyBackups.updateKeyBackupAuthData(ctx, txn, userID, version, authData) return d.keyBackups.updateKeyBackupAuthData(ctx, txn, userID, version, authData)
}) })
return return
@ -433,7 +433,7 @@ func (d *Database) UpdateKeyBackupAuthData(
func (d *Database) DeleteKeyBackup( func (d *Database) DeleteKeyBackup(
ctx context.Context, userID, version string, ctx context.Context, userID, version string,
) (exists bool, err error) { ) (exists bool, err error) {
err = sqlutil.WithTransaction(d.db, func(txn *sql.Tx) error { err = d.writer.Do(d.db, nil, func(txn *sql.Tx) error {
exists, err = d.keyBackups.deleteKeyBackup(ctx, txn, userID, version) exists, err = d.keyBackups.deleteKeyBackup(ctx, txn, userID, version)
return err return err
}) })
@ -443,7 +443,7 @@ func (d *Database) DeleteKeyBackup(
func (d *Database) GetKeyBackup( func (d *Database) GetKeyBackup(
ctx context.Context, userID, version string, ctx context.Context, userID, version string,
) (versionResult, algorithm string, authData json.RawMessage, deleted bool, err error) { ) (versionResult, algorithm string, authData json.RawMessage, deleted bool, err error) {
err = sqlutil.WithTransaction(d.db, func(txn *sql.Tx) error { err = d.writer.Do(d.db, nil, func(txn *sql.Tx) error {
versionResult, algorithm, authData, deleted, err = d.keyBackups.selectKeyBackup(ctx, txn, userID, version) versionResult, algorithm, authData, deleted, err = d.keyBackups.selectKeyBackup(ctx, txn, userID, version)
return err return err
}) })