lndhub.go/database/database.go

package database

import (
	"errors"
	"github.com/bumi/lndhub.go/database/models"
	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

const (
	Sqlite3    string = "sqlite3"
	Postgresql string = "postgres"
)

func getDbDialect(database string) (*gorm.Dialector, error) {
	var dbOpen gorm.Dialector
	var err error
	if database == Sqlite3 {
		sqliteDbUri := "./database/data.db"
		dbOpen = sqlite.Open(sqliteDbUri)
	} else if database == Postgresql {
		postgresDbUri := "host=localhost user=gorm2 password=gorm2 dbname=gorm2 port=5432 sslmode=disable"
		dbOpen = postgres.Open(postgresDbUri)
	} else {
		dbOpen = nil
		err = errors.New("non supported db dialect")
	}
	return &dbOpen, err
}

// Connect : Database connect
func Connect(database string) (*gorm.DB, error) {
	dbOpen, err := getDbDialect(database)
	if err != nil {
		return nil, err
	}

	db, err := gorm.Open(*dbOpen, &gorm.Config{})
	if err != nil {
		return nil, err
	}

	err = models.Migrate(db)
	if err != nil {
		return nil, err
	}

	return db, err
}