mirror of
https://github.com/matrix-org/dendrite.git
synced 2024-11-27 00:31:55 -06:00
5caae6f3a0
* bugfix: fix panic on new invite events from sytest I'm unsure why the previous code didn't work, but it's clearer, quicker and easier to read the `LastInsertID()` way. Previously, the code would panic as the SELECT would fail to find the last inserted row ID. * sqlite: Fix UNIQUE violations and close more cursors - Add missing `defer rows.Close()` - Do not have the state block NID as a PRIMARY KEY else it breaks for blocks with >1 state event in them. Instead, rejig the queries so we can still have monotonically increasing integers without using AUTOINCREMENT (which mandates PRIMARY KEY). * sqlite: Add missing variadic function * Use LastInsertId because empirically it works over the SELECT form (though I don't know why that is) * sqlite: Fix invite table by using the global stream pos rather than one specific to invites If we don't use the global, clients don't get notified about any invites because the position is too low. * linting: shadowing * sqlite: do not use last rowid, we already know the stream pos! * sqlite: Fix account data table in syncapi by commiting insert txns! * sqlite: Fix failing federation invite Was failing with 'database is locked' due to multiple write txns being taken out. * sqlite: Ensure we return exactly the number of events found in the database Previously we would return exactly the number of *requested* events, which meant that several zero-initialised events would bubble through the system, failing at JSON serialisation time. * sqlite: let's just ignore the problem for now.... * linting
138 lines
3.8 KiB
Go
138 lines
3.8 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 sqlite3
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/matrix-org/dendrite/common"
|
|
)
|
|
|
|
const roomAliasesSchema = `
|
|
CREATE TABLE IF NOT EXISTS roomserver_room_aliases (
|
|
alias TEXT NOT NULL PRIMARY KEY,
|
|
room_id TEXT NOT NULL,
|
|
creator_id TEXT NOT NULL
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS roomserver_room_id_idx ON roomserver_room_aliases(room_id);
|
|
`
|
|
|
|
const insertRoomAliasSQL = `
|
|
INSERT INTO roomserver_room_aliases (alias, room_id, creator_id) VALUES ($1, $2, $3)
|
|
`
|
|
|
|
const selectRoomIDFromAliasSQL = `
|
|
SELECT room_id FROM roomserver_room_aliases WHERE alias = $1
|
|
`
|
|
|
|
const selectAliasesFromRoomIDSQL = `
|
|
SELECT alias FROM roomserver_room_aliases WHERE room_id = $1
|
|
`
|
|
|
|
const selectCreatorIDFromAliasSQL = `
|
|
SELECT creator_id FROM roomserver_room_aliases WHERE alias = $1
|
|
`
|
|
|
|
const deleteRoomAliasSQL = `
|
|
DELETE FROM roomserver_room_aliases WHERE alias = $1
|
|
`
|
|
|
|
type roomAliasesStatements struct {
|
|
insertRoomAliasStmt *sql.Stmt
|
|
selectRoomIDFromAliasStmt *sql.Stmt
|
|
selectAliasesFromRoomIDStmt *sql.Stmt
|
|
selectCreatorIDFromAliasStmt *sql.Stmt
|
|
deleteRoomAliasStmt *sql.Stmt
|
|
}
|
|
|
|
func (s *roomAliasesStatements) prepare(db *sql.DB) (err error) {
|
|
_, err = db.Exec(roomAliasesSchema)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return statementList{
|
|
{&s.insertRoomAliasStmt, insertRoomAliasSQL},
|
|
{&s.selectRoomIDFromAliasStmt, selectRoomIDFromAliasSQL},
|
|
{&s.selectAliasesFromRoomIDStmt, selectAliasesFromRoomIDSQL},
|
|
{&s.selectCreatorIDFromAliasStmt, selectCreatorIDFromAliasSQL},
|
|
{&s.deleteRoomAliasStmt, deleteRoomAliasSQL},
|
|
}.prepare(db)
|
|
}
|
|
|
|
func (s *roomAliasesStatements) insertRoomAlias(
|
|
ctx context.Context, txn *sql.Tx, alias string, roomID string, creatorUserID string,
|
|
) (err error) {
|
|
insertStmt := common.TxStmt(txn, s.insertRoomAliasStmt)
|
|
_, err = insertStmt.ExecContext(ctx, alias, roomID, creatorUserID)
|
|
return
|
|
}
|
|
|
|
func (s *roomAliasesStatements) selectRoomIDFromAlias(
|
|
ctx context.Context, txn *sql.Tx, alias string,
|
|
) (roomID string, err error) {
|
|
selectStmt := common.TxStmt(txn, s.selectRoomIDFromAliasStmt)
|
|
err = selectStmt.QueryRowContext(ctx, alias).Scan(&roomID)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *roomAliasesStatements) selectAliasesFromRoomID(
|
|
ctx context.Context, txn *sql.Tx, roomID string,
|
|
) (aliases []string, err error) {
|
|
aliases = []string{}
|
|
selectStmt := common.TxStmt(txn, s.selectAliasesFromRoomIDStmt)
|
|
rows, err := selectStmt.QueryContext(ctx, roomID)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
defer rows.Close() // nolint: errcheck
|
|
|
|
for rows.Next() {
|
|
var alias string
|
|
if err = rows.Scan(&alias); err != nil {
|
|
return
|
|
}
|
|
|
|
aliases = append(aliases, alias)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (s *roomAliasesStatements) selectCreatorIDFromAlias(
|
|
ctx context.Context, txn *sql.Tx, alias string,
|
|
) (creatorID string, err error) {
|
|
selectStmt := common.TxStmt(txn, s.selectCreatorIDFromAliasStmt)
|
|
err = selectStmt.QueryRowContext(ctx, alias).Scan(&creatorID)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *roomAliasesStatements) deleteRoomAlias(
|
|
ctx context.Context, txn *sql.Tx, alias string,
|
|
) (err error) {
|
|
deleteStmt := common.TxStmt(txn, s.deleteRoomAliasStmt)
|
|
_, err = deleteStmt.ExecContext(ctx, alias)
|
|
return
|
|
}
|