Thanks to visit codestin.com
Credit goes to github.com

Skip to content
This repository was archived by the owner on Aug 30, 2024. It is now read-only.

feat: Add ActiveConnections function to track usage #390

Merged
merged 4 commits into from
Jul 20, 2021
Merged

feat: Add ActiveConnections function to track usage #390

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
6c8bfdb
feat: Add ActiveConnections function tto track usage
kylecarbs Jul 20, 2021
eff3e3e
Fix linting errors
kylecarbs Jul 20, 2021
0181650
Merge branch 'master' into activeconns
kylecarbs Jul 20, 2021
bbeffa1
Fix syntax err
kylecarbs Jul 20, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions wsnet/dial.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -232,6 +232,17 @@ func (d *Dialer) Closed() <-chan struct{} {
return d.closedChan
}

// ActiveConnections returns the amount of active connections.
// DialContext opens a connection, and close will end it.
func (d *Dialer) ActiveConnections() int {
stats, ok := d.rtc.GetStats().GetConnectionStats(d.rtc)
if !ok {
return -1
}
// Subtract 1 for the control channel.
return int(stats.DataChannelsRequested-stats.DataChannelsClosed) - 1
}

// Close closes the RTC connection.
// All data channels dialed will be closed.
func (d *Dialer) Close() error {
Expand Down
32 changes: 32 additions & 0 deletions wsnet/dial_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -243,6 +243,38 @@ func TestDial(t *testing.T) {
t.Error("didn't close in time")
}
})

t.Run("Active Connections", func(t *testing.T) {
t.Parallel()

listener, err := net.Listen("tcp", "0.0.0.0:0")
if err != nil {
t.Error(err)
return
}
go func() {
_, _ = listener.Accept()
}()
connectAddr, listenAddr := createDumbBroker(t)
_, err = Listen(context.Background(), slogtest.Make(t, nil), listenAddr, "")
if err != nil {
t.Error(err)
return
}
dialer, err := DialWebsocket(context.Background(), connectAddr, nil)
if err != nil {
t.Error(err)
}
conn, _ := dialer.DialContext(context.Background(), listener.Addr().Network(), listener.Addr().String())
assert.Equal(t, 1, dialer.ActiveConnections())
_ = conn.Close()
assert.Equal(t, 0, dialer.ActiveConnections())
_, _ = dialer.DialContext(context.Background(), listener.Addr().Network(), listener.Addr().String())
conn, _ = dialer.DialContext(context.Background(), listener.Addr().Network(), listener.Addr().String())
assert.Equal(t, 2, dialer.ActiveConnections())
_ = conn.Close()
assert.Equal(t, 1, dialer.ActiveConnections())
})
}

func BenchmarkThroughput(b *testing.B) {
Expand Down