#include "function/list/functions/list_unique_function.h"
#include "function/list/vector_list_functions.h"
#include "function/scalar_function.h"
using namespace lbug::common;
namespace lbug {
namespace function {
struct ListDistinct {
static void operation(common::list_entry_t& input, common::list_entry_t& result,
common::ValueVector& inputVector, common::ValueVector& resultVector) {
auto numUniqueValues = ListUnique::appendListElementsToValueSet(input, inputVector);
result = common::ListVector::addList(&resultVector, numUniqueValues);
auto resultDataVector = common::ListVector::getDataVector(&resultVector);
auto resultDataVectorBuffer =
common::ListVector::getListValuesWithOffset(&resultVector, result, 0 );
ListUnique::appendListElementsToValueSet(input, inputVector, nullptr,
[&resultDataVector, &resultDataVectorBuffer](common::ValueVector& dataVector,
uint64_t pos) -> void {
resultDataVector->copyFromVectorData(resultDataVectorBuffer, &dataVector,
dataVector.getData() + pos * dataVector.getNumBytesPerValue());
resultDataVectorBuffer += dataVector.getNumBytesPerValue();
});
}
};
static std::unique_ptr<FunctionBindData> bindFunc(const ScalarBindFuncInput& input) {
return FunctionBindData::getSimpleBindData(input.arguments, input.arguments[0]->getDataType());
}
function_set ListDistinctFunction::getFunctionSet() {
function_set result;
auto function = std::make_unique<ScalarFunction>(name,
std::vector<LogicalTypeID>{LogicalTypeID::LIST}, LogicalTypeID::LIST,
ScalarFunction::UnaryExecNestedTypeFunction<list_entry_t, list_entry_t, ListDistinct>);
function->bindFunc = bindFunc;
result.push_back(std::move(function));
return result;
}
} }