com.redis.om.spring.EnhancedRedisQueryEngine Maven / Gradle / Ivy

Go to download
package com.redis.om.spring;

import java.util.Collection;
import java.util.Comparator;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import org.springframework.data.keyvalue.core.CriteriaAccessor;
import org.springframework.data.keyvalue.core.QueryEngine;
import org.springframework.data.keyvalue.core.SortAccessor;
import org.springframework.data.keyvalue.core.SpelSortAccessor;
import org.springframework.data.keyvalue.core.query.KeyValueQuery;
import org.springframework.data.redis.core.RedisKeyValueAdapter;
import org.springframework.data.redis.repository.query.RedisOperationChain;
import org.springframework.expression.spel.standard.SpelExpressionParser;

public class EnhancedRedisQueryEngine extends QueryEngine> {

  private static final Log logger = LogFactory.getLog(EnhancedRedisQueryEngine.class);

  public EnhancedRedisQueryEngine() {
    this(new RedisCriteriaAccessor(), new SpelSortAccessor(new SpelExpressionParser()));
  }

  public EnhancedRedisQueryEngine(CriteriaAccessor criteriaAccessor,
      SortAccessor> sortAccessor) {
    super(criteriaAccessor, sortAccessor);
  }

  /* (non-Javadoc)
   * 
   * @see
   * org.springframework.data.keyvalue.core.QueryEngine#execute(java.lang.Object,
   * java.lang.Object, int, int, java.lang.String, java.lang.Class) */
  @Override
  @SuppressWarnings("unchecked")
  public  Collection execute(RedisOperationChain criteria, Comparator sort, long offset, int rows,
      String keyspace, Class type) {
    logger.debug(String.format("Executing %s", criteria));
    List result = List.of();
    // TODO: implement me!

    if (sort != null) {
      result.sort((Comparator) sort);
    }

    return result;
  }

  @Override
  public Collection execute(RedisOperationChain criteria, Comparator sort, long offset, int rows,
      String keyspace) {
    return execute(criteria, sort, offset, rows, keyspace, Object.class);
  }

  @Override
  public long count(RedisOperationChain criteria, String keyspace) {
    if (criteria == null || criteria.isEmpty()) {
      return this.getAdapter().count(keyspace);
    }

    return 0;
  }

  static class RedisCriteriaAccessor implements CriteriaAccessor {

    @Override
    public RedisOperationChain resolve(KeyValueQuery query) {
      return (RedisOperationChain) query.getCriteria();
    }
  }

}