MetricQuery.java

/*
 * Copyright 2011 Facebook, Inc.
 *
 * 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 com.facebook.tsdb.tsdash.server.model;

import java.util.ArrayList;
import java.util.HashMap;

import org.json.simple.JSONArray;
import org.json.simple.JSONObject;

public class MetricQuery {

    public String name;
    public HashMap<String, String> tags = null;
    public String[] orders;
    public String aggregator = null;
    public String[] dissolveTags;
    public boolean rate = false;

    public static HashMap<String, String> decodeTags(JSONObject tagsObj) {
        HashMap<String, String> tags = new HashMap<String, String>();
        for (Object tagKeyObj : tagsObj.keySet()) {
            tags.put((String) tagKeyObj, (String) tagsObj.get(tagKeyObj));
        }
        return tags;
    }

    private static String[] decodeArray(JSONArray jsonArray) {
        String[] array = new String[jsonArray.size()];
        for (int i = 0; i < array.length; i++) {
            array[i] = (String) jsonArray.get(i);
        }
        return array;
    }

    public ArrayList<String> getDissolveList() {
        ArrayList<String> toDissolve = new ArrayList<String>();
        if (aggregator == null) {
            return toDissolve;
        }
        // the host is dissolved by default
        toDissolve.add("host");
        for (String userSpecified : dissolveTags) {
            if (!userSpecified.equals("host")) {
                toDissolve.add(userSpecified);
            }
        }
        return toDissolve;
    }

    public static MetricQuery fromJSONObject(JSONObject src) {
        MetricQuery newQuery = new MetricQuery();
        newQuery.name = (String) src.get("name");
        if (src.get("rate") != null) {
            newQuery.rate = (Boolean) src.get("rate");
        }
        newQuery.tags = decodeTags((JSONObject) src.get("tags"));
        newQuery.aggregator = (String) src.get("aggregator");
        newQuery.orders = decodeArray((JSONArray) src.get("orders"));
        newQuery.dissolveTags =
            decodeArray((JSONArray) src.get("dissolveTags"));
        return newQuery;
    }

    @Override
    public String toString() {
        String ret = "Name: " + name + '\n';
        ret += "tags: " + tags + '\n';
        ret += "orders: " + orders + '\n';
        ret += "aggregator: " + aggregator + '\n';
        ret += "dissolve: " + dissolveTags + '\n';
        ret += "rate: " + rate + '\n';
        ret += '\n';
        return ret;
    }
}