private void blockUntilRunningAndAssignElasticIpsToInstancesOrPutIntoBadMap(
     Set<RunningInstance> input, Map<NodeMetadata, Exception> badNodes) {
   Map<RegionAndName, RunningInstance> instancesById =
       Maps.uniqueIndex(input, instanceToRegionAndName);
   for (Map.Entry<RegionAndName, RunningInstance> entry : instancesById.entrySet()) {
     RegionAndName id = entry.getKey();
     RunningInstance instance = entry.getValue();
     try {
       logger.debug("<< allocating elastic IP instance(%s)", id);
       String ip = client.getElasticIPAddressServices().allocateAddressInRegion(id.getRegion());
       // block until instance is running
       logger.debug(">> awaiting status running instance(%s)", id);
       AtomicReference<NodeMetadata> node =
           newReference(runningInstanceToNodeMetadata.apply(instance));
       nodeRunning.apply(node);
       logger.trace("<< running instance(%s)", id);
       logger.debug(">> associating elastic IP %s to instance %s", ip, id);
       client
           .getElasticIPAddressServices()
           .associateAddressInRegion(id.getRegion(), ip, id.getName());
       logger.trace("<< associated elastic IP %s to instance %s", ip, id);
       // add mapping of instance to ip into the cache
       elasticIpCache.put(id, ip);
     } catch (RuntimeException e) {
       badNodes.put(runningInstanceToNodeMetadata.apply(instancesById.get(id)), e);
     }
   }
 }
Пример #2
0
  // Fires up an instance, finds its root volume ID, takes a snapshot, then
  // terminates the instance.
  private Snapshot createSnapshot() throws RunNodesException {

    String instanceId = null;
    try {
      RunningInstance instance =
          getOnlyElement(
              concat(
                  ec2Client
                      .getInstanceServices()
                      .runInstancesInRegion(regionId, null, imageId, 1, 1)));
      instanceId = instance.getId();

      assertTrue(runningTester.apply(instance), instanceId + "didn't achieve the state running!");

      instance =
          getOnlyElement(
              concat(
                  ec2Client.getInstanceServices().describeInstancesInRegion(regionId, instanceId)));
      BlockDevice device = instance.getEbsBlockDevices().get("/dev/sda1");
      assertNotNull(device, "device: /dev/sda1 not present on: " + instance);
      Snapshot snapshot =
          ec2Client
              .getElasticBlockStoreServices()
              .createSnapshotInRegion(regionId, device.getVolumeId());
      snapshotsToDelete.add(snapshot.getId());
      return snapshot;
    } finally {
      if (instanceId != null)
        ec2Client.getInstanceServices().terminateInstancesInRegion(regionId, instanceId);
    }
  }
Пример #3
0
 @Override
 @AfterClass(groups = {"integration", "live"})
 protected void tearDownContext() {
   for (String imageId : imagesToDeregister) client.deregisterImageInRegion(regionId, imageId);
   for (String snapshotId : snapshotsToDelete)
     ec2Client.getElasticBlockStoreServices().deleteSnapshotInRegion(regionId, snapshotId);
   super.tearDownContext();
 }
Пример #4
0
 public void testDescribeImages() {
   for (String region : ec2Client.getConfiguredRegions()) {
     Set<? extends Image> allResults = client.describeImagesInRegion(region);
     assertNotNull(allResults);
     assert allResults.size() >= 2 : allResults.size();
     Iterator<? extends Image> iterator = allResults.iterator();
     String id1 = iterator.next().getId();
     String id2 = iterator.next().getId();
     Set<? extends Image> twoResults = client.describeImagesInRegion(region, imageIds(id1, id2));
     assertNotNull(twoResults);
     assertEquals(twoResults.size(), 2);
     iterator = twoResults.iterator();
     assertEquals(iterator.next().getId(), id1);
     assertEquals(iterator.next().getId(), id2);
   }
 }
Пример #5
0
  @Override
  @BeforeClass(groups = {"integration", "live"})
  public void setupContext() {
    super.setupContext();
    ec2Client = view.unwrap(EC2ApiMetadata.CONTEXT_TOKEN).getApi();
    runningTester = retry(new InstanceStateRunning(ec2Client), 600, 5, SECONDS);

    client = ec2Client.getAMIServices();
    if (ebsTemplate != null) {
      Template template = view.getComputeService().templateBuilder().from(ebsTemplate).build();
      regionId = template.getLocation().getId();
      imageId = template.getImage().getProviderId();
      for (Image image : client.describeImagesInRegion(regionId)) {
        if (ebsBackedImageName.equals(image.getName()))
          client.deregisterImageInRegion(regionId, image.getId());
      }
    }
  }
  protected Set<RunningInstance> createNodesInRegionAndZone(
      String region,
      String zone,
      String group,
      int count,
      Template template,
      RunInstancesOptions instanceOptions) {
    int countStarted = 0;
    int tries = 0;
    Set<RunningInstance> started = ImmutableSet.<RunningInstance>of();

    while (countStarted < count && tries++ < count) {
      if (logger.isDebugEnabled())
        logger.debug(
            ">> running %d instance region(%s) zone(%s) ami(%s) params(%s)",
            count - countStarted,
            region,
            zone,
            template.getImage().getProviderId(),
            instanceOptions.buildFormParameters());

      started =
          ImmutableSet.copyOf(
              concat(
                  started,
                  client
                      .getInstanceServices()
                      .runInstancesInRegion(
                          region,
                          zone,
                          template.getImage().getProviderId(),
                          1,
                          count - countStarted,
                          instanceOptions)));

      countStarted = size(started);
      if (countStarted < count)
        logger.debug(
            ">> not enough instances (%d/%d) started, attempting again", countStarted, count);
    }
    return started;
  }
 public Image getImageInRegion(String region, String id) {
   return getOnlyElement(
       client
           .getAMIServices()
           .describeImagesInRegion(region, DescribeImagesOptions.Builder.imageIds(id)));
 }