SNOW-2086682: Add telemetry for XML reader with row tag (#3355)

Author: sfc-gh-jdu

src/snowflake/snowpark/_internal/utils.py

@@ -200,6 +200,8 @@
 XML_ROW_TAG_STRING = "ROWTAG"
 XML_ROW_DATA_COLUMN_NAME = "ROW_DATA"
 XML_READER_FILE_PATH = os.path.join(os.path.dirname(__file__), "xml_reader.py")
+XML_READER_API_SIGNATURE = "DataFrameReader.xml[rowTag]"
+XML_READER_SQL_COMMENT = f"/* Python:snowflake.snowpark.{XML_READER_API_SIGNATURE} */"
 
 QUERY_TAG_STRING = "QUERY_TAG"
 SKIP_LEVELS_TWO = (

src/snowflake/snowpark/dataframe_reader.py

@@ -58,6 +58,7 @@
     XML_ROW_TAG_STRING,
     XML_ROW_DATA_COLUMN_NAME,
     XML_READER_FILE_PATH,
+    XML_READER_API_SIGNATURE,
     INFER_SCHEMA_FORMAT_TYPES,
     SNOWFLAKE_PATH_PREFIXES,
     TempObjectType,
@@ -1159,7 +1160,10 @@ def _read_semi_structured_file(self, path: str, format: str) -> DataFrame:
                 _emit_ast=False,
             )
         df._reader = self
-        set_api_call_source(df, f"DataFrameReader.{format.lower()}")
+        if xml_reader_udtf:
+            set_api_call_source(df, XML_READER_API_SIGNATURE)
+        else:
+            set_api_call_source(df, f"DataFrameReader.{format.lower()}")
         return df
 
     @private_preview(version="1.29.0")