SNOW-1480779 Support Method Chain in the Telemetry (#121)

* create methodChain

* alias

* rollup

* groupByGroupingSets

* limit

* na

* chain

* add method chain

* update

* update test

* fix test

* fix test

* fix test

* fix test

* fix test

* reformat

* fix test

* rename

src/main/java/com/snowflake/snowpark_java/types/CloudProviderToken.java

@@ -1,8 +1,9 @@
 package com.snowflake.snowpark_java.types;
 
-/** The Snowflake class provides access to the CloudProviderToken secret object with the following properties: accessKeyId, 
- * secretAccessKey, and token.*/
-
+/**
+ * The Snowflake class provides access to the CloudProviderToken secret object with the following
+ * properties: accessKeyId, secretAccessKey, and token.
+ */
 public class CloudProviderToken {
   private final String accessKeyId;
   private final String secretAccessKey;

src/main/java/com/snowflake/snowpark_java/types/SnowflakeSecrets.java

@@ -32,8 +32,8 @@ public UsernamePassword getUsernamePassword(String secretName) {
   }
 
   /**
-   * Get the Cloud provider token from the secret. On success, it returns a valid object with access key id,
-   * secret access key and token.
+   * Get the Cloud provider token from the secret. On success, it returns a valid object with access
+   * key id, secret access key and token.
    *
    * @param secretName name of the secret object.
    */

src/main/scala/com/snowflake/snowpark/CopyableDataFrame.scala

@@ -19,8 +19,9 @@ import com.snowflake.snowpark.internal.analyzer._
 class CopyableDataFrame private[snowpark] (
     override private[snowpark] val session: Session,
     override private[snowpark] val plan: SnowflakePlan,
+    override private[snowpark] val methodChain: Seq[String],
     private val stagedFileReader: StagedFileReader)
-    extends DataFrame(session, plan) {
+    extends DataFrame(session, plan, methodChain) {
 
   /**
    * Executes a `COPY INTO <table_name>` command to
@@ -238,7 +239,7 @@ class CopyableDataFrame private[snowpark] (
    * @group basic
    */
   override def clone: CopyableDataFrame = action("clone", 2) {
-    new CopyableDataFrame(session, plan, stagedFileReader)
+    new CopyableDataFrame(session, plan, Seq(), stagedFileReader)
   }
 
   /**
@@ -261,11 +262,12 @@ class CopyableDataFrame private[snowpark] (
 
   @inline override protected def action[T](funcName: String)(func: => T): T = {
     val isScala: Boolean = this.session.conn.isScalaAPI
-    OpenTelemetry.action("CopyableDataFrame", funcName, isScala)(func)
+    OpenTelemetry.action("CopyableDataFrame", funcName, methodChainString, isScala)(func)
   }
   @inline protected def action[T](funcName: String, javaOffset: Int)(func: => T): T = {
     val isScala: Boolean = this.session.conn.isScalaAPI
-    OpenTelemetry.action("CopyableDataFrame", funcName, isScala, javaOffset)(func)
+    OpenTelemetry.action("CopyableDataFrame", funcName, methodChainString, isScala, javaOffset)(
+      func)
   }
 }
 
@@ -360,6 +362,10 @@ class CopyableDataFrameAsyncActor private[snowpark] (cdf: CopyableDataFrame)
 
   @inline override protected def action[T](funcName: String)(func: => T): T = {
     val isScala: Boolean = cdf.session.conn.isScalaAPI
-    OpenTelemetry.action("CopyableDataFrameAsyncActor", funcName, isScala)(func)
+    OpenTelemetry.action(
+      "CopyableDataFrameAsyncActor",
+      funcName,
+      cdf.methodChainString + ".async",
+      isScala)(func)
   }
 }