balki/pgweb
1
0
Fork 0
Code Pull Requests Activity

Add idle timeout into session manager

Browse Source
This commit is contained in:
Dan Sosedoff
2022-12-02 14:20:20 -06:00
parent 0a133dc395
commit 16726e2461
4 changed files with 31 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/api/session_manager_test.go
View File

@@ -3,12 +3,12 @@ package api
import (
"sort"
"testing"
"time"
"github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert"
"github.com/sosedoff/pgweb/pkg/client"
"github.com/sosedoff/pgweb/pkg/command"
)
func TestSessionManager(t *testing.T) {
@@ -60,20 +60,16 @@ func TestSessionManager(t *testing.T) {
})
t.Run("clean up stale sessions", func(t *testing.T) {
defer func() {
command.Opts.ConnectionIdleTimeout = 0
}()
manager := NewSessionManager(logrus.New())
conn := &client.Client{}
manager.Add("foo", conn)
command.Opts.ConnectionIdleTimeout = 0
assert.Equal(t, 1, manager.Len())
assert.Equal(t, 0, manager.Cleanup())
assert.Equal(t, 1, manager.Len())
command.Opts.ConnectionIdleTimeout = 1
conn.Query("select 1")
manager.SetIdleTimeout(time.Minute)
assert.Equal(t, 1, manager.Cleanup())
assert.Equal(t, 0, manager.Len())
assert.True(t, conn.IsClosed())