QueryCountIndicator.java

/**
 * EasySOA Registry
 * Copyright 2012-2013 Open Wide
 * 
 * This program 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 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 Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * Contact : easysoa-dev@googlegroups.com
 */

package org.easysoa.registry.indicators.rest;

import java.util.Map;

import org.easysoa.registry.utils.NXQLQueryHelper;
import org.nuxeo.ecm.core.api.ClientException;
import org.nuxeo.ecm.core.api.CoreSession;
import org.nuxeo.ecm.core.api.IterableQueryResult;
import org.nuxeo.ecm.core.query.sql.NXQL;

/**
 * NXQL-based indicator provider
 *
 * @author mkalam-alami
 *
 */
public abstract class QueryCountIndicator extends Indicator {

    private String valueQuery = null;
    private String totalQuery = null;

    public QueryCountIndicator(String valueQuery) {
        this.valueQuery = valueQuery;
    }

    public QueryCountIndicator(String valueQuery, String totalQuery) {
        this.valueQuery = valueQuery;
        this.totalQuery = totalQuery;
    }

    @Override
    public IndicatorValue compute(CoreSession session, String subprojectId,
            Map<String, IndicatorValue> computedIndicators, String visibility) throws ClientException {
        //subprojectId = SubprojectServiceImpl.getSubprojectIdOrCreateDefault(session, subprojectId);
        // TODO default or not ??
    	
    	String subprojectCriteria = NXQLQueryHelper.buildSubprojectCriteria(session, subprojectId, visibility);

        IterableQueryResult queryResult = session.queryAndFetch(valueQuery + subprojectCriteria, NXQL.NXQL);
        try {
            if (this.totalQuery == null) {
                return new IndicatorValue(this.getName(), this.getCategory(), queryResult.size(), -1);
            } else {
                IterableQueryResult totalQueryResult = null;
                try {
                    totalQueryResult = session.queryAndFetch(totalQuery, NXQL.NXQL);
                    
                    //TODO : Total query stored in the percentage value - how is calculated the percentage ??
                    return new IndicatorValue(this.getName(), this.getCategory(), queryResult.size(), totalQueryResult.size());
                } finally {
                    if (totalQueryResult != null) {
                        totalQueryResult.close();
                    }
                }
            }
        } finally {
            queryResult.close();
        }
    }
}