Examples with ListInstancesRequest com.amazonaws.services.elasticmapreduce.model.ListInstancesRequest used on opensource projects

Example 1 with ListInstancesRequest

use of com.amazonaws.services.elasticmapreduce.model.ListInstancesRequest in project herd by FINRAOS.

the class EmrDaoImpl method getEmrMasterInstance.

@Override
public Instance getEmrMasterInstance(String clusterId, AwsParamsDto awsParams) throws Exception {
    // Get the master EC2 instance
    ListInstancesRequest listInstancesRequest = new ListInstancesRequest().withClusterId(clusterId).withInstanceGroupTypes(InstanceGroupType.MASTER);
    List<Instance> instances = emrOperations.listClusterInstancesRequest(getEmrClient(awsParams), listInstancesRequest).getInstances();
    // Throw error in case there are no master instances found yet
    if (instances.size() == 0) {
        throw new IllegalArgumentException("No master instances found for the cluster \"" + clusterId + "\".");
    }
    // EMR has only one master node.
    return instances.get(0);
}

Example 2 with ListInstancesRequest

use of com.amazonaws.services.elasticmapreduce.model.ListInstancesRequest in project herd by FINRAOS.

the class EmrDaoImpl method addEmrMasterSecurityGroups.

@Override
public List<String> addEmrMasterSecurityGroups(String clusterId, List<String> securityGroups, AwsParamsDto awsParams) throws Exception {
    // Get the master EC2 instance
    ListInstancesRequest listInstancesRequest = new ListInstancesRequest().withClusterId(clusterId).withInstanceGroupTypes(InstanceGroupType.MASTER);
    List<Instance> instances = emrOperations.listClusterInstancesRequest(getEmrClient(awsParams), listInstancesRequest).getInstances();
    // Throw error in case there are no master instances found yet
    if (instances.size() == 0) {
        throw new IllegalArgumentException("No master instances found for the cluster \"" + clusterId + "\".");
    }
    for (Instance instance : instances) {
        ec2Dao.addSecurityGroupsToEc2Instance(instance.getEc2InstanceId(), securityGroups, awsParams);
    }
    return securityGroups;
}

Example 3 with ListInstancesRequest

use of com.amazonaws.services.elasticmapreduce.model.ListInstancesRequest in project herd by FINRAOS.

the class EmrDaoTest method getEmrMasterInstanceReturnsInstance.

@Test
public void getEmrMasterInstanceReturnsInstance() throws Exception {
    String clusterId = "clusterId";
    Instance expectedInstance = new Instance();
    /*
         * Stub EmrOperations.listClusterInstancesRequest() and assert correct parameters are being passed into the method when the call is made.
         */
    when(mockEmrOperations.listClusterInstancesRequest(any(), any())).thenAnswer(new Answer<ListInstancesResult>() {

        @Override
        public ListInstancesResult answer(InvocationOnMock invocation) throws Throwable {
            /*
                 * Assert correct parameters are used when calling this method.
                 */
            ListInstancesRequest listInstancesRequest = invocation.getArgument(1);
            assertEquals(clusterId, listInstancesRequest.getClusterId());
            List<String> instanceGroupTypes = listInstancesRequest.getInstanceGroupTypes();
            assertEquals(1, instanceGroupTypes.size());
            assertEquals("MASTER", instanceGroupTypes.get(0));
            ListInstancesResult listInstancesResult = new ListInstancesResult();
            listInstancesResult.setInstances(Arrays.asList(expectedInstance));
            return listInstancesResult;
        }
    });
    Instance actualInstance = emrDao.getEmrMasterInstance(clusterId, new AwsParamsDto());
    assertEquals(expectedInstance, actualInstance);
}