use getFieldFromColumnForASTLiteral for all types

Author: yakov-olkhovskiy

src/Analyzer/ConstantNode.cpp

@@ -199,16 +199,12 @@ QueryTreeNodePtr ConstantNode::cloneImpl() const
 
 ASTPtr ConstantNode::toASTImpl(const ConvertToASTOptions & options) const
 {
+    if (!options.add_cast_for_constants)
+        return std::make_shared<ASTLiteral>(getFieldFromColumnForASTLiteral(constant_value.getColumn(), 0, constant_value.getType()));
+
     const auto & constant_value_type = constant_value.getType();
     auto constant_value_ast = std::make_shared<ASTLiteral>(getValue());
 
-    if (!options.add_cast_for_constants)
-    {
-        if (WhichDataType(constant_value_type->getTypeId()).isDateTimeOrDateTime64())
-            return std::make_shared<ASTLiteral>(getFieldFromColumnForASTLiteral(constant_value.getColumn(), 0, constant_value.getType()));
-        return constant_value_ast;
-    }
-
     // Add cast if constant was created as a result of constant folding.
     // Constant folding may lead to type transformation and literal on shard
     // may have a different type.

tests/integration/test_storage_postgresql/test.py

@@ -885,6 +885,12 @@ def test_postgres_datetime(started_cluster):
     result = node1.query("SELECT ts FROM test_datetime WHERE ts > '2025-01-01'::DateTime64")
     assert result == "2025-01-02 03:04:05.678900\n"
 
+    result = node1.query("SELECT ts FROM test_datetime WHERE ts > '2025-01-01'::Nullable(DateTime)")
+    assert result == "2025-01-02 03:04:05.678900\n"
+
+    result = node1.query("SELECT ts FROM test_datetime WHERE ts > '2025-01-01'::Nullable(DateTime64)")
+    assert result == "2025-01-02 03:04:05.678900\n"
+
 
 if __name__ == "__main__":
     cluster.start()