Function updateMigrationTracking

scripts/develop/dbrecovery.go:457–487

updateMigrationTracking connects to the running server's database and captures current migration state. Called after the server health check passes.

(ctx context.Context, _ slog.Logger, cfg *devConfig)

// database and captures current migration state. Called after
// the server health check passes.
func updateMigrationTracking(ctx context.Context, _ slog.Logger, cfg *devConfig) error {
	pgURL := os.Getenv("CODER_PG_CONNECTION_URL")
	if pgURL == "" {
		var err error
		pgURL, err = builtinPostgresURL(cfg)
		if err != nil {
			return xerrors.Errorf("resolve builtin postgres URL: %w", err)
		}
	}

	db, err := connectDB(ctx, pgURL)
	if err != nil {
		return xerrors.Errorf("connect for tracking update: %w", err)
	}
	defer db.Close()

	if _, err := db.ExecContext(ctx, trackingDDL); err != nil {
		return xerrors.Errorf("ensure tracking table: %w", err)
	}

	dbVersion, _, err := currentMigrationVersion(ctx, db)
	if err != nil {
		return xerrors.Errorf("get db version: %w", err)
	}
	if dbVersion < 0 {
		return nil
	}

	migrDir := filepath.Join(cfg.projectRoot, "coderd", "database", "migrations")
	return captureDownSQL(ctx, db, migrDir, dbVersion)
}

func builtinPostgresURL(cfg *devConfig) (string, error) {
	pgDir := filepath.Join(cfg.configDir, "postgres")