215 lines
8.5 KiB
Go
215 lines
8.5 KiB
Go
// Copyright 2018 New Vector Ltd
|
|
//
|
|
// 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"
|
|
"encoding/json"
|
|
|
|
"github.com/matrix-org/dendrite/internal/sqlutil"
|
|
"github.com/matrix-org/dendrite/syncapi/storage/tables"
|
|
"github.com/matrix-org/dendrite/syncapi/types"
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
)
|
|
|
|
const outputRoomEventsTopologySchema = `
|
|
-- Stores output room events received from the roomserver.
|
|
CREATE TABLE IF NOT EXISTS syncapi_output_room_events_topology (
|
|
event_id TEXT PRIMARY KEY,
|
|
topological_position BIGINT NOT NULL,
|
|
stream_position BIGINT NOT NULL,
|
|
room_id TEXT NOT NULL,
|
|
|
|
UNIQUE(topological_position, room_id, stream_position)
|
|
);
|
|
-- The topological order will be used in events selection and ordering
|
|
-- CREATE UNIQUE INDEX IF NOT EXISTS syncapi_event_topological_position_idx ON syncapi_output_room_events_topology(topological_position, stream_position, room_id);
|
|
`
|
|
|
|
const insertEventInTopologySQL = "" +
|
|
"INSERT INTO syncapi_output_room_events_topology (event_id, topological_position, room_id, stream_position)" +
|
|
" VALUES ($1, $2, $3, $4)" +
|
|
" ON CONFLICT DO NOTHING"
|
|
|
|
const selectEventIDsInRangeASCSQL = "" +
|
|
"SELECT event_id FROM syncapi_output_room_events_topology" +
|
|
" WHERE room_id = $1 AND (" +
|
|
"(topological_position > $2 AND topological_position < $3) OR" +
|
|
"(topological_position = $4 AND stream_position >= $5)" +
|
|
") ORDER BY topological_position ASC, stream_position ASC LIMIT $6"
|
|
|
|
const selectEventIDsInRangeDESCSQL = "" +
|
|
"SELECT event_id FROM syncapi_output_room_events_topology" +
|
|
" WHERE room_id = $1 AND (" +
|
|
"(topological_position > $2 AND topological_position < $3) OR" +
|
|
"(topological_position = $4 AND stream_position <= $5)" +
|
|
") ORDER BY topological_position DESC, stream_position DESC LIMIT $6"
|
|
|
|
const selectPositionInTopologySQL = "" +
|
|
"SELECT topological_position, stream_position FROM syncapi_output_room_events_topology" +
|
|
" WHERE event_id = $1"
|
|
|
|
const selectMaxPositionInTopologySQL = "" +
|
|
"SELECT MAX(topological_position), stream_position FROM syncapi_output_room_events_topology" +
|
|
" WHERE room_id = $1 ORDER BY stream_position DESC"
|
|
|
|
const selectStreamToTopologicalPositionAscSQL = "" +
|
|
"SELECT topological_position FROM syncapi_output_room_events_topology WHERE room_id = $1 AND stream_position >= $2 ORDER BY topological_position ASC LIMIT 1;"
|
|
|
|
const selectStreamToTopologicalPositionDescSQL = "" +
|
|
"SELECT topological_position FROM syncapi_output_room_events_topology WHERE room_id = $1 AND stream_position <= $2 ORDER BY topological_position DESC LIMIT 1;"
|
|
|
|
const selectTopologicalEventSQL = "" +
|
|
"SELECT headered_event_json, topological_position, stream_position " +
|
|
" FROM syncapi_output_room_events_topology " +
|
|
" JOIN syncapi_output_room_events ON syncapi_output_room_events.event_id = syncapi_output_room_events_topology.event_id " +
|
|
" WHERE syncapi_output_room_events_topology.room_id = $1 AND topological_position <= $2 AND type = $3 " +
|
|
" ORDER BY topological_position DESC LIMIT 1"
|
|
|
|
type outputRoomEventsTopologyStatements struct {
|
|
db *sql.DB
|
|
insertEventInTopologyStmt *sql.Stmt
|
|
selectEventIDsInRangeASCStmt *sql.Stmt
|
|
selectEventIDsInRangeDESCStmt *sql.Stmt
|
|
selectPositionInTopologyStmt *sql.Stmt
|
|
selectMaxPositionInTopologyStmt *sql.Stmt
|
|
selectStreamToTopologicalPositionAscStmt *sql.Stmt
|
|
selectStreamToTopologicalPositionDescStmt *sql.Stmt
|
|
selectTopologicalEventStmt *sql.Stmt
|
|
}
|
|
|
|
func NewSqliteTopologyTable(db *sql.DB) (tables.Topology, error) {
|
|
s := &outputRoomEventsTopologyStatements{
|
|
db: db,
|
|
}
|
|
_, err := db.Exec(outputRoomEventsTopologySchema)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, sqlutil.StatementList{
|
|
{&s.insertEventInTopologyStmt, insertEventInTopologySQL},
|
|
{&s.selectEventIDsInRangeASCStmt, selectEventIDsInRangeASCSQL},
|
|
{&s.selectEventIDsInRangeDESCStmt, selectEventIDsInRangeDESCSQL},
|
|
{&s.selectPositionInTopologyStmt, selectPositionInTopologySQL},
|
|
{&s.selectMaxPositionInTopologyStmt, selectMaxPositionInTopologySQL},
|
|
{&s.selectStreamToTopologicalPositionAscStmt, selectStreamToTopologicalPositionAscSQL},
|
|
{&s.selectStreamToTopologicalPositionDescStmt, selectStreamToTopologicalPositionDescSQL},
|
|
{&s.selectTopologicalEventStmt, selectTopologicalEventSQL},
|
|
}.Prepare(db)
|
|
}
|
|
|
|
// insertEventInTopology inserts the given event in the room's topology, based
|
|
// on the event's depth.
|
|
func (s *outputRoomEventsTopologyStatements) InsertEventInTopology(
|
|
ctx context.Context, txn *sql.Tx, event *gomatrixserverlib.HeaderedEvent, pos types.StreamPosition,
|
|
) (types.StreamPosition, error) {
|
|
_, err := sqlutil.TxStmt(txn, s.insertEventInTopologyStmt).ExecContext(
|
|
ctx, event.EventID(), event.Depth(), event.RoomID(), pos,
|
|
)
|
|
return types.StreamPosition(event.Depth()), err
|
|
}
|
|
|
|
func (s *outputRoomEventsTopologyStatements) SelectEventIDsInRange(
|
|
ctx context.Context, txn *sql.Tx, roomID string,
|
|
minDepth, maxDepth, maxStreamPos types.StreamPosition,
|
|
limit int, chronologicalOrder bool,
|
|
) (eventIDs []string, err error) {
|
|
// Decide on the selection's order according to whether chronological order
|
|
// is requested or not.
|
|
var stmt *sql.Stmt
|
|
if chronologicalOrder {
|
|
stmt = sqlutil.TxStmt(txn, s.selectEventIDsInRangeASCStmt)
|
|
} else {
|
|
stmt = sqlutil.TxStmt(txn, s.selectEventIDsInRangeDESCStmt)
|
|
}
|
|
|
|
// Query the event IDs.
|
|
rows, err := stmt.QueryContext(ctx, roomID, minDepth, maxDepth, maxDepth, maxStreamPos, limit)
|
|
if err == sql.ErrNoRows {
|
|
// If no event matched the request, return an empty slice.
|
|
return []string{}, nil
|
|
} else if err != nil {
|
|
return
|
|
}
|
|
|
|
// Return the IDs.
|
|
var eventID string
|
|
for rows.Next() {
|
|
if err = rows.Scan(&eventID); err != nil {
|
|
return
|
|
}
|
|
eventIDs = append(eventIDs, eventID)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// selectPositionInTopology returns the position of a given event in the
|
|
// topology of the room it belongs to.
|
|
func (s *outputRoomEventsTopologyStatements) SelectPositionInTopology(
|
|
ctx context.Context, txn *sql.Tx, eventID string,
|
|
) (pos types.StreamPosition, spos types.StreamPosition, err error) {
|
|
stmt := sqlutil.TxStmt(txn, s.selectPositionInTopologyStmt)
|
|
err = stmt.QueryRowContext(ctx, eventID).Scan(&pos, &spos)
|
|
return
|
|
}
|
|
|
|
// SelectStreamToTopologicalPosition returns the closest position of a given event
|
|
// in the topology of the room it belongs to from the given stream position.
|
|
func (s *outputRoomEventsTopologyStatements) SelectStreamToTopologicalPosition(
|
|
ctx context.Context, txn *sql.Tx, roomID string, streamPos types.StreamPosition, backwardOrdering bool,
|
|
) (topoPos types.StreamPosition, err error) {
|
|
if backwardOrdering {
|
|
err = s.selectStreamToTopologicalPositionDescStmt.QueryRowContext(ctx, roomID, streamPos).Scan(&topoPos)
|
|
} else {
|
|
err = s.selectStreamToTopologicalPositionAscStmt.QueryRowContext(ctx, roomID, streamPos).Scan(&topoPos)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *outputRoomEventsTopologyStatements) SelectMaxPositionInTopology(
|
|
ctx context.Context, txn *sql.Tx, roomID string,
|
|
) (pos types.StreamPosition, spos types.StreamPosition, err error) {
|
|
stmt := sqlutil.TxStmt(txn, s.selectMaxPositionInTopologyStmt)
|
|
err = stmt.QueryRowContext(ctx, roomID).Scan(&pos, &spos)
|
|
return
|
|
}
|
|
|
|
// SelectTopologicalEvent selects an event before and including the given position by eventType and roomID. Returns the found event and the topology token.
|
|
// If not event was found, returns nil and sql.ErrNoRows.
|
|
func (s *outputRoomEventsTopologyStatements) SelectTopologicalEvent(
|
|
ctx context.Context, txn *sql.Tx, topologicalPosition int, eventType, roomID string,
|
|
) (*gomatrixserverlib.HeaderedEvent, types.TopologyToken, error) {
|
|
var (
|
|
eventBytes []byte
|
|
token types.TopologyToken
|
|
)
|
|
err := sqlutil.TxStmtContext(ctx, txn, s.selectTopologicalEventStmt).
|
|
QueryRowContext(ctx, roomID, topologicalPosition, eventType).
|
|
Scan(&eventBytes, &token.Depth, &token.PDUPosition)
|
|
|
|
if err != nil {
|
|
return nil, types.TopologyToken{}, err
|
|
}
|
|
|
|
var res *gomatrixserverlib.HeaderedEvent
|
|
if err = json.Unmarshal(eventBytes, &res); err != nil {
|
|
return nil, types.TopologyToken{}, err
|
|
}
|
|
|
|
return res, token, nil
|
|
}
|