Update FunctionsIsIPAddr.h

pingcap · Feb 14, 2023 · 32cef30 · 32cef30
1 parent b66005f
commit 32cef30
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/dbms/src/Functions/FunctionsIsIPAddr.h b/dbms/src/Functions/FunctionsIsIPAddr.h
@@ -234,17 +234,28 @@ class FunctionIsIPv4OrIsIPv6 : public IFunction
     }
     void executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) const override
     {
+        size_t size = block.getByPosition(arguments[0]).column->size();
+        auto col_res = ColumnUInt8::create();
+        ColumnUInt8::Container & vec_res = col_res->getData();
+        vec_res.resize(size);
+
+        /// Always null.
+        if (block.getByPosition(arguments[0]).type->onlyNull())
+        {
+            for (size_t i = 0; i < size; ++i)
+            {
+                vec_res[i] = 0;
+            }
+            block.getByPosition(result).column = std::move(col_res);
+            return;
+        }
+
         auto [column, nullmap] = removeNullable(block.getByPosition(arguments[0]).column.get());
         if (const auto * col_input = checkAndGetColumn<ColumnString>(column))
         {
-            size_t size = block.getByPosition(arguments[0]).column->size();
             const typename ColumnString::Chars_t & data = col_input->getChars();
             const typename ColumnString::Offsets & offsets = col_input->getOffsets();
 
-            auto col_res = ColumnUInt8::create();
-            ColumnUInt8::Container & vec_res = col_res->getData();
-            vec_res.resize(size);
-
             size_t prev_offset = 0;
             for (size_t i = 0; i < size; ++i)
             {