com.redis.om.spring.search.stream.predicates.numeric.EqualPredicate Maven / Gradle / Ivy

Go to download
package com.redis.om.spring.search.stream.predicates.numeric;

import java.lang.reflect.Field;
import java.time.Instant;
import java.time.LocalDate;
import java.time.ZoneId;

import com.redis.om.spring.search.stream.predicates.BaseAbstractPredicate;

import io.redisearch.querybuilder.Node;
import io.redisearch.querybuilder.QueryBuilder;
import io.redisearch.querybuilder.Values;

public class EqualPredicate extends BaseAbstractPredicate {
  private T value;

  public EqualPredicate(Field field, T value) {
    super(field);
    this.value = value;
  }

  public T getValue() {
    return value;
  }

  @Override
  public Node apply(Node root) {
    Class cls = getValue().getClass();
    if (cls == LocalDate.class) {
      LocalDate localDate = (LocalDate) getValue();
      Instant instant = localDate.atStartOfDay(ZoneId.systemDefault()).toInstant();
      Long unixTime = instant.getEpochSecond();
      return QueryBuilder.intersect(root).add(getField().getName(), Values.eq(unixTime));
    } else if (cls == Integer.class) {
      return QueryBuilder.intersect(root).add(getField().getName(), Values.eq(Integer.valueOf(getValue().toString())));
    } else if (cls == Double.class) {
      return QueryBuilder.intersect(root).add(getField().getName(), Values.eq(Double.valueOf(getValue().toString())));
    } else {
      return root;
    }
  }

}