added support for measure columns

Author: shivam2680

src/databricks/sql/backend/thrift_backend.py

@@ -764,14 +764,16 @@ def _col_to_description(col, field=None, session_id_hex=None):
         else:
             precision, scale = None, None
 
-        # Extract variant type from field if available
+        # Extract variant/measure type from field if available
         if field is not None:
             try:
-                # Check for variant type in metadata
+                # Check for variant/measure type in metadata
                 if field.metadata and b"Spark:DataType:SqlName" in field.metadata:
                     sql_type = field.metadata.get(b"Spark:DataType:SqlName")
                     if sql_type == b"VARIANT":
                         cleaned_type = "variant"
+                    if sql_type and b"measure" in sql_type:
+                        cleaned_type += " measure"
             except Exception as e:
                 logger.debug(f"Could not extract variant type from field: {e}")
 

tests/unit/test_thrift_backend.py

@@ -2402,6 +2402,17 @@ def test_hive_schema_to_description(self):
                 ],
                 [("regular_col", "string"), ("variant_col", "variant")],
             ),
+            (
+                [
+                    ("measure_col", ttypes.TTypeId.DOUBLE_TYPE),
+                    ("int_measure_col", ttypes.TTypeId.INT_TYPE),
+                ],
+                [
+                    ("measure_col", {b"Spark:DataType:SqlName": b"double measure"}),
+                    ("int_measure_col", {b"Spark:DataType:SqlName": b"int measure"}),
+                ],
+                [("measure_col", "double measure"), ("int_measure_col", "int measure")],
+            ),
             (
                 [("regular_col", ttypes.TTypeId.STRING_TYPE)],
                 None,  # No arrow schema