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.ec2.transform.internal; 17 18 import java.util.List; 19 import java.util.regex.Pattern; 20 import software.amazon.awssdk.annotations.SdkInternalApi; 21 import software.amazon.awssdk.core.interceptor.Context; 22 import software.amazon.awssdk.core.interceptor.ExecutionAttributes; 23 import software.amazon.awssdk.core.interceptor.ExecutionInterceptor; 24 import software.amazon.awssdk.http.SdkHttpRequest; 25 import software.amazon.awssdk.services.ec2.model.DescribeSpotFleetRequestHistoryRequest; 26 import software.amazon.awssdk.services.ec2.model.RequestSpotFleetRequest; 27 28 /** 29 * A request handler that strips out millisecond precision from requests to 30 * RequestSpotFleet and DescribeSpotFleetRequestHistory, which don't expect 31 * timestamps to be so precise. 32 */ 33 @SdkInternalApi 34 public final class TimestampFormatInterceptor implements ExecutionInterceptor { 35 36 private static final Pattern PATTERN = Pattern.compile("\\.\\d\\d\\dZ"); 37 38 private static final String START_TIME = "StartTime"; 39 private static final String VALID_FROM = "SpotFleetRequestConfig.ValidFrom"; 40 private static final String VALID_UNTIL = "SpotFleetRequestConfig.ValidUntil"; 41 42 @Override modifyHttpRequest(Context.ModifyHttpRequest context, ExecutionAttributes executionAttributes)43 public SdkHttpRequest modifyHttpRequest(Context.ModifyHttpRequest context, ExecutionAttributes executionAttributes) { 44 SdkHttpRequest request = context.httpRequest(); 45 Object original = context.request(); 46 if (original instanceof DescribeSpotFleetRequestHistoryRequest) { 47 List<String> startTime = request.firstMatchingRawQueryParameters(START_TIME); 48 49 if (startTime != null && !startTime.isEmpty()) { 50 return request.toBuilder() 51 .putRawQueryParameter(START_TIME, sanitize(startTime.get(0))) 52 .build(); 53 } 54 55 } else if (original instanceof RequestSpotFleetRequest) { 56 List<String> validFrom = request.firstMatchingRawQueryParameters(VALID_FROM); 57 List<String> validUntil = request.firstMatchingRawQueryParameters(VALID_UNTIL); 58 59 return request.toBuilder().applyMutation(builder -> { 60 if (validFrom != null && !validFrom.isEmpty()) { 61 builder.putRawQueryParameter(VALID_FROM, sanitize(validFrom.get(0))); 62 } 63 if (validUntil != null && !validUntil.isEmpty()) { 64 builder.putRawQueryParameter(VALID_UNTIL, sanitize(validUntil.get(0))); 65 } 66 }).build(); 67 68 } 69 return request; 70 } 71 72 private String sanitize(String input) { 73 return PATTERN.matcher(input).replaceFirst("Z"); 74 } 75 } 76