/**
 * 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
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * 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.apache.storm.redis.trident.state;

import com.google.common.collect.Lists;
import java.util.List;
import org.apache.storm.redis.common.mapper.RedisDataTypeDescription;
import org.apache.storm.redis.common.mapper.RedisLookupMapper;
import org.apache.storm.trident.operation.TridentCollector;
import org.apache.storm.trident.state.BaseQueryFunction;
import org.apache.storm.trident.state.State;
import org.apache.storm.trident.tuple.TridentTuple;
import org.apache.storm.tuple.Values;

/**
 * AbstractRedisStateQuerier is base class of any RedisStateQuerier, which implements BaseQueryFunction.
 * <p/>
 * Derived classes should provide how to retrieve values from Redis,
 * and AbstractRedisStateQuerier takes care of rest things.
 *
 * @param <T> type of State
 */
public abstract class AbstractRedisStateQuerier<T extends State> extends BaseQueryFunction<T, List<Values>> {
    protected final RedisDataTypeDescription.RedisDataType dataType;
    protected final String additionalKey;
    private final RedisLookupMapper lookupMapper;

    /**
     * Constructor
     *
     * @param lookupMapper mapper for querying
     */
    public AbstractRedisStateQuerier(RedisLookupMapper lookupMapper) {
        this.lookupMapper = lookupMapper;

        RedisDataTypeDescription dataTypeDescription = lookupMapper.getDataTypeDescription();
        this.dataType = dataTypeDescription.getDataType();
        this.additionalKey = dataTypeDescription.getAdditionalKey();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public List<List<Values>> batchRetrieve(T state, List<TridentTuple> inputs) {
        List<List<Values>> values = Lists.newArrayList();

        List<String> keys = Lists.newArrayList();
        for (TridentTuple input : inputs) {
            keys.add(lookupMapper.getKeyFromTuple(input));
        }

        List<String> redisVals = retrieveValuesFromRedis(state, keys);
        for (int i = 0; i < redisVals.size(); i++) {
            values.add(lookupMapper.toTuple(inputs.get(i), redisVals.get(i)));
        }

        return values;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void execute(TridentTuple tuple, List<Values> values, TridentCollector collector) {
        for (Values value : values) {
            collector.emit(value);
        }
    }

    /**
     * Retrieves values from Redis that each value is corresponding to each key.
     *
     * @param state State for handling query
     * @param keys keys having state values
     * @return values which are corresponding to keys
     */
    protected abstract List<String> retrieveValuesFromRedis(T state, List<String> keys);
}
