Function GetPostgresClient

pkg/database/database.go:21–52

GetPostgresClient initializes or returns a database connection for a given DSN

(dsn string)

// GetPostgresClient initializes or returns a database connection for a given DSN
func GetPostgresClient(dsn string) (*gorm.DB, error) {
	mu.Lock()
	defer mu.Unlock()

	// Check if an instance already exists for this DSN
	if db, exists := instances[dsn]; exists {
		return db, nil
	}

	// Create a new instance since one doesn't exist
	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info), // Set log mode to `Info` to log all SQL queries
	})
	if err != nil {
		log.Printf("failed to connect to database: %v", err)
		return nil, err
	}

	// Execute initial setup queries
	if result := db.Exec("CREATE EXTENSION IF NOT EXISTS vector"); result.Error != nil {
		log.Printf("Failed to execute initial setup query: %v", result.Error)
		return nil, result.Error
	}

	// Auto-migrate the schema
	db.AutoMigrate(&Object{}, &Reference{}, &Config{}, &Shallow{}, &Index{}, &CodeEmbedding{})

	// Store the instance in the map
	instances[dsn] = db

	return db, nil
}

// ConstructPostgresDSN constructs the Data Source Name for a PostgreSQL connection
func ConstructPostgresDSN(host, user, password, dbname, port, sslmode, timezone string) string {