| /** |
| * 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. |
| **/ |
| |
| #include <cstddef> |
| #include <cstdint> |
| #include <cstring> |
| #include <memory> |
| #include <sstream> |
| #include <string> |
| #include <vector> |
| |
| #include "catalog/CatalogTypedefs.hpp" |
| #include "expressions/aggregation/AggregateFunction.hpp" |
| #include "expressions/aggregation/AggregateFunctionFactory.hpp" |
| #include "expressions/aggregation/AggregationHandle.hpp" |
| #include "expressions/aggregation/AggregationHandleMax.hpp" |
| #include "expressions/aggregation/AggregationID.hpp" |
| #include "storage/AggregationOperationState.hpp" |
| #include "storage/HashTableBase.hpp" |
| #include "storage/PackedPayloadHashTable.hpp" |
| #include "storage/StorageManager.hpp" |
| #include "storage/ValueAccessorMultiplexer.hpp" |
| #include "types/CharType.hpp" |
| #include "types/DatetimeIntervalType.hpp" |
| #include "types/DatetimeLit.hpp" |
| #include "types/DatetimeType.hpp" |
| #include "types/DoubleType.hpp" |
| #include "types/FloatType.hpp" |
| #include "types/IntType.hpp" |
| #include "types/IntervalLit.hpp" |
| #include "types/LongType.hpp" |
| #include "types/Type.hpp" |
| #include "types/TypeFactory.hpp" |
| #include "types/TypeID.hpp" |
| #include "types/TypedValue.hpp" |
| #include "types/VarCharType.hpp" |
| #include "types/YearMonthIntervalType.hpp" |
| #include "types/containers/ColumnVector.hpp" |
| #include "types/containers/ColumnVectorsValueAccessor.hpp" |
| |
| #include "types/operations/comparisons/Comparison.hpp" |
| #include "types/operations/comparisons/ComparisonFactory.hpp" |
| #include "types/operations/comparisons/ComparisonID.hpp" |
| |
| #include "gtest/gtest.h" |
| |
| namespace quickstep { |
| |
| class AggregationHandleMaxTest : public ::testing::Test { |
| protected: |
| static const int kNumSamples = 200; |
| static const int kIterations = 5; |
| |
| // Helper method that calls AggregationHandleMax::iterateUnaryInl() to |
| // aggregate 'value' into '*state'. |
| void iterateHandle(AggregationState *state, const TypedValue &value) { |
| static_cast<const AggregationHandleMax &>(*aggregation_handle_max_) |
| .iterateUnaryInl(static_cast<AggregationStateMax *>(state), value); |
| } |
| |
| void initializeHandle(const Type &type) { |
| aggregation_handle_max_.reset( |
| AggregateFunctionFactory::Get(AggregationID::kMax) |
| .createHandle(std::vector<const Type *>(1, &type))); |
| aggregation_handle_max_state_.reset( |
| aggregation_handle_max_->createInitialState()); |
| } |
| |
| static bool ApplyToTypesTest(TypeID typeID) { |
| const Type &type = |
| (typeID == kChar || typeID == kVarChar) |
| ? TypeFactory::GetType(typeID, static_cast<std::size_t>(10)) |
| : TypeFactory::GetType(typeID); |
| |
| return AggregateFunctionFactory::Get(AggregationID::kMax) |
| .canApplyToTypes(std::vector<const Type *>(1, &type)); |
| } |
| |
| static bool ResultTypeForArgumentTypeTest(TypeID input_type_id, |
| TypeID output_type_id) { |
| const Type *result_type = |
| AggregateFunctionFactory::Get(AggregationID::kMax) |
| .resultTypeForArgumentTypes(std::vector<const Type *>( |
| 1, &TypeFactory::GetType(input_type_id))); |
| return (result_type->getTypeID() == output_type_id); |
| } |
| |
| template <typename CppType> |
| static void CheckMaxValue(CppType expected, |
| const AggregationHandle &handle, |
| const AggregationState &state) { |
| EXPECT_EQ(expected, handle.finalize(state).getLiteral<CppType>()); |
| } |
| |
| template <typename CppType> |
| static void CheckMaxValue(CppType expected, const TypedValue &value) { |
| EXPECT_EQ(expected, value.getLiteral<CppType>()); |
| } |
| |
| static void CheckMaxString(const std::string &expected, |
| const AggregationHandle &handle, |
| const AggregationState &state) { |
| TypedValue value = handle.finalize(state); |
| |
| ASSERT_EQ(expected.length(), value.getAsciiStringLength()); |
| EXPECT_EQ(0, |
| std::strncmp(expected.c_str(), |
| static_cast<const char *>(value.getDataPtr()), |
| value.getAsciiStringLength())); |
| } |
| |
| // Static templated method to initialize data types. |
| template <typename CppType> |
| void SetDataType(int value, CppType *data) { |
| *data = value; |
| } |
| |
| template <typename GenericType> |
| void checkAggregationMaxGeneric() { |
| const GenericType &type = GenericType::Instance(true); |
| initializeHandle(type); |
| EXPECT_TRUE( |
| aggregation_handle_max_->finalize(*aggregation_handle_max_state_) |
| .isNull()); |
| |
| typename GenericType::cpptype val; |
| typename GenericType::cpptype max; |
| SetDataType(0, &max); |
| |
| iterateHandle(aggregation_handle_max_state_.get(), type.makeNullValue()); |
| for (int i = 0; i < kIterations; ++i) { |
| for (int j = 0; j < kNumSamples; ++j) { |
| if (type.getTypeID() == kInt || type.getTypeID() == kLong) { |
| SetDataType(i * kNumSamples + j - 10, &val); |
| } else { |
| SetDataType(static_cast<float>(i * kNumSamples + j - 10) / 10, &val); |
| } |
| iterateHandle(aggregation_handle_max_state_.get(), |
| type.makeValue(&val)); |
| if (max < val) { |
| max = val; |
| } |
| } |
| } |
| iterateHandle(aggregation_handle_max_state_.get(), type.makeNullValue()); |
| CheckMaxValue<typename GenericType::cpptype>( |
| max, *aggregation_handle_max_, *aggregation_handle_max_state_); |
| |
| // Test mergeStates(). |
| std::unique_ptr<AggregationState> merge_state( |
| aggregation_handle_max_->createInitialState()); |
| aggregation_handle_max_->mergeStates(*merge_state, |
| aggregation_handle_max_state_.get()); |
| |
| iterateHandle(merge_state.get(), type.makeNullValue()); |
| for (int i = 0; i < kIterations; ++i) { |
| for (int j = kNumSamples - 1; j >= 0; --j) { |
| if (type.getTypeID() == kInt || type.getTypeID() == kLong) { |
| SetDataType(i * kNumSamples + j - 20, &val); |
| } else { |
| SetDataType(static_cast<float>(i * kNumSamples + j - 20) / 10, &val); |
| } |
| iterateHandle(merge_state.get(), type.makeValue(&val)); |
| if (max < val) { |
| max = val; |
| } |
| } |
| } |
| aggregation_handle_max_->mergeStates(*merge_state, |
| aggregation_handle_max_state_.get()); |
| CheckMaxValue<typename GenericType::cpptype>( |
| max, *aggregation_handle_max_, *aggregation_handle_max_state_); |
| } |
| |
| template <typename GenericType> |
| ColumnVector* createColumnVectorGeneric(const Type &type, |
| typename GenericType::cpptype *max) { |
| NativeColumnVector *column = |
| new NativeColumnVector(type, kIterations * kNumSamples + 3); |
| |
| typename GenericType::cpptype val; |
| SetDataType(0, max); |
| |
| column->appendTypedValue(type.makeNullValue()); |
| for (int i = 0; i < kIterations; ++i) { |
| for (int j = 0; j < kNumSamples; ++j) { |
| if (type.getTypeID() == kInt || type.getTypeID() == kLong) { |
| SetDataType(i * kNumSamples + j - 10, &val); |
| } else { |
| SetDataType(static_cast<float>(i * kNumSamples + j - 10) / 10, &val); |
| } |
| column->appendTypedValue(type.makeValue(&val)); |
| if (*max < val) { |
| *max = val; |
| } |
| } |
| // One NULL in the middle. |
| if (i == kIterations / 2) { |
| column->appendTypedValue(type.makeNullValue()); |
| } |
| } |
| column->appendTypedValue(type.makeNullValue()); |
| |
| return column; |
| } |
| |
| template <typename GenericType> |
| void checkAggregationMaxGenericValueAccessor() { |
| const GenericType &type = GenericType::Instance(true); |
| initializeHandle(type); |
| EXPECT_TRUE( |
| aggregation_handle_max_->finalize(*aggregation_handle_max_state_) |
| .isNull()); |
| |
| std::unique_ptr<ColumnVectorsValueAccessor> accessor( |
| new ColumnVectorsValueAccessor()); |
| |
| typename GenericType::cpptype max; |
| accessor->addColumn(createColumnVectorGeneric<GenericType>(type, &max)); |
| |
| std::unique_ptr<AggregationState> va_state( |
| aggregation_handle_max_->accumulateValueAccessor( |
| {MultiSourceAttributeId(ValueAccessorSource::kBase, 0)}, |
| ValueAccessorMultiplexer(accessor.get()))); |
| |
| // Test the state generated directly by accumulateValueAccessor(), and also |
| // test after merging back. |
| CheckMaxValue<typename GenericType::cpptype>( |
| max, *aggregation_handle_max_, *va_state); |
| |
| aggregation_handle_max_->mergeStates(*va_state, |
| aggregation_handle_max_state_.get()); |
| CheckMaxValue<typename GenericType::cpptype>( |
| max, *aggregation_handle_max_, *aggregation_handle_max_state_); |
| } |
| |
| template <typename StringType> |
| void checkAggregationMaxString() { |
| const StringType &type = StringType::Instance(10, true); |
| initializeHandle(type); |
| EXPECT_TRUE( |
| aggregation_handle_max_->finalize(*aggregation_handle_max_state_) |
| .isNull()); |
| |
| std::unique_ptr<UncheckedComparator> fast_comparator_; |
| fast_comparator_.reset( |
| ComparisonFactory::GetComparison(ComparisonID::kGreater) |
| .makeUncheckedComparatorForTypes(type, type)); |
| std::string string_literal; |
| std::string max = ""; |
| int val; |
| iterateHandle(aggregation_handle_max_state_.get(), type.makeNullValue()); |
| for (int i = 0; i < kIterations; ++i) { |
| for (int j = 0; j < kNumSamples; ++j) { |
| val = i * kNumSamples + j; |
| std::ostringstream oss; |
| oss << "max" << val; |
| string_literal = oss.str(); |
| |
| iterateHandle( |
| aggregation_handle_max_state_.get(), |
| type.makeValue(string_literal.c_str(), string_literal.length() + 1) |
| .ensureNotReference()); |
| if (fast_comparator_->compareDataPtrs(string_literal.c_str(), |
| max.c_str())) { |
| max = string_literal; |
| } |
| } |
| } |
| iterateHandle(aggregation_handle_max_state_.get(), type.makeNullValue()); |
| CheckMaxString( |
| max, *aggregation_handle_max_, *aggregation_handle_max_state_); |
| |
| // Test mergeStates(). |
| std::unique_ptr<AggregationState> merge_state( |
| aggregation_handle_max_->createInitialState()); |
| aggregation_handle_max_->mergeStates(*merge_state, |
| aggregation_handle_max_state_.get()); |
| |
| iterateHandle(merge_state.get(), type.makeNullValue()); |
| for (int i = 0; i < kIterations; ++i) { |
| for (int j = kNumSamples - 1; j >= 0; --j) { |
| val = i * kNumSamples + j; |
| std::ostringstream oss; |
| oss << "test" << val; |
| string_literal = oss.str(); |
| |
| iterateHandle( |
| merge_state.get(), |
| type.makeValue(string_literal.c_str(), string_literal.length() + 1) |
| .ensureNotReference()); |
| if (fast_comparator_->compareDataPtrs(string_literal.c_str(), |
| max.c_str())) { |
| max = string_literal; |
| } |
| } |
| } |
| aggregation_handle_max_->mergeStates(*merge_state, |
| aggregation_handle_max_state_.get()); |
| CheckMaxString( |
| max, *aggregation_handle_max_, *aggregation_handle_max_state_); |
| } |
| |
| template <typename ColumnVectorType> |
| ColumnVector* createColumnVectorString(const Type &type, std::string *max) { |
| ColumnVectorType *column = |
| new ColumnVectorType(type, kIterations * kNumSamples + 3); |
| std::unique_ptr<UncheckedComparator> fast_comparator_; |
| fast_comparator_.reset( |
| ComparisonFactory::GetComparison(ComparisonID::kGreater) |
| .makeUncheckedComparatorForTypes(type, type)); |
| std::string string_literal; |
| *max = ""; |
| int val; |
| column->appendTypedValue(type.makeNullValue()); |
| for (int i = 0; i < kIterations; ++i) { |
| for (int j = 0; j < kNumSamples; ++j) { |
| val = i * kNumSamples + j; |
| std::ostringstream oss; |
| oss << "max" << val; |
| string_literal = oss.str(); |
| |
| column->appendTypedValue( |
| type.makeValue(string_literal.c_str(), string_literal.length() + 1) |
| .ensureNotReference()); |
| if (fast_comparator_->compareDataPtrs(string_literal.c_str(), |
| max->c_str())) { |
| *max = string_literal; |
| } |
| } |
| // One NULL in the middle. |
| if (i == kIterations / 2) { |
| column->appendTypedValue(type.makeNullValue()); |
| } |
| } |
| column->appendTypedValue(type.makeNullValue()); |
| |
| return column; |
| } |
| |
| template <typename StringType, typename ColumnVectorType> |
| void checkAggregationMaxStringValueAccessor() { |
| const StringType &type = StringType::Instance(10, true); |
| initializeHandle(type); |
| EXPECT_TRUE( |
| aggregation_handle_max_->finalize(*aggregation_handle_max_state_) |
| .isNull()); |
| |
| std::string max; |
| std::unique_ptr<ColumnVectorsValueAccessor> accessor( |
| new ColumnVectorsValueAccessor()); |
| accessor->addColumn(createColumnVectorString<ColumnVectorType>(type, &max)); |
| |
| std::unique_ptr<AggregationState> va_state( |
| aggregation_handle_max_->accumulateValueAccessor( |
| {MultiSourceAttributeId(ValueAccessorSource::kBase, 0)}, |
| ValueAccessorMultiplexer(accessor.get()))); |
| |
| // Test the state generated directly by accumulateValueAccessor(), and also |
| // test after merging back. |
| CheckMaxString(max, *aggregation_handle_max_, *va_state); |
| |
| aggregation_handle_max_->mergeStates(*va_state, |
| aggregation_handle_max_state_.get()); |
| CheckMaxString( |
| max, *aggregation_handle_max_, *aggregation_handle_max_state_); |
| } |
| |
| std::unique_ptr<AggregationHandle> aggregation_handle_max_; |
| std::unique_ptr<AggregationState> aggregation_handle_max_state_; |
| std::unique_ptr<StorageManager> storage_manager_; |
| }; |
| |
| template <> |
| void AggregationHandleMaxTest::CheckMaxValue<float>( |
| float val, const AggregationHandle &handle, const AggregationState &state) { |
| EXPECT_FLOAT_EQ(val, handle.finalize(state).getLiteral<float>()); |
| } |
| |
| template <> |
| void AggregationHandleMaxTest::CheckMaxValue<double>( |
| double val, |
| const AggregationHandle &handle, |
| const AggregationState &state) { |
| EXPECT_DOUBLE_EQ(val, handle.finalize(state).getLiteral<double>()); |
| } |
| |
| template <> |
| void AggregationHandleMaxTest::SetDataType<DatetimeLit>(int value, |
| DatetimeLit *data) { |
| data->ticks = value; |
| } |
| |
| template <> |
| void AggregationHandleMaxTest::SetDataType<DatetimeIntervalLit>( |
| int value, DatetimeIntervalLit *data) { |
| data->interval_ticks = value; |
| } |
| |
| template <> |
| void AggregationHandleMaxTest::SetDataType<YearMonthIntervalLit>( |
| int value, YearMonthIntervalLit *data) { |
| data->months = value; |
| } |
| |
| typedef AggregationHandleMaxTest AggregationHandleMaxDeathTest; |
| |
| TEST_F(AggregationHandleMaxTest, IntTypeTest) { |
| checkAggregationMaxGeneric<IntType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, LongTypeTest) { |
| checkAggregationMaxGeneric<LongType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, FloatTypeTest) { |
| checkAggregationMaxGeneric<FloatType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, DoubleTypeTest) { |
| checkAggregationMaxGeneric<DoubleType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, DatetimeType) { |
| checkAggregationMaxGeneric<DatetimeType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, DatetimeIntervalType) { |
| checkAggregationMaxGeneric<DatetimeIntervalType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, YearMonthIntervalType) { |
| checkAggregationMaxGeneric<YearMonthIntervalType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, CharTypeTest) { |
| checkAggregationMaxString<CharType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, VarCharTypeTest) { |
| checkAggregationMaxString<VarCharType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, IntTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<IntType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, LongTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<LongType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, FloatTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<FloatType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, DoubleTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<DoubleType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, DatetimeTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<DatetimeType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, DatetimeIntervalTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<DatetimeIntervalType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, YearMonthIntervalTypeValueAccessorTest) { |
| checkAggregationMaxGenericValueAccessor<YearMonthIntervalType>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, CharTypeValueAccessorTest) { |
| checkAggregationMaxStringValueAccessor<CharType, NativeColumnVector>(); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, VarCharValueAccessorTypeTest) { |
| checkAggregationMaxStringValueAccessor<VarCharType, IndirectColumnVector>(); |
| } |
| |
| #ifdef QUICKSTEP_DEBUG |
| TEST_F(AggregationHandleMaxDeathTest, WrongTypeTest) { |
| const Type &int_non_null_type = IntType::Instance(false); |
| const Type &long_type = LongType::Instance(true); |
| const Type &double_type = DoubleType::Instance(true); |
| const Type &float_type = FloatType::Instance(true); |
| const Type &char_type = CharType::Instance(true, 10); |
| const Type &varchar_type = VarCharType::Instance(true, 10); |
| |
| initializeHandle(IntType::Instance(true)); |
| int int_val = 0; |
| std::int64_t long_val = 0; |
| double double_val = 0; |
| float float_val = 0; |
| |
| // Passes. |
| iterateHandle(aggregation_handle_max_state_.get(), |
| int_non_null_type.makeValue(&int_val)); |
| |
| EXPECT_DEATH(iterateHandle(aggregation_handle_max_state_.get(), |
| long_type.makeValue(&long_val)), |
| ""); |
| EXPECT_DEATH(iterateHandle(aggregation_handle_max_state_.get(), |
| double_type.makeValue(&double_val)), |
| ""); |
| EXPECT_DEATH(iterateHandle(aggregation_handle_max_state_.get(), |
| float_type.makeValue(&float_val)), |
| ""); |
| EXPECT_DEATH(iterateHandle(aggregation_handle_max_state_.get(), |
| char_type.makeValue("asdf", 5)), |
| ""); |
| EXPECT_DEATH(iterateHandle(aggregation_handle_max_state_.get(), |
| varchar_type.makeValue("asdf", 5)), |
| ""); |
| |
| // Test mergeStates() with incorrectly typed handles. |
| std::unique_ptr<AggregationHandle> aggregation_handle_max_long( |
| AggregateFunctionFactory::Get(AggregationID::kMax) |
| .createHandle(std::vector<const Type *>(1, &long_type))); |
| std::unique_ptr<AggregationState> aggregation_state_max_merge_long( |
| aggregation_handle_max_long->createInitialState()); |
| static_cast<const AggregationHandleMax &>(*aggregation_handle_max_long) |
| .iterateUnaryInl(static_cast<AggregationStateMax *>( |
| aggregation_state_max_merge_long.get()), |
| long_type.makeValue(&long_val)); |
| EXPECT_DEATH( |
| aggregation_handle_max_->mergeStates(*aggregation_state_max_merge_long, |
| aggregation_handle_max_state_.get()), |
| ""); |
| |
| std::unique_ptr<AggregationHandle> aggregation_handle_max_double( |
| AggregateFunctionFactory::Get(AggregationID::kMax) |
| .createHandle(std::vector<const Type *>(1, &double_type))); |
| std::unique_ptr<AggregationState> aggregation_state_max_merge_double( |
| aggregation_handle_max_double->createInitialState()); |
| static_cast<const AggregationHandleMax &>(*aggregation_handle_max_double) |
| .iterateUnaryInl(static_cast<AggregationStateMax *>( |
| aggregation_state_max_merge_double.get()), |
| double_type.makeValue(&double_val)); |
| EXPECT_DEATH( |
| aggregation_handle_max_->mergeStates(*aggregation_state_max_merge_double, |
| aggregation_handle_max_state_.get()), |
| ""); |
| |
| std::unique_ptr<AggregationHandle> aggregation_handle_max_float( |
| AggregateFunctionFactory::Get(AggregationID::kMax) |
| .createHandle(std::vector<const Type *>(1, &float_type))); |
| std::unique_ptr<AggregationState> aggregation_state_max_merge_float( |
| aggregation_handle_max_float->createInitialState()); |
| static_cast<const AggregationHandleMax &>(*aggregation_handle_max_float) |
| .iterateUnaryInl(static_cast<AggregationStateMax *>( |
| aggregation_state_max_merge_float.get()), |
| float_type.makeValue(&float_val)); |
| EXPECT_DEATH( |
| aggregation_handle_max_->mergeStates(*aggregation_state_max_merge_float, |
| aggregation_handle_max_state_.get()), |
| ""); |
| } |
| #endif |
| |
| TEST_F(AggregationHandleMaxTest, canApplyToTypeTest) { |
| EXPECT_TRUE(ApplyToTypesTest(kInt)); |
| EXPECT_TRUE(ApplyToTypesTest(kLong)); |
| EXPECT_TRUE(ApplyToTypesTest(kFloat)); |
| EXPECT_TRUE(ApplyToTypesTest(kDouble)); |
| EXPECT_TRUE(ApplyToTypesTest(kChar)); |
| EXPECT_TRUE(ApplyToTypesTest(kVarChar)); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, ResultTypeForArgumentTypeTest) { |
| EXPECT_TRUE(ResultTypeForArgumentTypeTest(kInt, kInt)); |
| EXPECT_TRUE(ResultTypeForArgumentTypeTest(kLong, kLong)); |
| EXPECT_TRUE(ResultTypeForArgumentTypeTest(kFloat, kFloat)); |
| EXPECT_TRUE(ResultTypeForArgumentTypeTest(kDouble, kDouble)); |
| } |
| |
| TEST_F(AggregationHandleMaxTest, GroupByTableMergeTest) { |
| const Type &int_non_null_type = IntType::Instance(false); |
| initializeHandle(int_non_null_type); |
| storage_manager_.reset(new StorageManager("./test_max_data")); |
| std::unique_ptr<AggregationStateHashTableBase> source_hash_table( |
| AggregationStateHashTableFactory::CreateResizable( |
| HashTableImplType::kSeparateChaining, |
| std::vector<const Type *>(1, &int_non_null_type), |
| 10, |
| {aggregation_handle_max_.get()}, |
| storage_manager_.get())); |
| std::unique_ptr<AggregationStateHashTableBase> destination_hash_table( |
| AggregationStateHashTableFactory::CreateResizable( |
| HashTableImplType::kSeparateChaining, |
| std::vector<const Type *>(1, &int_non_null_type), |
| 10, |
| {aggregation_handle_max_.get()}, |
| storage_manager_.get())); |
| |
| PackedPayloadHashTable *destination_hash_table_derived = |
| static_cast<PackedPayloadHashTable *>(destination_hash_table.get()); |
| |
| PackedPayloadHashTable *source_hash_table_derived = |
| static_cast<PackedPayloadHashTable *>(source_hash_table.get()); |
| |
| AggregationHandleMax *aggregation_handle_max_derived = |
| static_cast<AggregationHandleMax *>(aggregation_handle_max_.get()); |
| // We create three keys: first is present in both the hash tables, second key |
| // is present only in the source hash table while the third key is present |
| // the destination hash table only. |
| std::vector<TypedValue> common_key; |
| common_key.emplace_back(0); |
| std::vector<TypedValue> exclusive_source_key, exclusive_destination_key; |
| exclusive_source_key.emplace_back(1); |
| exclusive_destination_key.emplace_back(2); |
| |
| const int common_key_source_max = 3000; |
| TypedValue common_key_source_max_val(common_key_source_max); |
| |
| const int common_key_destination_max = 4000; |
| TypedValue common_key_destination_max_val(common_key_destination_max); |
| |
| const int exclusive_key_source_max = 100; |
| TypedValue exclusive_key_source_max_val(exclusive_key_source_max); |
| |
| const int exclusive_key_destination_max = 200; |
| TypedValue exclusive_key_destination_max_val(exclusive_key_destination_max); |
| |
| std::unique_ptr<AggregationStateMax> common_key_source_state( |
| static_cast<AggregationStateMax *>( |
| aggregation_handle_max_->createInitialState())); |
| std::unique_ptr<AggregationStateMax> common_key_destination_state( |
| static_cast<AggregationStateMax *>( |
| aggregation_handle_max_->createInitialState())); |
| std::unique_ptr<AggregationStateMax> exclusive_key_source_state( |
| static_cast<AggregationStateMax *>( |
| aggregation_handle_max_->createInitialState())); |
| std::unique_ptr<AggregationStateMax> exclusive_key_destination_state( |
| static_cast<AggregationStateMax *>( |
| aggregation_handle_max_->createInitialState())); |
| |
| // Create max value states for keys. |
| aggregation_handle_max_derived->iterateUnaryInl(common_key_source_state.get(), |
| common_key_source_max_val); |
| int actual_val = aggregation_handle_max_->finalize(*common_key_source_state) |
| .getLiteral<int>(); |
| EXPECT_EQ(common_key_source_max_val.getLiteral<int>(), actual_val); |
| |
| aggregation_handle_max_derived->iterateUnaryInl( |
| common_key_destination_state.get(), common_key_destination_max_val); |
| actual_val = aggregation_handle_max_->finalize(*common_key_destination_state) |
| .getLiteral<int>(); |
| EXPECT_EQ(common_key_destination_max_val.getLiteral<int>(), actual_val); |
| |
| aggregation_handle_max_derived->iterateUnaryInl( |
| exclusive_key_destination_state.get(), exclusive_key_destination_max_val); |
| actual_val = |
| aggregation_handle_max_->finalize(*exclusive_key_destination_state) |
| .getLiteral<int>(); |
| EXPECT_EQ(exclusive_key_destination_max_val.getLiteral<int>(), actual_val); |
| |
| aggregation_handle_max_derived->iterateUnaryInl( |
| exclusive_key_source_state.get(), exclusive_key_source_max_val); |
| actual_val = aggregation_handle_max_->finalize(*exclusive_key_source_state) |
| .getLiteral<int>(); |
| EXPECT_EQ(exclusive_key_source_max_val.getLiteral<int>(), actual_val); |
| |
| // Add the key-state pairs to the hash tables. |
| unsigned char buffer[100]; |
| buffer[0] = '\0'; |
| memcpy(buffer + 1, |
| common_key_source_state.get()->getPayloadAddress(), |
| aggregation_handle_max_.get()->getPayloadSize()); |
| source_hash_table_derived->upsertCompositeKey(common_key, buffer); |
| |
| memcpy(buffer + 1, |
| common_key_destination_state.get()->getPayloadAddress(), |
| aggregation_handle_max_.get()->getPayloadSize()); |
| destination_hash_table_derived->upsertCompositeKey(common_key, buffer); |
| |
| memcpy(buffer + 1, |
| exclusive_key_source_state.get()->getPayloadAddress(), |
| aggregation_handle_max_.get()->getPayloadSize()); |
| source_hash_table_derived->upsertCompositeKey(exclusive_source_key, buffer); |
| |
| memcpy(buffer + 1, |
| exclusive_key_destination_state.get()->getPayloadAddress(), |
| aggregation_handle_max_.get()->getPayloadSize()); |
| destination_hash_table_derived->upsertCompositeKey(exclusive_destination_key, |
| buffer); |
| |
| EXPECT_EQ(2u, destination_hash_table_derived->numEntries()); |
| EXPECT_EQ(2u, source_hash_table_derived->numEntries()); |
| |
| HashTableMerger merger(destination_hash_table_derived); |
| source_hash_table_derived->forEachCompositeKey(&merger); |
| |
| EXPECT_EQ(3u, destination_hash_table_derived->numEntries()); |
| |
| CheckMaxValue<int>( |
| common_key_destination_max_val.getLiteral<int>(), |
| aggregation_handle_max_derived->finalizeHashTableEntry( |
| destination_hash_table_derived->getSingleCompositeKey(common_key) + |
| 1)); |
| CheckMaxValue<int>( |
| exclusive_key_destination_max_val.getLiteral<int>(), |
| aggregation_handle_max_derived->finalizeHashTableEntry( |
| destination_hash_table_derived->getSingleCompositeKey( |
| exclusive_destination_key) + 1)); |
| CheckMaxValue<int>( |
| exclusive_key_source_max_val.getLiteral<int>(), |
| aggregation_handle_max_derived->finalizeHashTableEntry( |
| source_hash_table_derived->getSingleCompositeKey( |
| exclusive_source_key) + 1)); |
| } |
| |
| } // namespace quickstep |
