40 lines
961 B
Go
40 lines
961 B
Go
package postgres
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/matrix-org/dendrite/currentstateserver/storage/shared"
|
|
"github.com/matrix-org/dendrite/internal/config"
|
|
"github.com/matrix-org/dendrite/internal/sqlutil"
|
|
)
|
|
|
|
type Database struct {
|
|
shared.Database
|
|
db *sql.DB
|
|
writer sqlutil.Writer
|
|
sqlutil.PartitionOffsetStatements
|
|
}
|
|
|
|
// NewDatabase creates a new sync server database
|
|
func NewDatabase(dbProperties *config.DatabaseOptions) (*Database, error) {
|
|
var d Database
|
|
var err error
|
|
if d.db, err = sqlutil.Open(dbProperties); err != nil {
|
|
return nil, err
|
|
}
|
|
d.writer = sqlutil.NewDummyWriter()
|
|
if err = d.PartitionOffsetStatements.Prepare(d.db, d.writer, "currentstate"); err != nil {
|
|
return nil, err
|
|
}
|
|
currRoomState, err := NewPostgresCurrentRoomStateTable(d.db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
d.Database = shared.Database{
|
|
DB: d.db,
|
|
Writer: d.writer,
|
|
CurrentRoomState: currRoomState,
|
|
}
|
|
return &d, nil
|
|
}
|