es.bsc.amon.controller.QueriesDBMapper.java Source code

Introduction

Here is the source code for es.bsc.amon.controller.QueriesDBMapper.java

Source

/*
 * Author: Mario Macias (Barcelona Supercomputing Center). 2014
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * This library 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 Lesser General Public License for more details:
 *
 * http://www.gnu.org/licenses/lgpl-2.1.html
 */

package es.bsc.amon.controller;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.mongodb.BasicDBList;
import com.mongodb.BasicDBObject;
import com.mongodb.BasicDBObjectBuilder;
import com.mongodb.DBObject;
import com.mongodb.util.JSON;
import es.bsc.amon.DBManager;
import es.bsc.mongoal.QueryGenerator;
import play.Logger;
import play.libs.Json;

/**
 * Created by mmacias on 02/07/14.
 */
public class QueriesDBMapper {
    public static final QueriesDBMapper instance = new QueriesDBMapper();
    private static es.bsc.mongoal.QueryGenerator mongoAlQG = null;

    private QueriesDBMapper() {
    }

    /**
     * Uses the MongoDB Json query language
     * @param query
     * @return
     */
    public ArrayNode aggregate(JsonNode query) {
        Object raw = JSON.parse(query.toString());
        ArrayNode ret = new ArrayNode(JsonNodeFactory.instance);

        if (raw instanceof BasicDBObject) {
            ret = (ArrayNode) Json.parse(EventsDBMapper.getInstance().aggregate((BasicDBObject) raw).toString());
        } else if (raw instanceof BasicDBList) {
            ret = (ArrayNode) Json.parse(EventsDBMapper.getInstance().aggregate((BasicDBList) raw).toString());
        }

        return ret;
    }

    /**
     * Uses the MongoAL query language
     */
    public ArrayNode aggregate(String query) {
        if (mongoAlQG == null) {
            mongoAlQG = new QueryGenerator(DBManager.instance.getDatabase());
        }

        BasicDBList dbl = new BasicDBList();
        Iterable<DBObject> it = mongoAlQG.query(query);
        for (DBObject dbo : it) {
            dbl.add(dbo);
        }
        return (ArrayNode) Json.parse(dbl.toString());

    }

    public static final String START = "start";
    public static final String END = "end";
    public static final String APPID = "appId";
    public static final String NODEID = "nodeId";
    public static final String INSTANCEID = "instanceId";
    public static final String OP = "op";
    public static final String DATA = "data";

    public enum Operation {
        sum, avg, max, min, first, last, count, array
    };
}