a6f995eb45
* Move Updater structs to shared and use it for postgres * Add constructors for NewXXXUpdater and a useTxns flag In sqlite, we set useTxns=false and comment why. * Handle nil txn * Handle nil in transaction * Missed one * Close the txn at the right time * Don't close the transaction as we reuse it between calls
60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
// Copyright 2017-2018 New Vector Ltd
|
|
// Copyright 2019-2020 The Matrix.org Foundation C.I.C.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package shared
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
)
|
|
|
|
// StatementList is a list of SQL statements to prepare and a pointer to where to store the resulting prepared statement.
|
|
type StatementList []struct {
|
|
Statement **sql.Stmt
|
|
SQL string
|
|
}
|
|
|
|
// Prepare the SQL for each statement in the list and assign the result to the prepared statement.
|
|
func (s StatementList) Prepare(db *sql.DB) (err error) {
|
|
for _, statement := range s {
|
|
if *statement.Statement, err = db.Prepare(statement.SQL); err != nil {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
type transaction struct {
|
|
ctx context.Context
|
|
txn *sql.Tx
|
|
}
|
|
|
|
// Commit implements types.Transaction
|
|
func (t *transaction) Commit() error {
|
|
if t.txn == nil {
|
|
// The Updater structs can operate in useTxns=false mode. The code will still call this though.
|
|
return nil
|
|
}
|
|
return t.txn.Commit()
|
|
}
|
|
|
|
// Rollback implements types.Transaction
|
|
func (t *transaction) Rollback() error {
|
|
if t.txn == nil {
|
|
// The Updater structs can operate in useTxns=false mode. The code will still call this though.
|
|
return nil
|
|
}
|
|
return t.txn.Rollback()
|
|
}
|