Emitting trace events in Scala¶

Adding support for the Telemetry API¶

To use Telemetry methods, you must make the Snowflake Telemetry library, which is included with Snowflake, available to your handler code.

• In the PACKAGES clause in your CREATE PROCEDURE or CREATE FUNCTION statement, include the com.snowflake:telemetry package. The PACKAGES clause makes the included Snowflake Telemetry API available to your code.

  Code in the following example uses the PACKAGES clause to reference the Telemetry library as well as the Snowpark library (which is required for stored procedures written in Scala – for more information, see Writing stored procedures in Scala).

  CREATE OR REPLACE PROCEDURE MYPROC(...)
    RETURNS ...
    LANGUAGE SCALA
    ...
    PACKAGES = ('com.snowflake:snowpark:latest', 'com.snowflake:telemetry:latest')
    ...
  

  Copy

• Import the com.snowflake.telemetry package in your handler code.

  import com.snowflake.telemetry.Telemetry
  

  Copy

Adding trace events¶

You can add trace events by calling the Telemetry.addEvent method, passing a name for the event. You can also optionally associate attributes – key-value pairs – with an event.

The addEvent method has the following signatures:

public static void addEvent(String name)
public static void addEvent(String name, Attributes attributes)

Code in the following example adds an event called testEvent, associating with the event two attributes: key and result.

// Adding an event without attributes.
Telemetry.addEvent("testEvent")

// Adding an event with attributes.
Attributes eventAttributes = Attributes.of(
  AttributeKey.stringKey("key"), "run",
  AttributeKey.longKey("result"), Long.valueOf(123))
Telemetry.addEvent("testEventWithAttributes", eventAttributes)

Adding these events results in two rows in the event table, each with a different value in the RECORD column:

{
  "name": "testEvent"
}

{
  "name": "testEventWithAttributes"
}

The testEventWithAttributes event row includes the following attributes in the row’s RECORD_ATTRIBUTES column:

{
  "key": "run",
  "result": 123
}

Adding span attributes¶

You can set attributes – key-value pairs – associated with spans by calling the Telemetry.setSpanAttribute method.

The setSpanAttribute method has the following signatures:

public static void setSpanAttribute(String key, boolean value)
public static void setSpanAttribute(String key, long value)
public static void setSpanAttribute(String key, double value)
public static void setSpanAttribute(String key, String value)

For details on spans, see How Snowflake represents trace events.

Code in the following example creates four attributes and sets their values:

// Setting span attributes.
Telemetry.setSpanAttribute("example.boolean", true)
Telemetry.setSpanAttribute("example.long", 2L)
Telemetry.setSpanAttribute("example.double", 2.5)
Telemetry.setSpanAttribute("example.string", "testAttribute")

Setting these attributes results in the following in the event table’s RECORD_ATTRIBUTES column:

{
  "example.boolean": true,
  "example.long": 2,
  "example.double": 2.5,
  "example.string": "testAttribute"
}

Adding custom spans¶

You can add custom spans that are separate from the default span created by Snowflake. For details on custom spans, see Adding custom spans to a trace.

Code in the following example uses the OpenTelemetry API (https://javadoc.io/doc/io.opentelemetry/opentelemetry-api/latest/index.html) and OpenTelemetry context propagation API (https://www.javadoc.io/doc/io.opentelemetry/opentelemetry-context-prop/latest/index.html) to create a new my.span span. It then adds an event to the new span. Finally, the code ends the span to have the span’s event data captured in the event table. If the code doesn’t call the Span.end method, data is not captured in the event table.

CREATE OR REPLACE FUNCTION testScalaUserSpans(x varchar) RETURNS VARCHAR
LANGUAGE SCALA
RUNTIME_VERSION = 2.12
PACKAGES = ('com.snowflake:telemetry:latest')
HANDLER = 'TestScalaClass.run'
AS
$$
class TestScalaClass {
  import com.snowflake.telemetry.Telemetry
  import io.opentelemetry.api.GlobalOpenTelemetry
  import io.opentelemetry.api.trace.Tracer
  import io.opentelemetry.api.trace.Span
  import io.opentelemetry.context.Scope

  def run(x: String): String = {
    val tracer: Tracer = GlobalOpenTelemetry.getTracerProvider().get("my.tracer")
    val span: Span = tracer.spanBuilder("my.span").startSpan()
    span.addEvent("test event from scala")
    span.end()
    return x
  }
}
$$;

Examples¶

CREATE OR REPLACE PROCEDURE do_tracing()
RETURNS STRING
LANGUAGE SCALA
RUNTIME_VERSION = '2.12'
PACKAGES=('com.snowflake:snowpark:latest', 'com.snowflake:telemetry:latest')
HANDLER = 'ProcedureHandler.run'
AS
$$
import com.snowflake.snowpark_java.Session
import com.snowflake.telemetry.Telemetry
import io.opentelemetry.api.common.AttributeKey
import io.opentelemetry.api.common.Attributes

class ProcedureHandler {

  def run(session: Session): String = {

    // Set span attribute.
    Telemetry.setSpanAttribute("example.proc.do_tracing", "begin")

    // Add an event without attributes.
    Telemetry.addEvent("run_method_start")

    // Add an event with attributes.
    val eventAttributes: Attributes = Attributes.of(
      AttributeKey.stringKey("example.method.name"), "run")
    Telemetry.addEvent("event_with_attributes", eventAttributes)

    // Set span attribute.
    Telemetry.setSpanAttribute("example.proc.do_tracing", "complete")

    return "SUCCESS"
  }
}
$$;