From 7c6322c5b03232a0dfd603a54b2e0037bf817122 Mon Sep 17 00:00:00 2001
From: Sofia Antipushina <ssb_99@mail.ru>
Date: Sat, 16 May 2020 02:06:25 +0300
Subject: [PATCH] Add support for many columns

---
 src/AggregateFunctions/AggregateFunctionDistinct.h | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/AggregateFunctions/AggregateFunctionDistinct.h b/src/AggregateFunctions/AggregateFunctionDistinct.h
index 5580cc3b4df..b87183f15d6 100644
--- a/src/AggregateFunctions/AggregateFunctionDistinct.h
+++ b/src/AggregateFunctions/AggregateFunctionDistinct.h
@@ -42,12 +42,13 @@ class AggregateFunctionDistinct final : public IAggregateFunctionHelper<Aggregat
 {
 private:
     AggregateFunctionPtr nested_func;
+    size_t num_arguments;
     mutable AggregateFunctionDistinctData storage;
 
 public:
     AggregateFunctionDistinct(AggregateFunctionPtr nested, const DataTypes & arguments)
     : IAggregateFunctionHelper<AggregateFunctionDistinct>(arguments, {})
-    , nested_func(nested)
+    , nested_func(nested), num_arguments(arguments.size())
     {
         if (arguments.empty())
             throw Exception("Aggregate function " + getName() + " require at least one argument", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
@@ -90,7 +91,8 @@ public:
     void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
     {
         SipHash hash;
-        columns[0]->updateHashWithValue(row_num, hash);
+        for (size_t i = 0; i < num_arguments; ++i)
+            columns[i]->updateHashWithValue(row_num, hash);
 
         UInt128 key;
         hash.get128(key.low, key.high);