RadiatorPage.java

/*
 * Copyright 2010-2012 Capgemini
 * Licensed 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 net.sf.appstatus.web.pages;

import static net.sf.appstatus.web.HtmlUtils.applyLayout;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang3.StringUtils;

import net.sf.appstatus.core.batch.IBatchManager;
import net.sf.appstatus.core.check.ICheckResult;
import net.sf.appstatus.web.IPage;
import net.sf.appstatus.web.StatusWebHandler;

/**
 * This is a radiator page.
 * <p>
 * This page displays a quick overview of the application and reloads every
 * minute.
 *
 * @author Nicolas Richeton
 *
 */
public class RadiatorPage implements IPage {

	private static final int STATUS_ERROR = 2;
	private static final int STATUS_MAINTENANCE = 3;
	private static final int STATUS_OK = 0;
	private static final int STATUS_WARN = 1;

	public void doGet(final StatusWebHandler webHandler, final HttpServletRequest req, final HttpServletResponse resp)
			throws UnsupportedEncodingException, IOException {

		// Setup response
		resp.setContentType("text/html");
		resp.setCharacterEncoding("UTF-8");

		// Get Health checks
		final List<ICheckResult> results = webHandler.getAppStatus().checkAll(req.getLocale());
		int status = webHandler.getAppStatus().isMaintenance() ? STATUS_MAINTENANCE : STATUS_OK;
		for (final ICheckResult r : results) {

			if (r.getCode() != ICheckResult.OK && !r.isFatal() && status == STATUS_OK) {
				status = STATUS_WARN;
			}

			if (r.getCode() != ICheckResult.OK && r.isFatal()) {
				status = STATUS_ERROR;
				break;
			}
		}

		String btnClass;

		switch (status) {
		case STATUS_WARN:
			btnClass = "btn-warning";
			break;
		case STATUS_ERROR:
			btnClass = "btn-danger";
			break;
		case STATUS_MAINTENANCE:
			btnClass = "btn-info";
			break;
		default:
			btnClass = "btn-success";
		}

		// Get batchs status.
		final IBatchManager manager = webHandler.getAppStatus().getBatchManager();

		String batchStatus = " progress-success ";
		String active = StringUtils.EMPTY;
		int width = 0;

		if (manager != null) {
			batchStatus = manager.getErrorBatches().size() > 0 ? " progress-danger " : " progress-success ";
			active = manager.getRunningBatches().size() > 0 ? " progress-striped active " : "";
			width = manager.getRunningBatches().size() + manager.getFinishedBatches().size() > 0 ? 100 : 0;
		}

		final Map<String, String> model = new HashMap<String, String>();
		model.put("applicationName", webHandler.getApplicationName());
		model.put("batchBtnClass", btnClass);
		model.put("batchStatus", batchStatus);
		model.put("batchActive", active);
		model.put("batchBarWidth", width + "%");

		resp.getWriter().append(applyLayout(model, "radiatorLayout.html"));
	}

	public void doPost(final StatusWebHandler webHandler, final HttpServletRequest req,
			final HttpServletResponse resp) {
		// Nothing to do
	}

	public String getId() {
		return "radiator";
	}

	public String getName() {
		return "Radiator";
	}

}