You've already forked mariadb-columnstore-engine
							
							
				mirror of
				https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
				synced 2025-11-03 17:13:17 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			908 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			908 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* Copyright (C) 2014 InfiniDB, Inc.
 | 
						|
   Copyright (C) 2019 MariaDB Corporation
 | 
						|
 | 
						|
   This program is free software; you can redistribute it and/or
 | 
						|
   modify it under the terms of the GNU General Public License
 | 
						|
   as published by the Free Software Foundation; version 2 of
 | 
						|
   the License.
 | 
						|
 | 
						|
   This program is distributed in the hope that it will be useful,
 | 
						|
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
   GNU General Public License for more details.
 | 
						|
 | 
						|
   You should have received a copy of the GNU General Public License
 | 
						|
   along with this program; if not, write to the Free Software
 | 
						|
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 | 
						|
   MA 02110-1301, USA. */
 | 
						|
 | 
						|
//  $Id: idborderby.cpp 3932 2013-06-25 16:08:10Z xlou $
 | 
						|
 | 
						|
#include <iostream>
 | 
						|
#include <string>
 | 
						|
#include <stack>
 | 
						|
using namespace std;
 | 
						|
 | 
						|
#include "objectreader.h"
 | 
						|
#include "calpontselectexecutionplan.h"
 | 
						|
#include "rowgroup.h"
 | 
						|
 | 
						|
using namespace boost;
 | 
						|
 | 
						|
#include "errorids.h"
 | 
						|
#include "exceptclasses.h"
 | 
						|
using namespace logging;
 | 
						|
 | 
						|
#include "calpontsystemcatalog.h"
 | 
						|
using namespace execplan;
 | 
						|
 | 
						|
#include "resourcemanager.h"
 | 
						|
using namespace joblist;
 | 
						|
 | 
						|
using namespace rowgroup;
 | 
						|
 | 
						|
#include "idborderby.h"
 | 
						|
 | 
						|
#include "joblisttypes.h"
 | 
						|
#include "mcs_decimal.h"
 | 
						|
 | 
						|
// See agg_arg_charsets in sql_type.h to see conversion rules for
 | 
						|
// items that have different char sets
 | 
						|
