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.machinelearning.internal; 17 18 import java.util.UUID; 19 import software.amazon.awssdk.annotations.SdkInternalApi; 20 import software.amazon.awssdk.core.SdkRequest; 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.services.machinelearning.model.CreateBatchPredictionRequest; 25 import software.amazon.awssdk.services.machinelearning.model.CreateDataSourceFromRdsRequest; 26 import software.amazon.awssdk.services.machinelearning.model.CreateDataSourceFromRedshiftRequest; 27 import software.amazon.awssdk.services.machinelearning.model.CreateDataSourceFromS3Request; 28 import software.amazon.awssdk.services.machinelearning.model.CreateEvaluationRequest; 29 import software.amazon.awssdk.services.machinelearning.model.CreateMlModelRequest; 30 31 /** 32 * CreateXxx API calls require a unique (for all time!) ID parameter for 33 * idempotency. If the user doesn't specify one, fill in a GUID. 34 */ 35 @SdkInternalApi 36 //TODO: They should be using the idempotency trait 37 public final class RandomIdInterceptor implements ExecutionInterceptor { 38 39 @Override modifyRequest(Context.ModifyRequest context, ExecutionAttributes executionAttributes)40 public SdkRequest modifyRequest(Context.ModifyRequest context, ExecutionAttributes executionAttributes) { 41 SdkRequest request = context.request(); 42 43 if (request instanceof CreateBatchPredictionRequest) { 44 CreateBatchPredictionRequest copy = (CreateBatchPredictionRequest) request; 45 46 if (copy.batchPredictionDataSourceId() == null) { 47 return copy.toBuilder().batchPredictionDataSourceId(UUID.randomUUID().toString()).build(); 48 } 49 50 return copy; 51 } else if (request instanceof CreateDataSourceFromRdsRequest) { 52 CreateDataSourceFromRdsRequest copy = 53 (CreateDataSourceFromRdsRequest) request; 54 55 if (copy.dataSourceId() == null) { 56 copy = copy.toBuilder().dataSourceId(UUID.randomUUID().toString()).build(); 57 } 58 59 return copy; 60 } else if (request instanceof CreateDataSourceFromRedshiftRequest) { 61 CreateDataSourceFromRedshiftRequest copy = 62 (CreateDataSourceFromRedshiftRequest) request; 63 64 if (copy.dataSourceId() == null) { 65 copy = copy.toBuilder().dataSourceId(UUID.randomUUID().toString()).build(); 66 } 67 68 return copy; 69 } else if (request instanceof CreateDataSourceFromS3Request) { 70 CreateDataSourceFromS3Request copy = (CreateDataSourceFromS3Request) request; 71 72 if (copy.dataSourceId() == null) { 73 copy = copy.toBuilder().dataSourceId(UUID.randomUUID().toString()).build(); 74 } 75 76 return copy; 77 } else if (request instanceof CreateEvaluationRequest) { 78 CreateEvaluationRequest copy = 79 (CreateEvaluationRequest) request; 80 81 if (copy.evaluationId() == null) { 82 copy = copy.toBuilder().evaluationId(UUID.randomUUID().toString()).build(); 83 } 84 85 return copy; 86 } else if (request instanceof CreateMlModelRequest) { 87 CreateMlModelRequest copy = (CreateMlModelRequest) request; 88 89 if (copy.mlModelId() == null) { 90 copy = copy.toBuilder().mlModelId(UUID.randomUUID().toString()).build(); 91 } 92 93 return copy; 94 } 95 96 return request; 97 } 98 99 } 100