package net.onrc.onos.core.metrics;

import com.codahale.metrics.Counter;
import com.codahale.metrics.Gauge;
import com.codahale.metrics.Histogram;
import com.codahale.metrics.Meter;
import com.codahale.metrics.Metric;
import com.codahale.metrics.MetricFilter;
import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.Timer;
import org.restlet.representation.Representation;
import org.restlet.resource.Get;
import org.restlet.resource.ServerResource;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Map;

/**
 * REST APIs for Metrics objects.
 */
public class MetricsResource extends ServerResource {

    /**
     * Metric filter to allow selecting metrics by name.
     */
    private static class MetricNameFilter implements MetricFilter {
        final HashSet<String> names;

        /**
         * Hide default constructor.
         */
        @SuppressWarnings("unused")
        private MetricNameFilter() {
            names = null;
        }

        /**
         * Initializes a filter for the given name list.
         *
         * @param nameListString comma separated list of strings of the
         *                       names of metrics to query
         */
        public MetricNameFilter(final String nameListString) {

            if (nameListString == null) {
                names = null;
            } else {
                List<String> nameList = Arrays.asList(nameListString.split(","));
                names = new HashSet<>();
                names.addAll(nameList);
            }
        }

        @Override
        public boolean matches(String s, Metric metric) {
            return names == null || names.contains(s);
        }
    }
    /**
     * REST API to get all of the system's metrics.
     *
     * @return a Representation object containing the metrics
     */
    @Get("json")
    @SuppressWarnings("rawtypes")
    public Representation retrieve() throws Exception {
        final MetricRegistry registry = OnosMetrics.getMetricsRegistry();
        final MetricsObjectResource result = new MetricsObjectResource();

        final List<MetricsObjectResource.TimerObjectResource> timers =
                new ArrayList<>();
        final List<MetricsObjectResource.GaugeObjectResource> gauges =
                new ArrayList<>();
        final List<MetricsObjectResource.CounterObjectResource> counters =
                new ArrayList<>();
        final List<MetricsObjectResource.MeterObjectResource> meters =
                new ArrayList<>();
        final List<MetricsObjectResource.HistogramObjectResource> histograms =
                new ArrayList<>();

        final String metricIdsString = getQuery().getValues("ids");

        final MetricFilter filter = new MetricNameFilter(metricIdsString);

        for (final Map.Entry<String, Timer> timer :
             registry.getTimers(filter).entrySet()) {
            timers.add(new MetricsObjectResource.TimerObjectResource(
                    timer.getKey(), timer.getValue()));
        }
        result.setTimers(timers);

        for (final Map.Entry<String, Gauge> gauge :
             registry.getGauges(filter).entrySet()) {
            gauges.add(new MetricsObjectResource.GaugeObjectResource(
                    gauge.getKey(), gauge.getValue()));
        }
        result.setGauges(gauges);

        for (final Map.Entry<String, Counter> counter :
             registry.getCounters(filter).entrySet()) {
            counters.add(new MetricsObjectResource.CounterObjectResource(
                    counter.getKey(), counter.getValue()));
        }
        result.setCounters(counters);

        for (final Map.Entry<String, Meter> meter :
             registry.getMeters(filter).entrySet()) {
            meters.add(new MetricsObjectResource.MeterObjectResource(
                    meter.getKey(), meter.getValue()));
        }
        result.setMeters(meters);

        for (final Map.Entry<String, Histogram> histogram :
             registry.getHistograms(filter).entrySet()) {
            histograms.add(new MetricsObjectResource.HistogramObjectResource(
                    histogram.getKey(), histogram.getValue()));
        }
        result.setHistograms(histograms);

        return toRepresentation(result, null);
    }

}