Method EnsureConn

internal/engine/sqlite/analyzer/analyze.go:188–232

EnsureConn initializes the database connection 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 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 {
	a.mu.Lock()
	defer a.mu.Unlock()

	if a.conn != nil {
		return nil
	}

	var uri string
	applyMigrations := a.db.Managed
	if a.db.Managed {
		// For managed databases, create an in-memory database
		uri = ":memory:"
	} else if a.dbg.OnlyManagedDatabases {
		return fmt.Errorf("database: connections disabled via SQLCDEBUG=databases=managed")
	} else {
		uri = a.replacer.Replace(a.db.URI)
		// For in-memory databases, we need to apply migrations since the database starts empty
		if isInMemoryDatabase(uri) {
			applyMigrations = true
		}
	}

	conn, err := sqlite3.Open(uri)
	if err != nil {
		return fmt.Errorf("failed to open sqlite database: %w", err)
	}
	a.conn = conn

	// Apply migrations for managed or in-memory databases
	if applyMigrations {
		for _, m := range migrations {
			if len(strings.TrimSpace(m)) == 0 {
				continue
			}
			if err := a.conn.Exec(m); err != nil {
				a.conn.Close()
				a.conn = nil
				return fmt.Errorf("migration failed: %s: %w", m, err)
			}
		}
	}

	return nil
}

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