org.devmaster.elasticsearch.script.NextOccurrenceSearchScript.java Source code

Introduction

Here is the source code for org.devmaster.elasticsearch.script.NextOccurrenceSearchScript.java

Source

/*
 * 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 org.devmaster.elasticsearch.script;

import org.devmaster.elasticsearch.index.mapper.Recurring;
import org.joda.time.LocalDate;

import java.text.ParseException;
import java.util.HashMap;
import java.util.Map;

public class NextOccurrenceSearchScript extends AbstractRecurringSearchScript {

    public static final String SCRIPT_NAME = "nextOccurrence";

    private static final String PARAM_FIELD = "field";
    private static final String PARAM_FROM = "from";

    public NextOccurrenceSearchScript(Map<String, String> paramMap) {
        super(paramMap);
    }

    public static class Factory extends AbstractRecurringSearchScript.AbstractFactory<NextOccurrenceSearchScript> {

        public Factory() {
            super(NextOccurrenceSearchScript.class, buildParams());
        }

        private static Map<String, Boolean> buildParams() {
            Map<String, Boolean> map = new HashMap<>();
            map.put(PARAM_FIELD, true);
            map.put(PARAM_FROM, false);
            return map;
        }

    }

    @Override
    public Object run() {
        Recurring recurring = getRecurring(getParamValueFor(PARAM_FIELD));
        if (recurring != null) {
            String fromParam = getParamValueFor(PARAM_FROM);
            LocalDate date = fromParam != null ? new LocalDate(fromParam) : LocalDate.now();
            try {
                LocalDate nextOccurrence = recurring.getNextOccurrence(date);
                return nextOccurrence != null ? nextOccurrence.toString() : null;
            } catch (ParseException ignored) {
            }
        }
        return null;
    }

}