Skip to content

Commit 8d27b89

Browse files
committed
Use AwsV4HttpSigner instead of Aws4Signer
Signed-off-by: Andy Qin <[email protected]>
1 parent 85a3135 commit 8d27b89

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -58,21 +58,21 @@
5858
import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
5959
import software.amazon.awssdk.auth.credentials.AwsCredentials;
6060
import software.amazon.awssdk.auth.credentials.AwsSessionCredentials;
61-
import software.amazon.awssdk.auth.signer.Aws4Signer;
62-
import software.amazon.awssdk.auth.signer.params.Aws4SignerParams;
6361
import software.amazon.awssdk.core.sync.RequestBody;
6462
import software.amazon.awssdk.http.SdkHttpFullRequest;
6563
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;
6666
import software.amazon.awssdk.regions.Region;
6767

6868
@Log4j2
6969
public class ConnectorUtils {
7070

71-
private static final Aws4Signer signer;
71+
private static final AwsV4HttpSigner signer;
7272
public static final String SKIP_VALIDATE_MISSING_PARAMETERS = "skip_validating_missing_parameters";
7373

7474
static {
75-
signer = Aws4Signer.create();
75+
signer = AwsV4HttpSigner.create();
7676
}
7777

7878
public static RemoteInferenceInputDataSet processInput(
@@ -293,14 +293,14 @@ public static SdkHttpFullRequest signRequest(
293293
? AwsBasicCredentials.create(accessKey, secretKey)
294294
: AwsSessionCredentials.create(accessKey, secretKey, sessionToken);
295295

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();
304304
}
305305

306306
public static SdkHttpFullRequest buildSdkRequest(

0 commit comments

Comments
 (0)