feat: bootstrap foundation application
This commit is contained in:
79
apps/server/internal/database/database.go
Normal file
79
apps/server/internal/database/database.go
Normal file
@@ -0,0 +1,79 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
libsql "github.com/tursodatabase/go-libsql"
|
||||
|
||||
"github.com/tim/md-hub-secure/apps/server/internal/config"
|
||||
)
|
||||
|
||||
type DB interface {
|
||||
SQL() *sql.DB
|
||||
Close() error
|
||||
}
|
||||
|
||||
type dbHandle struct {
|
||||
sql *sql.DB
|
||||
}
|
||||
|
||||
func Open(ctx context.Context, cfg config.DatabaseConfig) (DB, error) {
|
||||
if err := os.MkdirAll(filepath.Dir(cfg.Path), 0o755); err != nil {
|
||||
return nil, fmt.Errorf("create database directory: %w", err)
|
||||
}
|
||||
|
||||
var (
|
||||
db *sql.DB
|
||||
err error
|
||||
)
|
||||
if cfg.PrimaryURL == "" {
|
||||
db, err = sql.Open("libsql", "file:"+cfg.Path)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("open local libsql database: %w", err)
|
||||
}
|
||||
} else {
|
||||
var opts []libsql.Option
|
||||
if cfg.AuthToken != "" {
|
||||
opts = append(opts, libsql.WithAuthToken(cfg.AuthToken))
|
||||
}
|
||||
|
||||
connector, err := libsql.NewEmbeddedReplicaConnector(cfg.Path, cfg.PrimaryURL, opts...)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("create libsql connector: %w", err)
|
||||
}
|
||||
|
||||
db = sql.OpenDB(connector)
|
||||
}
|
||||
|
||||
db.SetConnMaxLifetime(5 * time.Minute)
|
||||
db.SetMaxOpenConns(1)
|
||||
db.SetMaxIdleConns(1)
|
||||
|
||||
if err := db.PingContext(ctx); err != nil {
|
||||
return nil, fmt.Errorf("ping database: %w", err)
|
||||
}
|
||||
|
||||
var journalMode string
|
||||
if err := db.QueryRowContext(ctx, "PRAGMA journal_mode=WAL").Scan(&journalMode); err != nil {
|
||||
return nil, fmt.Errorf("enable wal mode: %w", err)
|
||||
}
|
||||
|
||||
if _, err := db.ExecContext(ctx, "PRAGMA foreign_keys=ON"); err != nil {
|
||||
return nil, fmt.Errorf("enable foreign keys: %w", err)
|
||||
}
|
||||
|
||||
return &dbHandle{sql: db}, nil
|
||||
}
|
||||
|
||||
func (d *dbHandle) SQL() *sql.DB {
|
||||
return d.sql
|
||||
}
|
||||
|
||||
func (d *dbHandle) Close() error {
|
||||
return d.sql.Close()
|
||||
}
|
||||
Reference in New Issue
Block a user