|
11 | 11 | import com.azure.core.credential.AzureKeyCredential;
|
12 | 12 | import com.azure.core.credential.TokenCredential;
|
13 | 13 | import com.azure.core.http.HttpClient;
|
| 14 | +import com.azure.core.http.HttpHeaders; |
14 | 15 | import com.azure.core.http.HttpPipeline;
|
15 | 16 | import com.azure.core.http.HttpPipelineBuilder;
|
| 17 | +import com.azure.core.http.policy.AddHeadersPolicy; |
16 | 18 | import com.azure.core.http.policy.BearerTokenAuthenticationPolicy;
|
17 | 19 | import com.azure.core.http.policy.CookiePolicy;
|
18 | 20 | import com.azure.core.http.policy.HttpLogOptions;
|
|
31 | 33 | import java.net.URL;
|
32 | 34 | import java.util.ArrayList;
|
33 | 35 | import java.util.List;
|
| 36 | +import java.util.HashMap; |
34 | 37 | import java.util.Map;
|
35 | 38 | import java.util.Objects;
|
36 | 39 |
|
@@ -299,7 +302,9 @@ private List<HttpPipelinePolicy> createHttpPipelineAuthPolicies() {
|
299 | 302 | if (tokenCredential != null) {
|
300 | 303 | pipelinePolicies.add(new BearerTokenAuthenticationPolicy(tokenCredential,
|
301 | 304 | "https://communication.azure.com//.default"));
|
302 |
| - pipelinePolicies.add(new TokenCredentialAddHostHeaderPolicy(hostName)); |
| 305 | + Map<String, String> httpHeaders = new HashMap<>(); |
| 306 | + httpHeaders.put("x-ms-host", hostName); |
| 307 | + pipelinePolicies.add(new AddHeadersPolicy(new HttpHeaders(httpHeaders))); |
303 | 308 | } else if (azureKeyCredential != null) {
|
304 | 309 | pipelinePolicies.add(new HmacAuthenticationPolicy(azureKeyCredential));
|
305 | 310 | } else {
|
|
0 commit comments