/*
* 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.apache.provisionr.amazon.activities;
import com.amazonaws.AmazonWebServiceRequest;
import com.amazonaws.services.ec2.model.BlockDeviceMapping;
import com.amazonaws.services.ec2.model.EbsBlockDevice;
import com.amazonaws.services.ec2.model.LaunchSpecification;
import com.amazonaws.services.ec2.model.RequestSpotInstancesRequest;
import com.amazonaws.services.ec2.model.RunInstancesRequest;
import com.amazonaws.services.ec2.model.SpotInstanceType;
import com.google.common.base.Charsets;
import static com.google.common.base.Preconditions.checkNotNull;
import com.google.common.base.Strings;
import com.google.common.base.Throwables;
import com.google.common.collect.Lists;
import com.google.common.io.Resources;
import java.io.IOException;
import java.util.Calendar;
import java.util.List;
import net.schmizz.sshj.common.Base64;
import org.activiti.engine.delegate.DelegateExecution;
import org.apache.provisionr.amazon.core.ImageTable;
import org.apache.provisionr.amazon.core.ImageTableQuery;
import org.apache.provisionr.amazon.core.KeyPairs;
import org.apache.provisionr.amazon.core.ProviderClientCache;
import org.apache.provisionr.amazon.core.SecurityGroups;
import org.apache.provisionr.amazon.options.ProviderOptions;
import org.apache.provisionr.amazon.options.SoftwareOptions;
import org.apache.provisionr.api.hardware.BlockDevice;
import org.apache.provisionr.api.pool.Pool;
import org.apache.provisionr.api.provider.Provider;
public abstract class RunInstances extends AmazonActivity {
public static final String DEFAULT_ARCH = "amd64";
public static final String DEFAULT_TYPE = "instance-store";
public static final String DEFAULT_AMI_ID = "ami-0cdf4965"; // Ubuntu 12.10 x64
protected RunInstances(ProviderClientCache providerClientCache) {
super(providerClientCache);
}
protected RunInstancesRequest createOnDemandInstancesRequest(Pool pool, DelegateExecution execution)
throws IOException {
return (RunInstancesRequest) createRequest(pool, execution, false);
}
protected RequestSpotInstancesRequest createSpotInstancesRequest(Pool pool, DelegateExecution execution)
throws IOException {
return (RequestSpotInstancesRequest) createRequest(pool, execution, true);
}
private AmazonWebServiceRequest createRequest(Pool pool, DelegateExecution execution, boolean spot)
throws IOException {
final String businessKey = execution.getProcessBusinessKey();
final String securityGroupName = SecurityGroups.formatNameFromBusinessKey(businessKey);
final String keyPairName = KeyPairs.formatNameFromBusinessKey(businessKey);
final String instanceType = pool.getHardware().getType();
final String imageId = getImageIdFromPoolConfigurationOrQueryImageTable(
pool, pool.getProvider(), instanceType);
final String userData = Resources.toString(Resources.getResource(RunInstances.class,
"/org/apache/provisionr/amazon/userdata.sh"), Charsets.UTF_8);
List<BlockDevice> blockDevices = pool.getHardware().getBlockDevices();
List<BlockDeviceMapping> blockDeviceMappings = Lists.newArrayList();
if (blockDevices != null && blockDevices.size() > 0) {
for (BlockDevice device : blockDevices) {
blockDeviceMappings.add(new BlockDeviceMapping()
.withDeviceName(device.getName())
.withEbs(new EbsBlockDevice()
.withVolumeSize(device.getSize())
.withDeleteOnTermination(true)
));
}
}
if (spot) {
Calendar validUntil = Calendar.getInstance();
validUntil.add(Calendar.MINUTE, 10);
final String spotPrice = checkNotNull(pool.getProvider().getOption(ProviderOptions.SPOT_BID),
"The bid for spot instances was not specified");
LaunchSpecification ls = new LaunchSpecification()
.withInstanceType(instanceType)
.withKeyName(keyPairName)
.withImageId(imageId)
.withBlockDeviceMappings(blockDeviceMappings)
.withSecurityGroups(Lists.newArrayList(securityGroupName))
.withUserData(Base64.encodeBytes(userData.getBytes(Charsets.UTF_8)));
return new RequestSpotInstancesRequest()
.withSpotPrice(spotPrice)
.withLaunchSpecification(ls)
.withLaunchGroup(businessKey)
.withInstanceCount(pool.getExpectedSize())
.withType(SpotInstanceType.OneTime)
.withValidUntil(validUntil.getTime());
} else {
return new RunInstancesRequest()
.withClientToken(businessKey)
.withSecurityGroups(securityGroupName)
.withKeyName(keyPairName)
.withInstanceType(instanceType)
.withImageId(imageId)
.withBlockDeviceMappings(blockDeviceMappings)
.withMinCount(pool.getMinSize())
.withMaxCount(pool.getExpectedSize())
.withUserData(Base64.encodeBytes(userData.getBytes(Charsets.UTF_8)));
}
}
private String getImageIdFromPoolConfigurationOrQueryImageTable(
Pool pool, Provider provider, String instanceType
) {
final String imageId = pool.getSoftware().getImageId();
if (!Strings.isNullOrEmpty(imageId)) {
return "default".equals(imageId) ? DEFAULT_AMI_ID : imageId;
}
ImageTable imageTable;
try {
imageTable = ImageTable.fromCsvResource("/org/apache/provisionr/amazon/ubuntu.csv");
} catch (IOException e) {
throw Throwables.propagate(e);
}
final String region = provider.getOptionOr(ProviderOptions.REGION, ProviderOptions.DEFAULT_REGION);
final String version = provider.getOptionOr(SoftwareOptions.BASE_OPERATING_SYSTEM_VERSION,
SoftwareOptions.DEFAULT_BASE_OPERATING_SYSTEM_VERSION);
ImageTableQuery query = imageTable.query()
.filterBy("region", region)
.filterBy("version", version)
.filterBy("arch", DEFAULT_ARCH);
if (instanceType.equals("t1.micro")) {
query.filterBy("type", "ebs");
} else {
query.filterBy("type", DEFAULT_TYPE);
}
return query.singleResult();
}
}