Add get_stats

Author: mickel8

lib/ex_webrtc/app.ex

@@ -0,0 +1,10 @@
+defmodule ExWebRTC.App do
+  use Application
+
+  def start(_type, _args) do
+    IO.inspect(:loading_ex_webrtc_app)
+    children = [{Registry, keys: :unique, name: ExWebRTC.Registry}]
+    Supervisor.start_link(children, strategy: :one_for_one)
+  end
+end
+

lib/ex_webrtc/ice_transport.ex

@@ -14,6 +14,7 @@ defmodule ExWebRTC.ICETransport do
   @callback restart(pid()) :: :ok
   @callback send_data(pid(), binary()) :: :ok
   @callback set_remote_credentials(pid(), ufrag :: binary(), pwd :: binary()) :: :ok
+  @callback get_stats(pid()) :: map()
   @callback stop(pid()) :: :ok
 end
 
@@ -43,5 +44,7 @@ defmodule ExWebRTC.DefaultICETransport do
   @impl true
   defdelegate set_remote_credentials(pid, ufrag, pwd), to: ICEAgent
   @impl true
+  defdelegate get_stats(pid), to: ICEAgent
+  @impl true
   defdelegate stop(pid), to: ICEAgent
 end

lib/ex_webrtc/peer_connection.ex

@@ -59,6 +59,11 @@ defmodule ExWebRTC.PeerConnection do
   @type connection_state() :: :closed | :failed | :disconnected | :new | :connecting | :connected
 
   #### API ####
+  @spec get_all_peer_connections() :: [pid()]
+  def get_all_peer_connections() do
+    Registry.select(ExWebRTC.Registry, [{{:_, :"$1", :_}, [], [:"$1"]}])
+  end
+
   @spec start_link(Configuration.options()) :: GenServer.on_start()
   def start_link(options \\ []) do
     configuration = Configuration.from_options!(options)
@@ -157,6 +162,11 @@ defmodule ExWebRTC.PeerConnection do
     GenServer.call(peer_connection, {:remove_track, sender_id})
   end
 
+  @spec get_stats(peer_connection()) :: %{String.t() => term()}
+  def get_stats(peer_connection) do
+    GenServer.call(peer_connection, :get_stats)
+  end
+
   @spec send_rtp(peer_connection(), String.t(), ExRTP.Packet.t()) :: :ok
   def send_rtp(peer_connection, track_id, packet) do
     GenServer.cast(peer_connection, {:send_rtp, track_id, packet})
@@ -171,6 +181,7 @@ defmodule ExWebRTC.PeerConnection do
 
   @impl true
   def init({owner, config}) do
+    {:ok, _} = Registry.register(ExWebRTC.Registry, self(), self())
     ice_config = [stun_servers: config.ice_servers, ip_filter: config.ice_ip_filter, on_data: nil]
     {:ok, ice_pid} = DefaultICETransport.start_link(:controlled, ice_config)
     {:ok, dtls_transport} = DTLSTransport.start_link(DefaultICETransport, ice_pid)
@@ -182,6 +193,7 @@ defmodule ExWebRTC.PeerConnection do
 
     state = %{
       owner: owner,
+      stats_id: Utils.generate_id(),
       config: config,
       current_local_desc: nil,
       pending_local_desc: nil,
@@ -558,6 +570,23 @@ defmodule ExWebRTC.PeerConnection do
     end
   end
 
+  @impl true
+  def handle_call(:get_stats, _from, state) do
+    timestamp = System.os_time(:millisecond)
+
+    stats = %{
+      state.stats_id => %{
+        id: state.stats_id,
+        type: :peer_connection,
+        timestamp: timestamp,
+        datachannels_opened: 0,
+        datachannels_closed: 0
+      }
+    }
+
+    {:reply, stats, state}
+  end
+
   @impl true
   def handle_cast({:send_rtp, track_id, packet}, state) do
     # TODO: iterating over transceivers is not optimal

mix.exs

@@ -33,6 +33,7 @@ defmodule ExWebRTC.MixProject do
 
   def application do
     [
+      mod: {ExWebRTC.App, []},
       extra_applications: [:logger]
     ]
   end