com.fcorti.aaar.GetNodeIdsModifiedBeforeWebScript.java Source code

Java tutorial

Introduction

Here is the source code for com.fcorti.aaar.GetNodeIdsModifiedBeforeWebScript.java

Source

/*
Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements.  See the NOTICE file distributed with
this work for additional information regarding copyright ownership.
The ASF licenses this file to You 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.fcorti.aaar;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.alfresco.model.ContentModel;
import org.alfresco.service.cmr.repository.StoreRef;
import org.alfresco.service.cmr.search.LimitBy;
import org.alfresco.service.cmr.search.PermissionEvaluationMode;
import org.alfresco.service.cmr.search.ResultSet;
import org.alfresco.service.cmr.search.ResultSetRow;
import org.alfresco.service.cmr.search.SearchParameters;
import org.alfresco.service.cmr.search.SearchService;
import org.apache.commons.math3.exception.OutOfRangeException;
import org.springframework.extensions.webscripts.Cache;
import org.springframework.extensions.webscripts.DeclarativeWebScript;
import org.springframework.extensions.webscripts.Status;
import org.springframework.extensions.webscripts.WebScriptException;
import org.springframework.extensions.webscripts.WebScriptRequest;

/**
 * Get nodes modified after a date time for A.A.A.R. analytics.
 *
 * @author fcorti@gmail.com
 * @since 2.1.0
 */
public class GetNodeIdsModifiedBeforeWebScript extends DeclarativeWebScript {

    private static final String PARAMETER_BASETYPE = "baseType";
    private static final String PARAMETER_DATE = "dt";
    private static final String PARAMETER_LIMIT = "limit";
    private static final int PARAMETER_LIMIT_DEFAULT = 10000;
    private static final String PARAMETER_SKIP = "skip";
    private static final int PARAMETER_SKIP_DEFAULT = 0;
    private static final String DATE_FORMAT = "yyyy-MM-dd";

    private SearchService searchService;

    protected Map<String, Object> executeImpl(WebScriptRequest req, Status status, Cache cache) {

        // Parameters.
        Map<String, Object> parameters = null;
        try {
            parameters = getParameters(req);
        } catch (Exception e) {
            throw new WebScriptException(Status.STATUS_BAD_REQUEST, e.getMessage());
        }

        // Query execution.
        ResultSet resultSet = searchService.query(getSearchParameters(parameters));
        Iterator<ResultSetRow> resultIterator = resultSet.iterator();

        // New parameter values.
        Date newParameterDt = (Date) parameters.get(PARAMETER_DATE);
        int newParameterSkip = (int) parameters.get(PARAMETER_SKIP);

        // Result composition.
        List<String> results = new ArrayList<String>();
        while (resultIterator.hasNext()) {

            ResultSetRow resultSetRow = resultIterator.next();
            resultSetRow.getValues();

            results.add(String.valueOf(resultSetRow.getValue(ContentModel.PROP_NODE_DBID)));

            ++newParameterSkip;
        }

        resultSet.close();

        // Parameter values.
        parameters.replace(PARAMETER_DATE, getDateAsString((Date) parameters.get(PARAMETER_DATE), DATE_FORMAT));
        parameters.replace(PARAMETER_LIMIT, String.valueOf(parameters.get(PARAMETER_LIMIT)));
        parameters.replace(PARAMETER_SKIP, String.valueOf(parameters.get(PARAMETER_SKIP)));

        // New parameters values.
        Map<String, Object> newParameters = new HashMap<String, Object>();
        newParameters.put(PARAMETER_BASETYPE, parameters.get(PARAMETER_BASETYPE));
        newParameters.put(PARAMETER_DATE, getDateAsString(newParameterDt, DATE_FORMAT));
        newParameters.put(PARAMETER_LIMIT, parameters.get(PARAMETER_LIMIT));
        newParameters.put(PARAMETER_SKIP, String.valueOf(newParameterSkip));

        // Model definition.
        Map<String, Object> model = new HashMap<String, Object>();
        model.put("count", String.valueOf(results.size()));
        model.put("results", results);
        model.put("parameters", parameters);
        model.put("newParameters", newParameters);

        return model;
    }

