• Home
• com.liferay
• com.liferay.jenkins.results.parser
• 1.0.1492
• source code
• AmazonVM.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.liferay.jenkins.results.parser.vm.amazon.AmazonVM Maven / Gradle / Ivy

/**
 * SPDX-FileCopyrightText: (c) 2000 Liferay, Inc. https://liferay.com
 * SPDX-License-Identifier: LGPL-2.1-or-later OR LicenseRef-Liferay-DXP-EULA-2.0.0-2023-06
 */

package com.liferay.jenkins.results.parser.vm.amazon;

import com.amazonaws.auth.AWSStaticCredentialsProvider;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.regions.Regions;
import com.amazonaws.services.ec2.AmazonEC2;
import com.amazonaws.services.ec2.AmazonEC2ClientBuilder;
import com.amazonaws.services.ec2.model.DescribeInstancesRequest;
import com.amazonaws.services.ec2.model.DescribeInstancesResult;
import com.amazonaws.services.ec2.model.EbsInstanceBlockDeviceSpecification;
import com.amazonaws.services.ec2.model.Instance;
import com.amazonaws.services.ec2.model.InstanceBlockDeviceMapping;
import com.amazonaws.services.ec2.model.InstanceBlockDeviceMappingSpecification;
import com.amazonaws.services.ec2.model.InstanceState;
import com.amazonaws.services.ec2.model.ModifyInstanceAttributeRequest;
import com.amazonaws.services.ec2.model.Reservation;
import com.amazonaws.services.ec2.model.RunInstancesRequest;
import com.amazonaws.services.ec2.model.RunInstancesResult;
import com.amazonaws.services.ec2.model.TerminateInstancesRequest;

import com.liferay.jenkins.results.parser.JenkinsResultsParserUtil;
import com.liferay.jenkins.results.parser.vm.VM;

import java.util.List;

/**
 * @author Kiyoshi Lee
 */
public abstract class AmazonVM extends VM {

	@Override
	public void create() {
		RunInstancesRequest runInstancesRequest = new RunInstancesRequest();

		runInstancesRequest.withImageId(_imageId);
		runInstancesRequest.withInstanceType(_instanceType);
		runInstancesRequest.withKeyName(_keyName);
		runInstancesRequest.withMaxCount(1);
		runInstancesRequest.withMinCount(1);
		runInstancesRequest.withSecurityGroupIds("sg-9ce452fb");

		RunInstancesResult runInstancesResult = _amazonEC2.runInstances(
			runInstancesRequest);

		Reservation instanceReservation = runInstancesResult.getReservation();

		String instanceReservationId = instanceReservation.getReservationId();

		DescribeInstancesResult describeInstancesResult =
			_amazonEC2.describeInstances();

		List reservations =
			describeInstancesResult.getReservations();

		for (Reservation reservation : reservations) {
			String reservationId = reservation.getReservationId();

			if (reservationId.equals(instanceReservationId)) {
				List instances = reservation.getInstances();

				Instance instance = instances.get(0);

				_instanceId = instance.getInstanceId();

				break;
			}
		}

		_waitForInstanceState("running");

		_amazonEC2.modifyInstanceAttribute(
			_getModifyInstanceAttributeRequest());
	}

	@Override
	public void delete() {
		TerminateInstancesRequest terminateInstancesRequest =
			new TerminateInstancesRequest();

		terminateInstancesRequest.withInstanceIds(_instanceId);

		_amazonEC2.terminateInstances(terminateInstancesRequest);

		_waitForInstanceState("terminated");
	}

	public String getInstanceId() {
		return _instanceId;
	}

	public String getPublicDnsName() {
		Instance instance = _getInstance();

		return instance.getPublicDnsName();
	}

	protected AmazonVM(
		String awsAccessKeyId, String awsSecretAccessKey, String instanceId) {

		_instanceId = instanceId;

		BasicAWSCredentials basicAWSCredentials = new BasicAWSCredentials(
			awsAccessKeyId, awsSecretAccessKey);

		AmazonEC2ClientBuilder amazonEC2ClientBuilder =
			AmazonEC2ClientBuilder.standard();

		amazonEC2ClientBuilder.withCredentials(
			new AWSStaticCredentialsProvider(basicAWSCredentials));
		amazonEC2ClientBuilder.withRegion(Regions.US_WEST_1);

		_amazonEC2 = amazonEC2ClientBuilder.build();
	}

