How to specify different retry policy for different exceptions in Spring-core 7.

Something like below was possible in Spring-core 6 and below.

final SimpleRetryPolicy retryWithMaxAttempts = new SimpleRetryPolicy(3);
final AlwaysRetryPolicy connectionRetryPolicy = new AlwaysRetryPolicy();
ExceptionCauseClassifierRetryPolicy exceptionClassifierRetryPolicy  = new ExceptionCauseClassifierRetryPolicy();
final Map, RetryPolicy> policyMap = new HashMap<>();
policyMap.put(SQLException.class, retryWithMaxAttempts);
policyMap.put(SocketException.class, connectionRetryPolicy);
exceptionClassifierRetryPolicy.setPolicyMap(policyMap);
return exceptionClassifierRetryPolicy;

Now it looks like totally messed up in Spring-core 7. Any help or suggestion?