Backport the association option `disable_joins` from Rails main, and
allow it to take a lambda as argument, so the behavior can be
controlled by a feature flag.