com.amazonaws.codegen.AddWaiters Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aws-java-sdk-code-generator Show documentation
The AWS SDK for Java - Code Generator module holds the classes and templates required to generate the AWS Java SDK clients for AWS services.
There is a newer version: 1.11.8
Show newest version
/*
 * Copyright 2010-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

package com.amazonaws.codegen;

import com.amazonaws.codegen.model.intermediate.AcceptorModel;
import com.amazonaws.codegen.model.intermediate.OperationModel;
import com.amazonaws.codegen.model.intermediate.WaiterDefinitionModel;
import com.amazonaws.codegen.model.service.*;
import com.amazonaws.jmespath.JmesPathExpression;
import com.amazonaws.util.IOUtils;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static com.amazonaws.codegen.AstJsonToAstJava.fromAstJsonToAstJava;

class AddWaiters {

    private static final ObjectMapper mapper = new ObjectMapper();

    private final Waiters waiters;
    private final Map operations;

    AddWaiters(Waiters waiters, Map operations){
        this.waiters = waiters;
        this.operations = operations;
    }

    Map constructWaiters() throws IOException{


        Map javaWaiterModels = new HashMap<>();
        Map argumentToAstMap = new HashMap<>();

        for (Map.Entry entry : waiters.getWaiters().entrySet()) {

            final String waiterName = entry.getKey();
            final WaiterDefinition waiterDefinition = entry.getValue();
            List acceptors = new ArrayList<>();

            WaiterDefinitionModel waiterDefinitionModel = new WaiterDefinitionModel();

            waiterDefinitionModel.setDelay(waiterDefinition.getDelay());
            waiterDefinitionModel.setMaxAttempts(waiterDefinition.getMaxAttempts());
            waiterDefinitionModel.setWaiterName(waiterName);
            waiterDefinitionModel.setOperationModel(operations.get(waiterDefinition.getOperation()));
            for(Acceptor acceptor: waiterDefinition.getAcceptors()){
                AcceptorModel acceptorModel = new AcceptorModel();
                acceptorModel.setMatcher(acceptor.getMatcher());
                acceptorModel.setState(acceptor.getState());
                acceptorModel.setExpected(acceptor.getExpected());
                acceptorModel.setArgument(acceptor.getArgument());
                acceptorModel.setAst(getAstFromArgument(acceptor.getArgument(), argumentToAstMap));

                acceptors.add(acceptorModel);
            }
            waiterDefinitionModel.setAcceptors(acceptors);
            javaWaiterModels.put(waiterName, waiterDefinitionModel);
        }

        return javaWaiterModels;
    }

    private static JmesPathExpression getAstFromArgument(String argument, Map argumentToAstMap) throws IOException{
        if(argument != null && !argumentToAstMap.containsKey(argument)) {

            ProcessBuilder pb = new ProcessBuilder("/apollo/env/SDETools/bin/brazil-runtime-exec", "python3.4",
                    System.getProperty("codeGenSourceDirectory") + "/bin/jp-to-ast.py", argument);
            Process p = pb.start();

            JsonNode jsonNode = mapper.readTree(IOUtils.toString(p.getInputStream()));
            JmesPathExpression ast = fromAstJsonToAstJava(jsonNode);

            argumentToAstMap.put(argument, ast);
            IOUtils.closeQuietly(p.getInputStream(), null);

            return ast;

        }
        else if(argument != null){
            return argumentToAstMap.get(argument);
        }
        return null;
    }
}