pt.webdetails.cda.cache.Query.java Source code

Introduction

Here is the source code for pt.webdetails.cda.cache.Query.java

Source

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

package pt.webdetails.cda.cache;

import java.io.OutputStream;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import org.apache.commons.io.output.NullOutputStream;
import org.hibernate.Session;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import pt.webdetails.cda.CdaEngine;
import pt.webdetails.cda.PluginHibernateException;
import pt.webdetails.cda.query.QueryOptions;
import pt.webdetails.cda.settings.CdaSettings;
import pt.webdetails.cda.settings.SettingsManager;
import pt.webdetails.cda.utils.PluginHibernateUtil;

/**
 *
 * @author pdpi
 */
public abstract class Query implements Serializable {

    private static final long serialVersionUID = 1L;

    private long id;
    private String cdaFile;
    private String dataAccessId;
    private List<CachedParam> parameters;
    private int hitCount, missCount;

    public Query() {
    }

    public Query(JSONObject json) throws JSONException {
        this.cdaFile = getJsonString(json, "cdaFile");
        this.dataAccessId = getJsonString(json, "dataAccessId");

        this.hitCount = getJsonInt(json, "hitCount");
        this.missCount = getJsonInt(json, "missCount");

        if (json.has("parameters")) {
            this.parameters = new ArrayList<CachedParam>();
            Object params = json.get("parameters");

            if (params instanceof JSONArray) {
                for (int i = 0; i < ((JSONArray) params).length(); i++) {
                    this.parameters.add(new CachedParam(((JSONArray) params).getJSONObject(i)));
                }
            } else {
                String[] names = JSONObject.getNames((JSONObject) params);
                if (names != null) {
                    for (String name : names) {
                        this.parameters.add(new CachedParam(name, ((JSONObject) params).getString(name)));
                    }
                }
            }

        }
    }

    public long getId() {
        return id;
    }

    public void setId(long id) {
        this.id = id;
    }

    public String getCdaFile() {
        return cdaFile;
    }

    public void setCdaFile(String cdaFile) {
        this.cdaFile = cdaFile;
    }

    public String getDataAccessId() {
        return dataAccessId;
    }

    public void setDataAccessId(String dataAccessId) {
        this.dataAccessId = dataAccessId;
    }

    public int getHitCount() {
        return hitCount;
    }

    public void setHitCount(int hitCount) {
        this.hitCount = hitCount;
    }

    public int getMissCount() {
        return missCount;
    }

    public void setMissCount(int missCount) {
        this.missCount = missCount;
    }

    public List<CachedParam> getParameters() {
        return parameters;
    }

    public void setParameters(List<CachedParam> parameters) {
        this.parameters = parameters;
    }

    public void addParameter(CachedParam p) {
        if (parameters == null) {
            this.parameters = new ArrayList<CachedParam>();
        }
        this.parameters.add(p);
    }

    public void registerRequest(boolean hit) {
        if (hit) {
            setHitCount(getHitCount() + 1);
        } else {
            setMissCount(getMissCount() + 1);
        }
    }

    public JSONObject toJSON() {
        JSONObject output = new JSONObject();
        JSONObject params = new JSONObject();
        try {
            for (CachedParam param : getParameters()) {
                params.put(param.getName(), param.getValue());
            }
        } catch (JSONException jse) {
            CacheScheduleManager.logger
                    .error("Failed to build parameters for query: " + getCdaFile() + "/" + getDataAccessId());
            return null;
        }
        try {
            output.put("id", getId());

            output.put("cdaFile", getCdaFile());
            output.put("dataAccessId", getDataAccessId());
            output.put("parameters", params);

            output.put("hitCount", getHitCount());
            output.put("missCount", getMissCount());
        } catch (JSONException jse) {
            CacheScheduleManager.logger
                    .error("Failed to build JSON for query: " + getCdaFile() + "/" + getDataAccessId());
            return null;
        }
        return output;
    }

    public void execute() throws Exception {
        try {
            executeQuery();
        } catch (Exception e) {
            CacheScheduleManager.logger.error("Failed to execute query " + toString());
        }
    }

    protected void executeQuery() throws Exception {

        final CdaSettings cdaSettings = SettingsManager.getInstance().parseSettingsFile(getCdaFile());

        final QueryOptions queryOptions = new QueryOptions();
        queryOptions.setDataAccessId(getDataAccessId());
        // force query to be refreshed
        queryOptions.setCacheBypass(true);

        for (Object o : getParameters()) {
            CachedParam param = (CachedParam) o;
            queryOptions.addParameter(param.getName(), param.getValue());
        }

        OutputStream nullOut = new NullOutputStream();
        Date d = new Date();
        CdaEngine.getInstance().doQuery(nullOut, cdaSettings, queryOptions);
        setTimeElapsed(new Date().getTime() - d.getTime());
        CacheScheduleManager.logger
                .debug("Time elapsed: " + Double.toString(new Double(getTimeElapsed()) / 1000) + "s");
    }

    public abstract void setTimeElapsed(long timeElapsed);

    public abstract long getTimeElapsed();

    public void save() throws PluginHibernateException {
        if (getId() == 0) {
            Session s = PluginHibernateUtil.getSession();
            s.save(this);
        }
    }

    protected static String getJsonString(JSONObject json, String expr) {
        try {
            return json.getString(expr);
        } catch (Exception e) {
            return "";
        }
    }

    protected static int getJsonInt(JSONObject json, String expr) {
        try {
            return json.getInt(expr);
        } catch (Exception e) {
            return 0;
        }
    }

    protected static Date getJsonDate(JSONObject json, String expr) {
        try {
            return new Date(json.getLong(expr));
        } catch (Exception e) {
            return new Date(0);
        }
    }

    protected static boolean getJsonBoolean(JSONObject json, String expr) {
        try {
            return json.getBoolean(expr);
        } catch (Exception e) {
            return false;
        }
    }
}