Add push server component template

pushserver/storage/interface.go

@@ -0,0 +1,7 @@
+package storage
+
+import "github.com/matrix-org/dendrite/internal"
+
+type Database interface {
+	internal.PartitionStorer
+}

pushserver/storage/postgres/storage.go

@@ -0,0 +1,31 @@
+package postgres
+
+import (
+	"fmt"
+
+	"github.com/matrix-org/dendrite/internal/sqlutil"
+	"github.com/matrix-org/dendrite/pushserver/storage/shared"
+	"github.com/matrix-org/dendrite/setup/config"
+)
+
+type Database struct {
+	shared.Database
+	sqlutil.PartitionOffsetStatements
+}
+
+func Open(dbProperties *config.DatabaseOptions) (*Database, error) {
+	var d Database
+	var err error
+	if d.DB, err = sqlutil.Open(dbProperties); err != nil {
+		return nil, fmt.Errorf("sqlutil.Open: %w", err)
+	}
+	d.Writer = sqlutil.NewDummyWriter()
+
+	if err = d.PartitionOffsetStatements.Prepare(d.DB, d.Writer, "pushserver"); err != nil {
+		return nil, err
+	}
+
+	// Create the tables.
+
+	return &d, nil
+}

pushserver/storage/shared/storage.go

@@ -0,0 +1,12 @@
+package shared
+
+import (
+	"database/sql"
+
+	"github.com/matrix-org/dendrite/internal/sqlutil"
+)
+
+type Database struct {
+	DB     *sql.DB
+	Writer sqlutil.Writer
+}

pushserver/storage/sqlite3/storage.go

@@ -0,0 +1,31 @@
+package sqlite3
+
+import (
+	"fmt"
+
+	"github.com/matrix-org/dendrite/internal/sqlutil"
+	"github.com/matrix-org/dendrite/pushserver/storage/shared"
+	"github.com/matrix-org/dendrite/setup/config"
+)
+
+type Database struct {
+	shared.Database
+	sqlutil.PartitionOffsetStatements
+}
+
+func Open(dbProperties *config.DatabaseOptions) (*Database, error) {
+	var d Database
+	var err error
+	if d.DB, err = sqlutil.Open(dbProperties); err != nil {
+		return nil, fmt.Errorf("sqlutil.Open: %w", err)
+	}
+	d.Writer = sqlutil.NewExclusiveWriter()
+
+	if err = d.PartitionOffsetStatements.Prepare(d.DB, d.Writer, "pushserver"); err != nil {
+		return nil, err
+	}
+
+	// Create the tables.
+
+	return &d, nil
+}

pushserver/storage/storage.go

@@ -0,0 +1,23 @@
+// +build !wasm
+
+package storage
+
+import (
+	"fmt"
+
+	"github.com/matrix-org/dendrite/pushserver/storage/postgres"
+	"github.com/matrix-org/dendrite/pushserver/storage/sqlite3"
+	"github.com/matrix-org/dendrite/setup/config"
+)
+
+// Open opens a database connection.
+func Open(dbProperties *config.DatabaseOptions) (Database, error) {
+	switch {
+	case dbProperties.ConnectionString.IsSQLite():
+		return sqlite3.Open(dbProperties)
+	case dbProperties.ConnectionString.IsPostgres():
+		return postgres.Open(dbProperties)
+	default:
+		return nil, fmt.Errorf("unexpected database type")
+	}
+}

pushserver/storage/storage_wasm.go

@@ -0,0 +1,20 @@
+package storage
+
+import (
+	"fmt"
+
+	"github.com/matrix-org/dendrite/pushserver/storage/sqlite3"
+	"github.com/matrix-org/dendrite/setup/config"
+)
+
+// NewDatabase opens a new database
+func Open(dbProperties *config.DatabaseOptions) (Database, error) {
+	switch {
+	case dbProperties.ConnectionString.IsSQLite():
+		return sqlite3.Open(dbProperties)
+	case dbProperties.ConnectionString.IsPostgres():
+		return nil, fmt.Errorf("can't use Postgres implementation")
+	default:
+		return nil, fmt.Errorf("unexpected database type")
+	}
+}

pushserver/storage/tables/interface.go

@@ -0,0 +1 @@
+package tables