namespace ordering
 | 
						|
{
 | 
						|
int TinyIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  int8_t v1 = l->row1().getIntField(fSpec.fIndex);
 | 
						|
  int8_t v2 = l->row2().getIntField(fSpec.fIndex);
 | 
						|
  int8_t nullValue = static_cast<int8_t>(joblist::TINYINTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int SmallIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  int16_t v1 = l->row1().getIntField(fSpec.fIndex);
 | 
						|
  int16_t v2 = l->row2().getIntField(fSpec.fIndex);
 | 
						|
  int16_t nullValue = static_cast<int16_t>(joblist::SMALLINTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int IntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  int32_t v1 = l->row1().getIntField(fSpec.fIndex);
 | 
						|
  int32_t v2 = l->row2().getIntField(fSpec.fIndex);
 | 
						|
  int32_t nullValue = static_cast<int32_t>(joblist::INTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int BigIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  int64_t v1 = l->row1().getIntField(fSpec.fIndex);
 | 
						|
  int64_t v2 = l->row2().getIntField(fSpec.fIndex);
 | 
						|
  int64_t nullValue = static_cast<int64_t>(joblist::BIGINTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int WideDecimalCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  int128_t v1, v2;
 | 
						|
  l->row1().getInt128Field(fSpec.fIndex, v1);
 | 
						|
  l->row2().getInt128Field(fSpec.fIndex, v2);
 | 
						|
 | 
						|
  bool v1IsNull = v1 == datatypes::Decimal128Null;
 | 
						|
  bool v2IsNull = v2 == datatypes::Decimal128Null;
 | 
						|
 | 
						|
  if (v1IsNull || v2IsNull)
 | 
						|
  {
 | 
						|
    if (!v1IsNull && v2IsNull)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1IsNull && !v2IsNull)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int UTinyIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  uint8_t v1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint8_t v2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
  uint8_t nullValue = static_cast<uint8_t>(joblist::UTINYINTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int USmallIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  uint16_t v1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint16_t v2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
  uint16_t nullValue = static_cast<uint16_t>(joblist::USMALLINTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int UIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  uint32_t v1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint32_t v2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
  uint32_t nullValue = static_cast<uint32_t>(joblist::UINTNULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int UBigIntCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  uint64_t v1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint64_t v2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
 | 
						|
  if (v1 == joblist::UBIGINTNULL || v2 == joblist::UBIGINTNULL)
 | 
						|
  {
 | 
						|
    if (v1 != joblist::UBIGINTNULL && v2 == joblist::UBIGINTNULL)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == joblist::UBIGINTNULL && v2 != joblist::UBIGINTNULL)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int StringCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  bool b1 = l->row1().isNullValue(fSpec.fIndex);
 | 
						|
  bool b2 = l->row2().isNullValue(fSpec.fIndex);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  if (b1 == true || b2 == true)
 | 
						|
  {
 | 
						|
    if (b1 == false && b2 == true)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (b1 == true && b2 == false)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    auto const str1 = l->row1().getConstString(fSpec.fIndex);
 | 
						|
    auto const str2 = l->row2().getConstString(fSpec.fIndex);
 | 
						|
 | 
						|
    if (!cs)
 | 
						|
      cs = l->rowGroup()->getCharset(fSpec.fIndex);
 | 
						|
 | 
						|
    ret = fSpec.fAsc * cs->strnncollsp(str1.str(), str1.length(), str2.str(), str2.length());
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int DoubleCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  uint64_t uiv1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint64_t uiv2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  if (uiv1 == joblist::DOUBLENULL || uiv2 == joblist::DOUBLENULL)
 | 
						|
  {
 | 
						|
    if (uiv1 != joblist::DOUBLENULL && uiv2 == joblist::DOUBLENULL)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (uiv1 == joblist::DOUBLENULL && uiv2 != joblist::DOUBLENULL)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    double v1 = l->row1().getDoubleField(fSpec.fIndex);
 | 
						|
    double v2 = l->row2().getDoubleField(fSpec.fIndex);
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int FloatCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int32_t iv1 = l->row1().getIntField(fSpec.fIndex);
 | 
						|
  int32_t iv2 = l->row2().getIntField(fSpec.fIndex);
 | 
						|
  int32_t nullValue = static_cast<int32_t>(joblist::FLOATNULL);
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  if (iv1 == nullValue || iv2 == nullValue)
 | 
						|
  {
 | 
						|
    if (iv1 != nullValue && iv2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (iv1 == nullValue && iv2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    float v1 = l->row1().getFloatField(fSpec.fIndex);
 | 
						|
    float v2 = l->row2().getFloatField(fSpec.fIndex);
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int LongDoubleCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  long double v1 = l->row1().getLongDoubleField(fSpec.fIndex);
 | 
						|
  long double v2 = l->row2().getLongDoubleField(fSpec.fIndex);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  if (v1 == joblist::LONGDOUBLENULL || v2 == joblist::LONGDOUBLENULL)
 | 
						|
  {
 | 
						|
    if (v1 != joblist::LONGDOUBLENULL && v2 == joblist::LONGDOUBLENULL)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == joblist::LONGDOUBLENULL && v2 != joblist::LONGDOUBLENULL)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int DateCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  uint32_t v1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint32_t v2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
  uint32_t nullValue = static_cast<uint32_t>(joblist::DATENULL);
 | 
						|
 | 
						|
  if (v1 == nullValue || v2 == nullValue)
 | 
						|
  {
 | 
						|
    if (v1 != nullValue && v2 == nullValue)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == nullValue && v2 != nullValue)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int DatetimeCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
  uint64_t v1 = l->row1().getUintField(fSpec.fIndex);
 | 
						|
  uint64_t v2 = l->row2().getUintField(fSpec.fIndex);
 | 
						|
 | 
						|
  if (v1 == joblist::DATETIMENULL || v2 == joblist::DATETIMENULL)
 | 
						|
  {
 | 
						|
    if (v1 != joblist::DATETIMENULL && v2 == joblist::DATETIMENULL)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (v1 == joblist::DATETIMENULL && v2 != joblist::DATETIMENULL)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (v1 > v2)
 | 
						|
      ret = fSpec.fAsc;
 | 
						|
    else if (v1 < v2)
 | 
						|
      ret = -fSpec.fAsc;
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int TimeCompare::operator()(IdbCompare* l, Row::Pointer r1, Row::Pointer r2)
 | 
						|
{
 | 
						|
  l->row1().setData(r1);
 | 
						|
  l->row2().setData(r2);
 | 
						|
 | 
						|
  int64_t v1 = l->row1().getIntField(fSpec.fIndex);
 | 
						|
  int64_t v2 = l->row2().getIntField(fSpec.fIndex);
 | 
						|
 | 
						|
  bool b1 = (joblist::TIMENULL == (uint64_t)v1);
 | 
						|
  bool b2 = (joblist::TIMENULL == (uint64_t)v2);
 | 
						|
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  if (b1 == true || b2 == true)
 | 
						|
  {
 | 
						|
    if (b1 == false && b2 == true)
 | 
						|
      ret = fSpec.fNf;
 | 
						|
    else if (b1 == true && b2 == false)
 | 
						|
      ret = -fSpec.fNf;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    // ((int64_t) -00:00:26) > ((int64_t) -00:00:25)
 | 
						|
    // i.e. For 2 negative TIME values, we invert the order of
 | 
						|
    // comparison operations to force "-00:00:26" to appear before
 | 
						|
    // "-00:00:25" in ascending order.
 | 
						|
    if (v1 < 0 && v2 < 0)
 | 
						|
    {
 | 
						|
      // Unset the MSB.
 | 
						|
      v1 &= ~(1ULL << 63);
 | 
						|
      v2 &= ~(1ULL << 63);
 | 
						|
      if (v1 < v2)
 | 
						|
        ret = fSpec.fAsc;
 | 
						|
      else if (v1 > v2)
 | 
						|
        ret = -fSpec.fAsc;
 | 
						|
    }
 | 
						|
    else
 | 
						|
    {
 | 
						|
      if (v1 > v2)
 | 
						|
        ret = fSpec.fAsc;
 | 
						|
      else if (v1 < v2)
 | 
						|
        ret = -fSpec.fAsc;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
bool CompareRule::less(Row::Pointer r1, Row::Pointer r2) const
 | 
						|
{
 | 
						|
  for (auto* compare : fCompares)
 | 
						|
  {
 | 
						|
    int c = (*compare)(fIdbCompare, r1, r2);
 | 
						|
 | 
						|
    if (c < 0)
 | 
						|
      return true;
 | 
						|
    else if (c > 0)
 | 
						|
      return false;
 | 
						|
  }
 | 
						|
 | 
						|
  return false;
 | 
						|
}
 | 
						|
 | 
						|
void CompareRule::revertRules()
 | 
						|
{
 | 
						|
  auto fCompareIter = fCompares.begin();
 | 
						|
  for (; fCompareIter != fCompares.end(); fCompareIter++)
 | 
						|
  {
 | 
						|
    (*fCompareIter)->revertSortSpec();
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
void CompareRule::compileRules(const std::vector<IdbSortSpec>& spec, const rowgroup::RowGroup& rg)
 | 
						|
{
 | 
						|
  const vector<CalpontSystemCatalog::ColDataType>& types = rg.getColTypes();
 | 
						|
  const auto& offsets = rg.getOffsets();
 | 
						|
 | 
						|
  for (auto spec_el : spec)
 | 
						|
  {
 | 
						|
    switch (types[spec_el.fIndex])
 | 
						|
    {
 | 
						|
      case CalpontSystemCatalog::TINYINT:
 | 
						|
      {
 | 
						|
        Compare* c = new TinyIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::SMALLINT:
 | 
						|
      {
 | 
						|
        Compare* c = new SmallIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::MEDINT:
 | 
						|
      case CalpontSystemCatalog::INT:
 | 
						|
      {
 | 
						|
        Compare* c = new IntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::BIGINT:
 | 
						|
      {
 | 
						|
        Compare* c = new BigIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::DECIMAL:
 | 
						|
      case CalpontSystemCatalog::UDECIMAL:
 | 
						|
      {
 | 
						|
        Compare* c;
 | 
						|
        uint32_t len = rg.getColumnWidth(spec_el.fIndex);
 | 
						|
        switch (len)
 | 
						|
        {
 | 
						|
          case datatypes::MAXDECIMALWIDTH:
 | 
						|
            c = new WideDecimalCompare(spec_el, offsets[spec_el.fIndex]);
 | 
						|
            break;
 | 
						|
          case datatypes::MAXLEGACYWIDTH: c = new BigIntCompare(spec_el); break;
 | 
						|
          case 1: c = new TinyIntCompare(spec_el); break;
 | 
						|
          case 2: c = new SmallIntCompare(spec_el); break;
 | 
						|
          case 4: c = new IntCompare(spec_el); break;
 | 
						|
        }
 | 
						|
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::UTINYINT:
 | 
						|
      {
 | 
						|
        Compare* c = new UTinyIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::USMALLINT:
 | 
						|
      {
 | 
						|
        Compare* c = new USmallIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::UMEDINT:
 | 
						|
      case CalpontSystemCatalog::UINT:
 | 
						|
      {
 | 
						|
        Compare* c = new UIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::UBIGINT:
 | 
						|
      {
 | 
						|
        Compare* c = new UBigIntCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::CHAR:
 | 
						|
      case CalpontSystemCatalog::VARCHAR:
 | 
						|
      case CalpontSystemCatalog::TEXT:
 | 
						|
      {
 | 
						|
        Compare* c = new StringCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::DOUBLE:
 | 
						|
      case CalpontSystemCatalog::UDOUBLE:
 | 
						|
      {
 | 
						|
        Compare* c = new DoubleCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::FLOAT:
 | 
						|
      case CalpontSystemCatalog::UFLOAT:
 | 
						|
      {
 | 
						|
        Compare* c = new FloatCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::LONGDOUBLE:
 | 
						|
      {
 | 
						|
        Compare* c = new LongDoubleCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::DATE:
 | 
						|
      {
 | 
						|
        Compare* c = new DateCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::DATETIME:
 | 
						|
      case CalpontSystemCatalog::TIMESTAMP:
 | 
						|
      {
 | 
						|
        Compare* c = new DatetimeCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
      case CalpontSystemCatalog::TIME:
 | 
						|
      {
 | 
						|
        Compare* c = new TimeCompare(spec_el);
 | 
						|
        fCompares.push_back(c);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      default:
 | 
						|
      {
 | 
						|
        break;
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
void IdbCompare::initialize(const RowGroup& rg)
 | 
						|
{
 | 
						|
  fRowGroup = rg;
 | 
						|
  fRowGroup.initRow(&fRow1);
 | 
						|
  fRowGroup.initRow(&fRow2);
 | 
						|
}
 | 
						|
 | 
						|
void IdbCompare::setStringTable(bool b)
 | 
						|
{
 | 
						|
  fRowGroup.setUseStringTable(b);
 | 
						|
  fRowGroup.initRow(&fRow1);
 | 
						|
  fRowGroup.initRow(&fRow2);
 | 
						|
}
 | 
						|
 | 
						|
// OrderByData class implementation
 | 
						|
OrderByData::OrderByData(const std::vector<IdbSortSpec>& spec, const rowgroup::RowGroup& rg)
 | 
						|
{
 | 
						|
  IdbCompare::initialize(rg);
 | 
						|
  fRule.compileRules(spec, rg);
 | 
						|
  fRule.fIdbCompare = this;
 | 
						|
}
 | 
						|
 | 
						|
// OrderByData class dtor
 | 
						|
OrderByData::~OrderByData()
 | 
						|
{
 | 
						|
  // delete compare objects
 | 
						|
  auto i = fRule.fCompares.begin();
 | 
						|
 | 
						|
  while (i != fRule.fCompares.end())
 | 
						|
  {
 | 
						|
    if (*i)
 | 
						|
    {
 | 
						|
      delete *i;
 | 
						|
      *i = NULL;
 | 
						|
    }
 | 
						|
    i++;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
// IdbOrderBy class implementation
 | 
						|
IdbOrderBy::IdbOrderBy()
 | 
						|
 : fDistinct(false), fMemSize(0), fRowsPerRG(rowgroup::rgCommonSize), fErrorCode(0), fRm(nullptr)
 | 
						|
{
 | 
						|
}
 | 
						|
 | 
						|
IdbOrderBy::~IdbOrderBy()
 | 
						|
{
 | 
						|
  // returnRGDataMemory2RM() returns all memory before the dtor is called.
 | 
						|
  if (fRm && fMemSize > 0)
 | 
						|
    fRm->returnMemory(fMemSize, fSessionMemLimit);
 | 
						|
 | 
						|
  // delete compare objects
 | 
						|
  vector<Compare*>::iterator i = fRule.fCompares.begin();
 | 
						|
 | 
						|
  while (i != fRule.fCompares.end())
 | 
						|
    delete *i++;
 | 
						|
}
 | 
						|
 | 
						|
// INV fRm is not NULL here
 | 
						|
void IdbOrderBy::initialize(const RowGroup& rg)
 | 
						|
{
 | 
						|
  // initialize rows
 | 
						|
  IdbCompare::initialize(rg);
 | 
						|
 | 
						|
  auto newSize = rg.getSizeWithStrings(fRowsPerRG);
 | 
						|
  if (!fRm->getMemory(newSize, fSessionMemLimit))
 | 
						|
  {
 | 
						|
    cerr << IDBErrorInfo::instance()->errorMsg(fErrorCode) << " @" << __FILE__ << ":" << __LINE__;
 | 
						|
    throw IDBExcept(fErrorCode);
 | 
						|
  }
 | 
						|
  fMemSize += newSize;
 | 
						|
  fData.reinit(fRowGroup, fRowsPerRG);
 | 
						|
  fRowGroup.setData(&fData);
 | 
						|
  fRowGroup.resetRowGroup(0);
 | 
						|
  fRowGroup.initRow(&fRow0);
 | 
						|
  fRowGroup.getRow(0, &fRow0);
 | 
						|
 | 
						|
  // set compare functors
 | 
						|
  fRule.compileRules(fOrderByCond, fRowGroup);
 | 
						|
 | 
						|
  fRowGroup.initRow(&row1);
 | 
						|
  fRowGroup.initRow(&row2);
 | 
						|
 | 
						|
  // These two blocks contain structs with memory accounting.
 | 
						|
    fOrderByQueue.reset(new SortingPQ(rowgroup::rgCommonSize, fRm->getAllocator<OrderByRow>()));
 | 
						|
 | 
						|
  if (fDistinct)
 | 
						|
  {
 | 
						|
    fDistinctMap.reset(new DistinctMap_t(10, Hasher(this, getKeyLength()), Eq(this, getKeyLength()),
 | 
						|
                                         utils::STLPoolAllocator<rowgroup::Row::Pointer>(fRm)));
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
bool IdbOrderBy::getData(RGData& data)
 | 
						|
{
 | 
						|
  if (fDataQueue.empty())
 | 
						|
    return false;
 | 
						|
 | 
						|
  data = fDataQueue.front();
 | 
						|
  fDataQueue.pop();
 | 
						|
 | 
						|
  return true;
 | 
						|
}
 | 
						|
 | 
						|
bool EqualCompData::operator()(Row::Pointer a, Row::Pointer b)
 | 
						|
{
 | 
						|
  bool eq = true;
 | 
						|
  fRow1.setData(a);
 | 
						|
  fRow2.setData(b);
 | 
						|
 | 
						|
  for (vector<uint64_t>::const_iterator i = fIndex.begin(); i != fIndex.end() && eq; i++)
 | 
						|
  {
 | 
						|
    CalpontSystemCatalog::ColDataType type = fRow1.getColType(*i);
 | 
						|
 | 
						|
    switch (type)
 | 
						|
    {
 | 
						|
      case CalpontSystemCatalog::TINYINT:
 | 
						|
      case CalpontSystemCatalog::SMALLINT:
 | 
						|
      case CalpontSystemCatalog::MEDINT:
 | 
						|
      case CalpontSystemCatalog::INT:
 | 
						|
      case CalpontSystemCatalog::BIGINT:
 | 
						|
      case CalpontSystemCatalog::UTINYINT:
 | 
						|
      case CalpontSystemCatalog::USMALLINT:
 | 
						|
      case CalpontSystemCatalog::UMEDINT:
 | 
						|
      case CalpontSystemCatalog::UINT:
 | 
						|
      case CalpontSystemCatalog::UBIGINT:
 | 
						|
      case CalpontSystemCatalog::DATE:
 | 
						|
      case CalpontSystemCatalog::DATETIME:
 | 
						|
      case CalpontSystemCatalog::TIMESTAMP:
 | 
						|
      case CalpontSystemCatalog::TIME:
 | 
						|
      {
 | 
						|
        // equal compare. ignore sign and null
 | 
						|
        eq = (fRow1.getUintField(*i) == fRow2.getUintField(*i));
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::DECIMAL:
 | 
						|
      case CalpontSystemCatalog::UDECIMAL:
 | 
						|
      {
 | 
						|
        // equal compare. ignore sign and null
 | 
						|
        if (UNLIKELY(fRow1.getColumnWidth(*i) < datatypes::MAXDECIMALWIDTH))
 | 
						|
        {
 | 
						|
          eq = (fRow1.getUintField(*i) == fRow2.getUintField(*i));
 | 
						|
        }
 | 
						|
        else if (fRow1.getColumnWidth(*i) == datatypes::MAXDECIMALWIDTH)
 | 
						|
        {
 | 
						|
          datatypes::TSInt128 left = fRow1.getTSInt128Field(*i);
 | 
						|
          datatypes::TSInt128 right = fRow2.getTSInt128Field(*i);
 | 
						|
          eq = left.getValue() == right.getValue();
 | 
						|
        }
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::CHAR:
 | 
						|
      case CalpontSystemCatalog::VARCHAR:
 | 
						|
      {
 | 
						|
        eq = (fRow1.getStringField(*i) == fRow2.getStringField(*i));
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::DOUBLE:
 | 
						|
      case CalpontSystemCatalog::UDOUBLE:
 | 
						|
      {
 | 
						|
        eq = (fRow1.getDoubleField(*i) == fRow2.getDoubleField(*i));
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::FLOAT:
 | 
						|
      case CalpontSystemCatalog::UFLOAT:
 | 
						|
      {
 | 
						|
        eq = (fRow1.getFloatField(*i) == fRow2.getFloatField(*i));
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      case CalpontSystemCatalog::LONGDOUBLE:
 | 
						|
      {
 | 
						|
        eq = (fRow1.getLongDoubleField(*i) == fRow2.getLongDoubleField(*i));
 | 
						|
        break;
 | 
						|
      }
 | 
						|
 | 
						|
      default:
 | 
						|
      {
 | 
						|
        eq = false;
 | 
						|
        uint64_t ec = ERR_WF_UNKNOWN_COL_TYPE;
 | 
						|
        cerr << IDBErrorInfo::instance()->errorMsg(ec, type) << " @" << __FILE__ << ":" << __LINE__;
 | 
						|
        throw IDBExcept(IDBErrorInfo::instance()->errorMsg(ec, type), ec);
 | 
						|
        break;
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return eq;
 | 
						|
}
 | 
						|
 | 
						|
uint64_t IdbOrderBy::Hasher::operator()(const Row::Pointer& p) const
 | 
						|
{
 | 
						|
  Row& row = ts->row1;
 | 
						|
  row.setPointer(p);
 | 
						|
  // MCOL-1829 Row::hash uses colcount - 1 as an array idx down a callstack.
 | 
						|
  uint64_t ret = row.hash(colCount - 1);
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
bool IdbOrderBy::Eq::operator()(const Row::Pointer& d1, const Row::Pointer& d2) const
 | 
						|
{
 | 
						|
  Row &r1 = ts->row1, &r2 = ts->row2;
 | 
						|
  r1.setPointer(d1);
 | 
						|
  r2.setPointer(d2);
 | 
						|
  // MCOL-1829 Row::equals uses 2nd argument as key columns container size boundary
 | 
						|
  bool ret = r1.equals(r2, colCount - 1);
 | 
						|
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
}  // namespace ordering
 |