org.fao.geonet.monitor.health.DashboardAppHealthCheck.java Source code

Introduction

Here is the source code for org.fao.geonet.monitor.health.DashboardAppHealthCheck.java

Source

/*
 * Copyright (C) 2001-2016 Food and Agriculture Organization of the
 * United Nations (FAO-UN), United Nations World Food Programme (WFP)
 * and United Nations Environment Programme (UNEP)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 *
 * Contact: Jeroen Ticheler - FAO - Viale delle Terme di Caracalla 2,
 * Rome - Italy. email: geonetwork@osgeo.org
 */

package org.fao.geonet.monitor.health;

import com.yammer.metrics.core.HealthCheck;
import io.searchbox.core.Search;
import io.searchbox.core.SearchResult;
import jeeves.monitor.HealthCheckFactory;
import jeeves.server.context.ServiceContext;
import org.apache.commons.lang.StringUtils;
import org.apache.http.client.methods.HttpGet;
import org.fao.geonet.GeonetContext;
import org.fao.geonet.constants.Geonet;
import org.fao.geonet.kernel.search.EsSearchManager;
import org.fao.geonet.kernel.setting.SettingManager;
import org.fao.geonet.utils.GeonetHttpRequestFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.client.ClientHttpResponse;

/**
 * Checks to ensure that the Kibana is up and running.
 */
public class DashboardAppHealthCheck implements HealthCheckFactory {

    public HealthCheck create(final ServiceContext context) {
        return new HealthCheck(this.getClass().getSimpleName()) {
            @Override
            protected Result check() throws Exception {
                final GeonetHttpRequestFactory httpRequestFactory = context.getBean(GeonetHttpRequestFactory.class);
                final SettingManager settingManager = context.getBean(SettingManager.class);
                final EsSearchManager searchMan = context.getBean(EsSearchManager.class);
                final String dashboardAppUrl = searchMan.getClient().getDashboardAppUrl();

                if (StringUtils.isNotEmpty(dashboardAppUrl)) {
                    ClientHttpResponse httpResponse = null;
                    try {
                        String url = settingManager.getBaseURL() + "dashboards/api/status";
                        httpResponse = httpRequestFactory.execute(new HttpGet(url));

                        if (httpResponse.getRawStatusCode() == 200) {
                            return Result.healthy(String.format("Dashboard application is running."));
                        } else {
                            return Result.unhealthy("Dashboard application is not available currently. "
                                    + "This component is only required if you use dashboards.");
                        }
                    } catch (Throwable e) {
                        return Result.unhealthy(e);
                    } finally {
                        if (httpResponse != null) {
                            httpResponse.close();
                        }
                    }
                } else {
                    return Result.unhealthy("Dashboard application is not configured. "
                            + "Update config.properties to setup Kibana to use this feature.");
                }
            }
        };
    }
}