geode-core/src/main/java/org/apache/geode/cache/query/internal/OrderByComparatorMapped.java

/*
 * 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 org.apache.geode.cache.query.internal;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import it.unimi.dsi.fastutil.objects.Object2ObjectOpenCustomHashMap;

import org.apache.geode.cache.query.FunctionDomainException;
import org.apache.geode.cache.query.NameResolutionException;
import org.apache.geode.cache.query.QueryInvocationTargetException;
import org.apache.geode.cache.query.QueryService;
import org.apache.geode.cache.query.TypeMismatchException;
import org.apache.geode.cache.query.types.ObjectType;
import org.apache.geode.pdx.internal.PdxString;

public class OrderByComparatorMapped extends OrderByComparator {

  private final Map<Object, Object[]> orderByMap;

  public OrderByComparatorMapped(List<CompiledSortCriterion> orderByAttrs, ObjectType objType,
      ExecutionContext context) {
    super(orderByAttrs, objType, context);
    if (objType.isStructType()) {
      orderByMap = new Object2ObjectOpenCustomHashMap<Object, Object[]>(
          new StructBag.ObjectArrayFUHashingStrategy());
    } else {
      this.orderByMap = new HashMap<Object, Object[]>();
    }
  }

  @Override
  void addEvaluatedSortCriteria(Object row, ExecutionContext context)
      throws FunctionDomainException, TypeMismatchException, NameResolutionException,
      QueryInvocationTargetException {
    this.orderByMap.put(row, this.calculateSortCriteria(context, row));
  }

  @Override
  public int evaluateSortCriteria(Object obj1, Object obj2) {
    int result = -1;
    Object[] list1 = this.evaluateSortCriteria(obj1);
    Object[] list2 = this.evaluateSortCriteria(obj2);
    if (list1.length != list2.length) {
      Support.assertionFailed("Error Occurred due to improper sort criteria evaluation ");
    } else {
      for (int i = 0; i < list1.length; i++) {
        Object arr1[] = (Object[]) list1[i];
        Object arr2[] = (Object[]) list2[i];

        if (arr1[0] == null) {
          result = (arr2[0] == null ? 0 : -1);
        } else if (arr2[0] == null) {
          result = 1;
        } else if (arr1[0] == QueryService.UNDEFINED) {
          result = (arr2[0] == QueryService.UNDEFINED ? 0 : -1);
        } else if (arr2[0] == QueryService.UNDEFINED) {
          result = 1;
        } else {
          if (arr1[0] instanceof Number && arr2[0] instanceof Number) {
            Number num1 = (Number) arr1[0];
            Number num2 = (Number) arr2[0];
            result = Double.compare(num1.doubleValue(), num2.doubleValue());
          } else {
            if (arr1[0] instanceof PdxString && arr2[0] instanceof String) {
              arr2[0] = new PdxString((String) arr2[0]);
            } else if (arr2[0] instanceof PdxString && arr1[0] instanceof String) {
              arr1[0] = new PdxString((String) arr1[0]);
            }
            result = ((Comparable) arr1[0]).compareTo(arr2[0]);
          }
        }

        if (result != 0) {
          // not equal, change the sign based on the order by type (asc,
          // desc)
          if (((Boolean) arr1[1]).booleanValue()) {
            result *= -1;
          }
          break;
        }
      }
    }
    return result;
  }

  @Override
  protected Object[] evaluateSortCriteria(Object row) {
    return orderByMap.get(row);
  }

  private Object[] calculateSortCriteria(ExecutionContext context, Object row)
      throws FunctionDomainException, TypeMismatchException, NameResolutionException,
      QueryInvocationTargetException {
    if (orderByAttrs != null) {
      Object[] evaluatedResult = new Object[this.orderByAttrs.size()];
      int index = 0;
      for (CompiledSortCriterion csc : orderByAttrs) {
        Object[] arr = new Object[2];
        if (csc.getColumnIndex() == -1) {
          arr[0] = csc.evaluate(context);
        } else {
          arr[0] = csc.evaluate(row, context);
        }
        arr[1] = csc.getCriterion();
        evaluatedResult[index++] = arr;
      }
      return evaluatedResult;
    }
    return null;
  }
}