mautrix-discord/database/database.go

package database

import (
	"database/sql"

	_ "github.com/lib/pq"
	_ "github.com/mattn/go-sqlite3"

	log "maunium.net/go/maulogger/v2"

	"gitlab.com/beeper/discord/database/migrations"
)

type Database struct {
	*sql.DB
	log     log.Logger
	dialect string

	User       *UserQuery
	Portal     *PortalQuery
	Puppet     *PuppetQuery
	Message    *MessageQuery
	Reaction   *ReactionQuery
	Attachment *AttachmentQuery
}

func New(dbType, uri string, maxOpenConns, maxIdleConns int, baseLog log.Logger) (*Database, error) {
	conn, err := sql.Open(dbType, uri)
	if err != nil {
		return nil, err
	}

	if dbType == "sqlite3" {
		conn.Exec("PRAGMA foreign_keys = ON")
	}

	conn.SetMaxOpenConns(maxOpenConns)
	conn.SetMaxIdleConns(maxIdleConns)

	dbLog := baseLog.Sub("Database")

	if err := migrations.Run(conn, dbLog); err != nil {
		return nil, err
	}

	db := &Database{
		DB:      conn,
		log:     dbLog,
		dialect: dbType,
	}

	db.User = &UserQuery{
		db:  db,
		log: db.log.Sub("User"),
	}

	db.Portal = &PortalQuery{
		db:  db,
		log: db.log.Sub("Portal"),
	}

	db.Puppet = &PuppetQuery{
		db:  db,
		log: db.log.Sub("Puppet"),
	}

	db.Message = &MessageQuery{
		db:  db,
		log: db.log.Sub("Message"),
	}

	db.Reaction = &ReactionQuery{
		db:  db,
		log: db.log.Sub("Reaction"),
	}

	db.Attachment = &AttachmentQuery{
		db:  db,
		log: db.log.Sub("Attachment"),
	}

	return db, nil
}