From c3651c3a0aa0e034a61695f6d8505dcb4fc2a74b Mon Sep 17 00:00:00 2001
From: Vasily Nemkov <V.Nemkov@gmail.com>
Date: Tue, 10 Mar 2020 15:07:13 +0800
Subject: [PATCH] Fixed formatDateTime to utilize timezone of
 DateTime/DateTime64 argument

---
 dbms/src/Functions/formatDateTime.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dbms/src/Functions/formatDateTime.cpp b/dbms/src/Functions/formatDateTime.cpp
index 5a3ee7b1c48..ce6a30aac89 100644
--- a/dbms/src/Functions/formatDateTime.cpp
+++ b/dbms/src/Functions/formatDateTime.cpp
@@ -314,8 +314,8 @@ public:
         size_t result_size = pattern_to_fill.size();
 
         const DateLUTImpl * time_zone_tmp = nullptr;
-        if (arguments.size() == 3)
-            time_zone_tmp = &extractTimeZoneFromFunctionArguments(block, arguments, 2, 0);
+        if (std::is_same_v<DataType, DataTypeDateTime64> || std::is_same_v<DataType, DataTypeDateTime>)
+           time_zone_tmp = &extractTimeZoneFromFunctionArguments(block, arguments, 3, 0);
         else
             time_zone_tmp = &DateLUT::instance();