• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License").
5  * You may not use this file except in compliance with the License.
6  * A copy of the License is located at
7  *
8  *  http://aws.amazon.com/apache2.0
9  *
10  * or in the "license" file accompanying this file. This file is distributed
11  * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
12  * express or implied. See the License for the specific language governing
13  * permissions and limitations under the License.
14  */
15 
16 package software.amazon.awssdk.services.s3.internal.handlers;
17 
18 import software.amazon.awssdk.annotations.SdkInternalApi;
19 import software.amazon.awssdk.core.interceptor.Context;
20 import software.amazon.awssdk.core.interceptor.ExecutionAttributes;
21 import software.amazon.awssdk.core.interceptor.ExecutionInterceptor;
22 import software.amazon.awssdk.http.SdkHttpRequest;
23 import software.amazon.awssdk.services.s3.model.PutObjectRequest;
24 import software.amazon.awssdk.services.s3.model.UploadPartRequest;
25 
26 /**
27  * Interceptor to add an 'Expect: 100-continue' header to the HTTP Request if it represents a PUT Object request.
28  */
29 @SdkInternalApi
30 //TODO: This should be generalized for all streaming requests
31 public final class StreamingRequestInterceptor implements ExecutionInterceptor {
32     @Override
modifyHttpRequest(Context.ModifyHttpRequest context, ExecutionAttributes executionAttributes)33     public SdkHttpRequest modifyHttpRequest(Context.ModifyHttpRequest context,
34                                             ExecutionAttributes executionAttributes) {
35         if (context.request() instanceof PutObjectRequest || context.request() instanceof UploadPartRequest) {
36             return context.httpRequest().toBuilder().putHeader("Expect", "100-continue").build();
37         }
38         return context.httpRequest();
39     }
40 }
41