	protected AmazonVM(
		String awsAccessKeyId, String awsSecretAccessKey, String imageId,
		String instanceType, String keyName) {

		_imageId = imageId;
		_instanceType = instanceType;
		_keyName = keyName;

		BasicAWSCredentials basicAWSCredentials = new BasicAWSCredentials(
			awsAccessKeyId, awsSecretAccessKey);

		AmazonEC2ClientBuilder amazonEC2ClientBuilder =
			AmazonEC2ClientBuilder.standard();

		amazonEC2ClientBuilder.withCredentials(
			new AWSStaticCredentialsProvider(basicAWSCredentials));
		amazonEC2ClientBuilder.withRegion(Regions.US_WEST_1);

		_amazonEC2 = amazonEC2ClientBuilder.build();
	}

	private String _getDeviceName() {
		Instance instance = _getInstance();

		List instanceBlockDeviceMappings =
			instance.getBlockDeviceMappings();

		InstanceBlockDeviceMapping instanceBlockDeviceMapping =
			instanceBlockDeviceMappings.get(0);

		return instanceBlockDeviceMapping.getDeviceName();
	}

	private EbsInstanceBlockDeviceSpecification
		_getEbsInstanceBlockDeviceSpecification() {

		EbsInstanceBlockDeviceSpecification
			ebsInstanceBlockDeviceSpecification =
				new EbsInstanceBlockDeviceSpecification();

		ebsInstanceBlockDeviceSpecification.withDeleteOnTermination(true);

		return ebsInstanceBlockDeviceSpecification;
	}

	private Instance _getInstance() {
		DescribeInstancesRequest describeInstancesRequest =
			new DescribeInstancesRequest();

		describeInstancesRequest.withInstanceIds(_instanceId);

		DescribeInstancesResult describeInstancesResult =
			_amazonEC2.describeInstances(describeInstancesRequest);

		List reservations =
			describeInstancesResult.getReservations();

		Reservation reservation = reservations.get(0);

		List instances = reservation.getInstances();

		return instances.get(0);
	}

	private InstanceBlockDeviceMappingSpecification
		_getInstanceBlockDeviceMappingSpecification() {

		InstanceBlockDeviceMappingSpecification
			instanceBlockDeviceMappingSpecification =
				new InstanceBlockDeviceMappingSpecification();

		instanceBlockDeviceMappingSpecification.withDeviceName(
			_getDeviceName());
		instanceBlockDeviceMappingSpecification.withEbs(
			_getEbsInstanceBlockDeviceSpecification());

		return instanceBlockDeviceMappingSpecification;
	}

	private String _getInstanceState() {
		Instance instance = _getInstance();

		InstanceState instanceState = instance.getState();

		return instanceState.getName();
	}

	private ModifyInstanceAttributeRequest
		_getModifyInstanceAttributeRequest() {

		ModifyInstanceAttributeRequest modifyInstanceAttributeRequest =
			new ModifyInstanceAttributeRequest();

		modifyInstanceAttributeRequest.withBlockDeviceMappings(
			_getInstanceBlockDeviceMappingSpecification());
		modifyInstanceAttributeRequest.withInstanceId(_instanceId);

		return modifyInstanceAttributeRequest;
	}

	private void _waitForInstanceState(String targetState) {
		String instanceState = _getInstanceState();

		System.out.println(
			JenkinsResultsParserUtil.combine(
				"Waiting for the EC2 instance state \"", targetState, "\""));

		long timeout =
			JenkinsResultsParserUtil.getCurrentTimeMillis() +
				MILLIS_TIMEOUT_DURATION;

		while (!instanceState.equals(targetState)) {
			if (JenkinsResultsParserUtil.getCurrentTimeMillis() >= timeout) {
				throw new RuntimeException(
					"Timeout occurred while waiting for EC2 instance state \"" +
						targetState + "\"");
			}

			JenkinsResultsParserUtil.sleep(1000 * 30);

			instanceState = _getInstanceState();
		}
	}

	private final AmazonEC2 _amazonEC2;
	private String _imageId;
	private String _instanceId;
	private String _instanceType;
	private String _keyName;

}