    /**
     * Get the search parameters.
     * 
     * @param parameters
     * @return
     */
    private static final SearchParameters getSearchParameters(Map<String, Object> parameters) {

        SearchParameters searchParameters = new SearchParameters();

        searchParameters.addStore(StoreRef.STORE_REF_WORKSPACE_SPACESSTORE);
        searchParameters.setLanguage(SearchService.LANGUAGE_FTS_ALFRESCO);
        searchParameters.setPermissionEvaluation(PermissionEvaluationMode.NONE);
        searchParameters.setUseInMemorySort(false);
        searchParameters.setLimitBy(LimitBy.FINAL_SIZE);
        searchParameters.setLimit((int) parameters.get(PARAMETER_LIMIT));
        searchParameters.setQuery(getQuery(parameters));
        searchParameters.setSkipCount((int) parameters.get(PARAMETER_SKIP));
        searchParameters.addSort("@" + ContentModel.PROP_NODE_DBID, true);

        return searchParameters;
    }

    /**
     * Get the query string.
     * 
     * @param parameters
     * @return
     */
    private static final String getQuery(Map<String, Object> parameters) {

        String query = "";

        // Base type filter.
        query += "(TYPE:\"" + parameters.get(PARAMETER_BASETYPE) + "\") ";

        // Date filter.
        query += "AND ";
        query += "(@" + ContentModel.PROP_MODIFIED + ":[MIN TO \""
                + getDateAsString((Date) parameters.get(PARAMETER_DATE), DATE_FORMAT) + "\">) ";

        return query;
    }

    /**
     * Get parameters from the request.
     * 
     * @param req
     * @return
     * @throws Exception 
     */
    private static final Map<String, Object> getParameters(WebScriptRequest req) throws Exception {

        Map<String, Object> parameters = new HashMap<String, Object>();

        // BaseType parameter.
        String baseTypeParameter = req.getParameter(PARAMETER_BASETYPE);
        if (baseTypeParameter == null) {
            throw new Exception("Parameter '" + PARAMETER_BASETYPE + "' not specified.");
        }
        baseTypeParameter = baseTypeParameter.trim();
        if (baseTypeParameter.isEmpty()) {
            throw new Exception("Parameter '" + PARAMETER_BASETYPE + "' cannot be empty.");
        }

        // Date parameter.
        String dateParameter = req.getParameter(PARAMETER_DATE);
        if (dateParameter == null) {
            throw new Exception("Parameter '" + PARAMETER_DATE + "' not specified.");
        }
        dateParameter = dateParameter.trim();
        if (dateParameter.isEmpty()) {
            throw new Exception("Parameter '" + PARAMETER_DATE + "' cannot be empty.");
        }
        Date dateParameterValue = null;
        try {
            dateParameterValue = (new SimpleDateFormat(DATE_FORMAT)).parse(dateParameter);
        } catch (Exception e) {
            throw new Exception(
                    "Parameter '" + PARAMETER_DATE + "' with a wrong format. Request '" + DATE_FORMAT + "'.");
        }

        // Limit parameter.
        String limitParameter = req.getParameter(PARAMETER_LIMIT);
        if (limitParameter == null) {
            limitParameter = "" + PARAMETER_LIMIT_DEFAULT;
        }
        limitParameter = limitParameter.trim();
        if (limitParameter.isEmpty()) {
            throw new Exception("Parameter '" + PARAMETER_LIMIT + "' cannot be empty.");
        }
        int limitParameterValue = Integer.parseInt(limitParameter);
        if (limitParameterValue <= 0) {
            throw new OutOfRangeException(limitParameterValue, 0, Integer.MAX_VALUE);
        }

        // Skip parameter.
        String skipParameter = req.getParameter(PARAMETER_SKIP);
        if (skipParameter == null) {
            skipParameter = "" + PARAMETER_SKIP_DEFAULT;
        }
        skipParameter = skipParameter.trim();
        if (skipParameter.isEmpty()) {
            throw new Exception("Parameter '" + PARAMETER_SKIP + "' cannot be empty.");
        }
        int skipParameterValue = Integer.parseInt(skipParameter);
        if (skipParameterValue < 0) {
            throw new OutOfRangeException(skipParameterValue, 0, Long.MAX_VALUE);
        }

        parameters.put(PARAMETER_BASETYPE, baseTypeParameter);
        parameters.put(PARAMETER_DATE, dateParameterValue);
        parameters.put(PARAMETER_LIMIT, limitParameterValue);
        parameters.put(PARAMETER_SKIP, skipParameterValue);

        return parameters;
    }

    /**
     * Get the string representation of the date.
     * 
     * @param date
     * @param format
     * @return
     */
    private static final String getDateAsString(Date date, String format) {
        return (new SimpleDateFormat(format)).format(date);
    }

    public void setSearchService(SearchService searchService) {
        this.searchService = searchService;
    }
}