mirror of
https://github.com/bitechdev/ResolveSpec.git
synced 2026-01-13 14:34:25 +00:00
276854768e
* Introduced NewConnectionFromDB function to create connections from existing *sql.DB instances. * Added ExistingDBProvider to wrap existing database connections for dbmanager features. * Implemented tests for NewConnectionFromDB and ExistingDBProvider functionalities.
68 lines
2.3 KiB
Go
68 lines
2.3 KiB
Go
package dbmanager
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/bitechdev/ResolveSpec/pkg/dbmanager/providers"
|
|
)
|
|
|
|
// createConnection creates a database connection based on the configuration
|
|
func createConnection(cfg ConnectionConfig) (Connection, error) {
|
|
// Validate configuration
|
|
if err := cfg.Validate(); err != nil {
|
|
return nil, fmt.Errorf("invalid connection configuration: %w", err)
|
|
}
|
|
|
|
// Create provider based on database type
|
|
provider, err := createProvider(cfg.Type)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Create connection wrapper based on database type
|
|
switch cfg.Type {
|
|
case DatabaseTypePostgreSQL, DatabaseTypeSQLite, DatabaseTypeMSSQL:
|
|
return newSQLConnection(cfg.Name, cfg.Type, cfg, provider), nil
|
|
case DatabaseTypeMongoDB:
|
|
return newMongoConnection(cfg.Name, cfg, provider), nil
|
|
default:
|
|
return nil, fmt.Errorf("%w: %s", ErrUnsupportedDatabase, cfg.Type)
|
|
}
|
|
}
|
|
|
|
// createProvider creates a database provider based on the database type
|
|
func createProvider(dbType DatabaseType) (Provider, error) {
|
|
switch dbType {
|
|
case DatabaseTypePostgreSQL:
|
|
return providers.NewPostgresProvider(), nil
|
|
case DatabaseTypeSQLite:
|
|
return providers.NewSQLiteProvider(), nil
|
|
case DatabaseTypeMSSQL:
|
|
return providers.NewMSSQLProvider(), nil
|
|
case DatabaseTypeMongoDB:
|
|
return providers.NewMongoProvider(), nil
|
|
default:
|
|
return nil, fmt.Errorf("%w: %s", ErrUnsupportedDatabase, dbType)
|
|
}
|
|
}
|
|
|
|
// Provider is an alias to the providers.Provider interface
|
|
// This allows dbmanager package consumers to use Provider without importing providers
|
|
type Provider = providers.Provider
|
|
|
|
// NewConnectionFromDB creates a new Connection from an existing *sql.DB
|
|
// This allows you to use dbmanager features (ORM wrappers, health checks, etc.)
|
|
// with a database connection that was opened outside of dbmanager
|
|
//
|
|
// Parameters:
|
|
// - name: A unique name for this connection
|
|
// - dbType: The database type (DatabaseTypePostgreSQL, DatabaseTypeSQLite, or DatabaseTypeMSSQL)
|
|
// - db: An existing *sql.DB connection
|
|
//
|
|
// Returns a Connection that wraps the existing *sql.DB
|
|
func NewConnectionFromDB(name string, dbType DatabaseType, db *sql.DB) Connection {
|
|
provider := providers.NewExistingDBProvider(db, name)
|
|
return newSQLConnection(name, dbType, ConnectionConfig{Name: name, Type: dbType}, provider)
|
|
}
|