org.restcom.stats.core.service.HistogramService.java Source code

Introduction

Here is the source code for org.restcom.stats.core.service.HistogramService.java

Source

/*
 * TeleStax, Open Source Cloud Communications
 * Copyright 2011-2016, Telestax Inc and individual contributors
 * by the @authors tag.
 *
 * This program is free software: you can redistribute it and/or modify
 * under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation; either version 3 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 */
package org.restcom.stats.core.service;

import com.mongodb.client.MongoCursor;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import javax.inject.Inject;
import javax.inject.Named;

import org.bson.Document;
import org.bson.conversions.Bson;
import org.restcom.stats.core.dto.GaugeDTO;
import org.restcom.stats.core.dto.HistogramDTO;
import org.restcom.stats.core.persistence.DatabaseManager;
import org.restcom.stats.core.type.MetricType;

/**
 * @author Ricardo Limonta
 */
@Named
public class HistogramService implements Serializable {

    @Inject
    private DatabaseManager dbm;

    /**
     * Insert histogram metrics from map.
     * @param value histogram metric attributes.
     */
    public void insertMetric(Map<String, Object> value) {
        dbm.getCollection(MetricType.HISTOGRAM.getCollectionName()).insertOne(new Document(value));
    }

    public List<HistogramDTO> retrieveMetrics(long fromTime, long toTime, String key) {
        List<HistogramDTO> histograms = new ArrayList<>();

        //create params list
        List<Bson> params = new ArrayList<>();

        //define match criteria
        params.add(new Document("$match", new Document("timestamp", new Document("$gte", fromTime))));
        params.add(new Document("$match", new Document("timestamp", new Document("$lte", toTime))));
        params.add(new Document("$match", new Document("key", key)));

        //define grouping criteria
        params.add(new Document("$group",
                new Document("_id", "$timestamp").append("totalCount", new Document("$sum", "$count"))));
        //define order criteria
        params.add(new Document("$sort", new Document("_id", 1)));

        //exec query
        MongoCursor<Document> result = dbm.getCollection(MetricType.HISTOGRAM.getCollectionName()).aggregate(params)
                .iterator();

        //convert document result into dto
        while (result.hasNext()) {
            Document statsDoc = result.next();
            histograms.add(new HistogramDTO(statsDoc.getLong("_id"), statsDoc.getInteger("totalCount")));
        }

        return histograms;
    }

    // https://github.com/RestComm/statistics-service/issues/1
    public List<HistogramDTO> retrieveSumMetrics(long fromTime, long toTime, String key) {
        List<HistogramDTO> histograms = new ArrayList<>();

        //create params list
        List<Bson> params = new ArrayList<>();

        //define match criteria
        params.add(new Document("$match", new Document("timestamp", new Document("$gte", fromTime))));
        params.add(new Document("$match", new Document("timestamp", new Document("$lte", toTime))));
        params.add(new Document("$match", new Document("key", key)));

        //define grouping criteria
        params.add(new Document("$group",
                new Document("_id", "null").append("totalCount", new Document("$sum", "$count"))));

        //exec query
        MongoCursor<Document> result = dbm.getCollection(MetricType.HISTOGRAM.getCollectionName()).aggregate(params)
                .iterator();

        //convert document result into dto
        while (result.hasNext()) {
            Document statsDoc = result.next();
            histograms.add(new HistogramDTO(toTime, statsDoc.getInteger("totalCount")));
        }

        return histograms;
    }
}