balki/pgweb
1
0
Fork 0
Code Pull Requests Activity

Verify client is closed

Browse Source
This commit is contained in:
Dan Sosedoff
2022-12-02 13:59:22 -06:00
parent bc54c95140
commit 73dfcc46c3
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/api/session_manager_test.go
View File

@@ -75,5 +75,6 @@ func TestSessionManager(t *testing.T) {
command.Opts.ConnectionIdleTimeout = 1 command.Opts.ConnectionIdleTimeout = 1
assert.Equal(t, 1, manager.Cleanup()) assert.Equal(t, 1, manager.Cleanup())
assert.Equal(t, 0, manager.Len()) assert.Equal(t, 0, manager.Len())
assert.True(t, conn.IsClosed())
}) })
} }

4
pkg/client/client.go
View File

@@ -442,6 +442,10 @@ func (client *Client) Close() error {
return nil return nil
} }
func (c *Client) IsClosed() bool {
return c.closed
}
func (client *Client) IsIdle() bool { func (client *Client) IsIdle() bool {
mins := int(time.Since(client.lastQueryTime).Minutes()) mins := int(time.Since(client.lastQueryTime).Minutes())