db.go

package opao

import (
	"database/sql"
	"errors"
	"reflect"

	"github.com/OblivionOcean/opao/support"
	"github.com/OblivionOcean/opao/support/mysql"
	"github.com/OblivionOcean/opao/support/pg"
	"github.com/OblivionOcean/opao/support/sqlite"
)

const (
    EmptyQ = ""
)

type Database struct {
	Conn          *sql.DB
	sqlDriverName string
	support.ORM
}

func NewDatabase(sqlDriverName, linkInfo string) (*Database, error) {
	conn, err := sql.Open(sqlDriverName, linkInfo)
	if err != nil {
		return nil, err
	}
	db := &Database{Conn: conn}
	db.sqlDriverName = sqlDriverName
	db.ORM = support.ORM{}
	var driver func(*sql.DB, any, reflect.Type, string, []support.Elem, error) support.ObjectORM

	switch db.sqlDriverName {
	case "mysql":
		driver = mysql.NewMySQL
	case "postgres", "pg", "pgsql":
		driver = pg.NewPg
	case "sqlite3", "sqlite":
		driver = sqlite.NewSqlite
	default:
		return nil, errors.New("driver not supported")
	}

	db.ORM.Init(db.Conn, driver)

	return db, nil
}

func (db *Database) Close() error {
	if db.Conn == nil {
		return errors.New("database is not initialized")
	}
	return db.Conn.Close()
}

func (db *Database) GetConn() *sql.DB {
	return db.Conn
}