com.nitorcreations.willow.autoscaler.metrics.AutoScalingStatus Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of willow-autoscaler Show documentation

Willow elastic cloud auto scaler

There is a newer version: 2.0.2

package com.nitorcreations.willow.autoscaler.metrics;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import javax.inject.Named;
import javax.inject.Singleton;

import com.nitorcreations.willow.autoscaler.config.AutoScalingGroupConfig;
import com.nitorcreations.willow.autoscaler.deployment.AutoScalingGroupDeploymentStatus;
import com.nitorcreations.willow.messages.metrics.TimePoint;

@Named
@Singleton
public class AutoScalingStatus {

  //private Logger logger = Logger.getLogger(this.getClass().getCanonicalName());

  private Map statuses = new ConcurrentHashMap<>();

  private List groupConfigs = new ArrayList<>();

  public synchronized void initialize(List groups) {
    this.groupConfigs = groups;
    for (AutoScalingGroupConfig group : groups) {
      this.statuses.put(group.getName(), new AutoScalingGroupStatus(group));
    }
  }

  public AutoScalingGroupStatus getStatus(String group) {
    return statuses.get(group);
  }

  public void setDeploymentStatus(String group, AutoScalingGroupDeploymentStatus deploymentStatus) {
    AutoScalingGroupStatus currentStatus = getGroupStatus(group);
    currentStatus.setDeploymentStatus(deploymentStatus);
  }

  public void addMetricValue(String group, String metric, TimePoint metricValue) {
    AutoScalingGroupStatus currentStatus = getGroupStatus(group);
    currentStatus.addMetricValue(metric, metricValue);
  }

  public List getGroupConfigs() {
    return new ArrayList<>(groupConfigs);
  }

  private AutoScalingGroupStatus getGroupStatus(String group) {
    AutoScalingGroupStatus currentStatus = statuses.get(group);
    if (currentStatus == null) {
      throw new IllegalArgumentException("Unknown auto scaling group " + group);
    }
    return currentStatus;
  }

}