From acbb47a1005a2144dd390cd9e026d4b8f8373189 Mon Sep 17 00:00:00 2001
From: Joachim Bauch <bauch@struktur.de>
Date: Fri, 7 Aug 2020 11:39:52 +0200
Subject: [PATCH] Add callback on received RTT.

---
 src/signaling/client.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/signaling/client.go b/src/signaling/client.go
index 3523792..187ca2d 100644
--- a/src/signaling/client.go
+++ b/src/signaling/client.go
@@ -90,6 +90,7 @@ type Client struct {
 	OnLookupCountry   func(*Client) string
 	OnClosed          func(*Client)
 	OnMessageReceived func(*Client, []byte)
+	OnRTTReceived     func(*Client, time.Duration)
 }
 
 func NewClient(conn *websocket.Conn, remoteAddress string, agent string) (*Client, error) {
@@ -110,6 +111,7 @@ func NewClient(conn *websocket.Conn, remoteAddress string, agent string) (*Clien
 		OnLookupCountry:   func(client *Client) string { return unknownCountry },
 		OnClosed:          func(client *Client) {},
 		OnMessageReceived: func(client *Client, data []byte) {},
+		OnRTTReceived:     func(client *Client, rtt time.Duration) {},
 	}
 	return client, nil
 }
@@ -234,6 +236,7 @@ func (c *Client) ReadPump() {
 			} else {
 				log.Printf("Client from %s has RTT of %d ms (%s)", addr, rtt_ms, rtt)
 			}
+			c.OnRTTReceived(c, rtt)
 		}
 		return nil
 	})