Update the span name for websocket requests

Author: joshk

lib/nerves_hub/application.ex

@@ -54,6 +54,8 @@ defmodule NervesHub.Application do
       :httpc.set_option(:ipfamily, :inet6fb4)
     end
 
+    :ok = NervesHub.Telemetry.Customizations.setup()
+
     :ok = OpentelemetryBandit.setup()
     :ok = OpentelemetryPhoenix.setup(adapter: :bandit)
     :ok = OpentelemetryOban.setup(trace: [:jobs])

lib/nerves_hub/telemetry/customizations.ex

@@ -0,0 +1,23 @@
+defmodule NervesHub.Telemetry.Customizations do
+  alias OpenTelemetry.Tracer
+  require OpenTelemetry.Tracer
+
+  def setup() do
+    :telemetry.attach_many(
+      {__MODULE__, :bandit_customizations},
+      [
+        [:bandit, :request, :stop]
+      ],
+      &__MODULE__.handle_request/4,
+      nil
+    )
+  end
+
+  def handle_request([:bandit, :request, :stop], _measurements, %{conn: conn}, _config) do
+    if conn.status == 101 do
+      Tracer.update_name("WEBSOCKET #{conn.request_path}")
+    end
+
+    :ok
+  end
+end