Method EnsureConn

internal/engine/postgresql/analyzer/analyze.go:487–521

EnsureConn initializes the database connection pool if not already done. This is useful for database-only mode where we need to connect before analyzing queries.

(ctx context.Context, migrations []string)

// EnsureConn initializes the database connection pool if not already done.
// This is useful for database-only mode where we need to connect before analyzing queries.
func (a *Analyzer) EnsureConn(ctx context.Context, migrations []string) error {
	if a.pool != nil {
		return nil
	}

	var uri string
	if a.db.Managed {
		if a.client == nil {
			return fmt.Errorf("client is nil")
		}
		edb, err := a.client.CreateDatabase(ctx, &dbmanager.CreateDatabaseRequest{
			Engine:     "postgresql",
			Migrations: migrations,
		})
		if err != nil {
			return err
		}
		uri = edb.Uri
	} else if a.dbg.OnlyManagedDatabases {
		return fmt.Errorf("database: connections disabled via SQLCDEBUG=databases=managed")
	} else {
		uri = a.replacer.Replace(a.db.URI)
	}

	conf, err := pgxpool.ParseConfig(uri)
	if err != nil {
		return err
	}
	pool, err := pgxpool.NewWithConfig(ctx, conf)
	if err != nil {
		return err
	}
	a.pool = pool
	return nil
}

// GetColumnNames implements the expander.ColumnGetter interface.
// It prepares a query and returns the column names from the result set description.