|
58 | 58 | import software.amazon.awssdk.auth.credentials.AwsBasicCredentials; |
59 | 59 | import software.amazon.awssdk.auth.credentials.AwsCredentials; |
60 | 60 | import software.amazon.awssdk.auth.credentials.AwsSessionCredentials; |
61 | | -import software.amazon.awssdk.auth.signer.Aws4Signer; |
62 | | -import software.amazon.awssdk.auth.signer.params.Aws4SignerParams; |
63 | 61 | import software.amazon.awssdk.core.sync.RequestBody; |
64 | 62 | import software.amazon.awssdk.http.SdkHttpFullRequest; |
65 | 63 | import software.amazon.awssdk.http.SdkHttpMethod; |
| 64 | +import software.amazon.awssdk.http.auth.aws.signer.AwsV4HttpSigner; |
| 65 | +import software.amazon.awssdk.http.auth.spi.signer.SignedRequest; |
66 | 66 | import software.amazon.awssdk.regions.Region; |
67 | 67 |
|
68 | 68 | @Log4j2 |
69 | 69 | public class ConnectorUtils { |
70 | 70 |
|
71 | | - private static final Aws4Signer signer; |
| 71 | + private static final AwsV4HttpSigner signer; |
72 | 72 | public static final String SKIP_VALIDATE_MISSING_PARAMETERS = "skip_validating_missing_parameters"; |
73 | 73 |
|
74 | 74 | static { |
75 | | - signer = Aws4Signer.create(); |
| 75 | + signer = AwsV4HttpSigner.create(); |
76 | 76 | } |
77 | 77 |
|
78 | 78 | public static RemoteInferenceInputDataSet processInput( |
@@ -293,14 +293,14 @@ public static SdkHttpFullRequest signRequest( |
293 | 293 | ? AwsBasicCredentials.create(accessKey, secretKey) |
294 | 294 | : AwsSessionCredentials.create(accessKey, secretKey, sessionToken); |
295 | 295 |
|
296 | | - Aws4SignerParams params = Aws4SignerParams |
297 | | - .builder() |
298 | | - .awsCredentials(credentials) |
299 | | - .signingName(signingName) |
300 | | - .signingRegion(Region.of(region)) |
301 | | - .build(); |
302 | | - |
303 | | - return signer.sign(request, params); |
| 296 | + SignedRequest signedRequest = |
| 297 | + signer.sign( |
| 298 | + r -> r.identity(credentials) |
| 299 | + .request(request) |
| 300 | + .payload(request.contentStreamProvider().orElse(null)) |
| 301 | + .putProperty(AwsV4HttpSigner.SERVICE_SIGNING_NAME, signingName) |
| 302 | + .putProperty(AwsV4HttpSigner.REGION_NAME, region)); |
| 303 | + return (SdkHttpFullRequest) signedRequest.request(); |
304 | 304 | } |
305 | 305 |
|
306 | 306 | public static SdkHttpFullRequest buildSdkRequest( |
|
0 commit comments