SetHealthCheckOperation.java

/*******************************************************************************
 * Copyright (c) 2015 Pivotal Software, Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Pivotal Software, Inc. - initial API and implementation
 *******************************************************************************/
package org.springframework.ide.eclipse.boot.dash.cloudfoundry.ops;

import java.util.UUID;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.OperationCanceledException;
import org.springframework.ide.eclipse.boot.dash.cloudfoundry.CloudAppDashElement;
import org.springframework.ide.eclipse.boot.dash.cloudfoundry.client.ClientRequests;
import org.springframework.ide.eclipse.boot.dash.model.UserInteractions;
import org.springframework.ide.eclipse.boot.dash.util.CancelationTokens.CancelationToken;

/**
 * @author Kris De Volder
 */
public class SetHealthCheckOperation extends CloudApplicationOperation {

	private String hcType;
	private CloudAppDashElement app;
	private UserInteractions ui;
	private boolean confirmChange;
	private static final String CONFIRM_CHANGE_KEY = SetHealthCheckOperation.class.getName()+".confirm";

	public SetHealthCheckOperation(CloudAppDashElement app, String hcType, UserInteractions ui, boolean confirmChange, CancelationToken cancelationToken) {
		super("set-health-check "+app.getName()+" "+hcType, app.getCloudModel(), app.getName(), cancelationToken);
		this.app = app;
		this.hcType = hcType;
		this.ui = ui;
		this.confirmChange = confirmChange;
	}

	public SetHealthCheckOperation(CloudAppDashElement app, String hcType, CancelationToken cancelationToken) {
		this(app, hcType, null, false, cancelationToken);
	}

	@Override
	protected void doCloudOp(IProgressMonitor monitor) throws Exception, OperationCanceledException {
		monitor.beginTask(getName(), 2);
		try {
			ClientRequests client = getClientRequests();
			UUID guid = app.getAppGuid();
			String current = client.getHealthCheck(guid);
			//When current==null it means that there's no 'health-check' info in the info returned by
			//cloudcontroller. Probably this means app has no support for this and so we shouldn't try to
			//set it.
			monitor.worked(1);
			if (current!=null) {
				if (!current.equals(hcType)) {
					boolean confirmed = true;
					if (confirmChange) {
						confirmed = ui.yesNoWithToggle(CONFIRM_CHANGE_KEY, "Set health check for '"+app+"' to '"+hcType+"'?",
								"Health check is currently set to '"+current+"'. This may interfere with Spring Boot Devtools "+
								"reload functionality. Do you want to set it to '"+hcType+"'?",
								"Remember my decision"
						);
					}
					if (confirmed) {
						client.setHealthCheck(guid, hcType);
						app.setHealthCheck(hcType);
					}
					monitor.worked(1);
				}
			}
		} finally {
			monitor.done();
		}
	}

}