rackspace-autoscale/src/main/java/org/jclouds/rackspace/autoscale/v1/internal/ParseHelper.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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 org.jclouds.rackspace.autoscale.v1.internal;

import java.util.List;
import java.util.Map;

import org.jclouds.rackspace.autoscale.v1.domain.CreateScalingPolicy;
import org.jclouds.rackspace.autoscale.v1.domain.CreateScalingPolicy.ScalingPolicyType;
import org.jclouds.rackspace.autoscale.v1.domain.LaunchConfiguration;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Maps;
import com.google.common.primitives.Floats;
import com.google.common.primitives.Ints;

/**
 * 
 * Helper methods for parsing autoscale JSON
 */
public class ParseHelper {
   public static ImmutableMap<String, Object> buildLaunchConfigurationRequestMap(Map<String, Object> postParams) {
      LaunchConfiguration launchConfigurationRequest = (LaunchConfiguration) postParams.get("launchConfiguration");

      ImmutableMap.Builder<String, Object> launchConfigurationMapBuilder = ImmutableMap.builder();
      ImmutableMap.Builder<String, Object> argsBuilder = ImmutableMap.builder();
      ImmutableMap.Builder<String, Object> serverBuilder = ImmutableMap.builder();
      ImmutableList.Builder<Map<String, String>> networksBuilder = ImmutableList.builder();

      for (String networkId : launchConfigurationRequest.getNetworks()) {
         Map<String, String> network = Maps.newHashMap();
         network.put("uuid", networkId);
         networksBuilder.add(network);
      }

      serverBuilder.put("name", launchConfigurationRequest.getServerName());
      serverBuilder.put("imageRef", launchConfigurationRequest.getServerImageRef());
      serverBuilder.put("flavorRef", launchConfigurationRequest.getServerFlavorRef());
      serverBuilder.put("OS-DCF:diskConfig", launchConfigurationRequest.getServerDiskConfig());
      serverBuilder.put("metadata", launchConfigurationRequest.getServerMetadata());
      serverBuilder.put("personality", launchConfigurationRequest.getPersonalities());
      serverBuilder.put("networks", networksBuilder.build());

      argsBuilder.put("loadBalancers", launchConfigurationRequest.getLoadBalancers());
      argsBuilder.put("server", serverBuilder.build());

      launchConfigurationMapBuilder.put("type", launchConfigurationRequest.getType().toString());
      launchConfigurationMapBuilder.put("args", argsBuilder.build());

      return launchConfigurationMapBuilder.build();
   }

   @SuppressWarnings("unchecked")
   public static ImmutableList<Map<String, Object>> buildScalingPoliciesRequestList(Map<String, Object> postParams) {
      List<CreateScalingPolicy> scalingPoliciesRequest = (List<CreateScalingPolicy>) postParams.get("scalingPolicies");
      ImmutableList.Builder<Map<String, Object>> scalingPoliciesListBuilder = ImmutableList.builder();

      for (CreateScalingPolicy scalingPolicy : scalingPoliciesRequest) {
         scalingPoliciesListBuilder.add(buildScalingPolicyMap(scalingPolicy));
      }
      return scalingPoliciesListBuilder.build();
   }

   public static ImmutableMap<String, Object> buildScalingPolicyMap(CreateScalingPolicy scalingPolicy) {
      ImmutableMap.Builder<String, Object> scalingPolicyMapBuilder = ImmutableMap.builder();
      scalingPolicyMapBuilder.put("cooldown", scalingPolicy.getCooldown());
      scalingPolicyMapBuilder.put("type", scalingPolicy.getType().toString());
      scalingPolicyMapBuilder.put("name", scalingPolicy.getName());
      // A couple of different scaling policies are supported, such as percent or number based, or targeting specific numbers of instances
      String targetString = scalingPolicy.getTarget();
      Integer targetInt = Ints.tryParse(targetString);
      // Notes:
      // 1. Refactor when autoscale is complete and service is released to dry this code.
      // 2. Refactor to use simpler code for the number parsing. Polymorphism or a facade might work.
      // 3. Potentially remove or rework the enum code.
      Float targetFloat;
      if (targetInt != null) {
         scalingPolicyMapBuilder.put(scalingPolicy.getTargetType().toString(), targetInt);
      } else if ((targetFloat = Floats.tryParse(targetString)) != null) {
         scalingPolicyMapBuilder.put(scalingPolicy.getTargetType().toString(), targetFloat);
      } else {
         scalingPolicyMapBuilder.put(scalingPolicy.getTargetType().toString(), targetString);
      }

      if (scalingPolicy.getSchedulingType() != null
            && scalingPolicy.getType().equals(ScalingPolicyType.SCHEDULE)) {
         // Have to use getters to rebuild map
         scalingPolicyMapBuilder.put("args", ImmutableMap.of(scalingPolicy.getSchedulingType().toString(), scalingPolicy.getSchedulingString()));
      }

      return scalingPolicyMapBuilder.build();
   }
}