diff --git a/src/github.com/matrix-org/dendrite/roomserver/storage/reserved_rooms.go b/src/github.com/matrix-org/dendrite/roomserver/storage/reserved_rooms.go new file mode 100644 index 000000000..487da3dbe --- /dev/null +++ b/src/github.com/matrix-org/dendrite/roomserver/storage/reserved_rooms.go @@ -0,0 +1,73 @@ +// Copyright 2017 Vector Creations 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 storage + +import ( + "context" + "database/sql" + "time" +) + +const reservedRoomSchema = ` +-- The events table holds metadata for each event, the actual JSON is stored +-- separately to keep the size of the rows small. +CREATE SEQUENCE IF NOT EXISTS roomserver_event_nid_seq; +CREATE TABLE IF NOT EXISTS roomserver_reserved_rooms ( + --- The Room ID --- + room_id TEXT PRIMARY KEY, + reserved_since TIMESTAMP NOT NULL +); +` + +const insertReservedRoomSQL = "" + + "INSERT INTO roomserver_reserved_rooms (room_id, reserved_since)" + + " VALUES ($1, $2)" + +const selectReservedRoomSQL = "" + + "SELECT reserved_since FROM roomserver_reserved_rooms WHERE room_id = $1" + +type reservedRoomStatements struct { + insertReservedRoomStmt *sql.Stmt + selectReservedRoomStmt *sql.Stmt +} + +func (s *reservedRoomStatements) prepare(db *sql.DB) (err error) { + _, err = db.Exec(reservedRoomSchema) + if err != nil { + return + } + + return statementList{ + {&s.insertReservedRoomStmt, insertReservedRoomSQL}, + {&s.selectReservedRoomStmt, selectReservedRoomSQL}, + }.prepare(db) +} + +func (s *reservedRoomStatements) insertReservedRoom( + ctx context.Context, + roomID string, + reservedSince time.Time, +) error { + _, err := s.insertReservedRoomStmt.ExecContext(ctx, roomID, reservedSince) + return err +} + +func (s *reservedRoomStatements) selectReservedRoom( + ctx context.Context, roomID string, +) (time.Time, error) { + var reservedSince time.Time + err := s.selectReservedRoomStmt.QueryRowContext(ctx, roomID).Scan(&reservedSince) + return reservedSince, err +}