Add explicit value mappers for string types

trinodb · Jan 24, 2024 · ca38d29 · ca38d29
1 parent 7dfcdfc
commit ca38d29
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/trino/mapper.py b/trino/mapper.py
@@ -74,6 +74,13 @@ def map(self, value) -> Optional[Decimal]:
         return Decimal(value)
 
 
+class StringValueMapper(ValueMapper[str]):
+    def map(self, value: Any) -> Optional[str]:
+        if value is None:
+            return None
+        return str(value)
+
+
 class BinaryValueMapper(ValueMapper[bytes]):
     def map(self, value) -> Optional[bytes]:
         if value is None:
@@ -252,8 +259,12 @@ def _create_value_mapper(self, column) -> ValueMapper:
             return DoubleValueMapper()
         if col_type == 'decimal':
             return DecimalValueMapper()
+        if col_type in {'varchar', 'char'}:
+            return StringValueMapper()
         if col_type == 'varbinary':
             return BinaryValueMapper()
+        if col_type == 'json':
+            return StringValueMapper()
         if col_type == 'date':
             return DateValueMapper()
         if col_type == 'time':