mirrors_public/schollz_croc
0
0
Fork 0
mirror of https://github.com/schollz/croc.git synced 2025-10-11 13:21:00 +02:00
Code Activity

better cleaning of empty rooms

Browse source
This commit is contained in:
Zack Scholl 2019-05-02 20:57:55 -07:00
parent 1802ebcd00
commit 7d91f8200c
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/tcp/tcp.go
View file

@ -55,13 +55,18 @@ func (s *server) start() (err error) {
go func() { go func() {
for { for {
time.Sleep(10 * time.Minute) time.Sleep(10 * time.Minute)
roomsToDelete := []string{}
s.rooms.Lock() s.rooms.Lock()
for room := range s.rooms.rooms { for room := range s.rooms.rooms {
if time.Since(s.rooms.rooms[room].opened) > 3*time.Hour { if time.Since(s.rooms.rooms[room].opened) > 3*time.Hour {
delete(s.rooms.rooms, room) roomsToDelete = append(roomsToDelete,room)
} }
} }
s.rooms.Unlock() s.rooms.Unlock()
for _, room := range roomsToDelete {
s.deleteRoom(room)
}
} }
}() }()

4
src/tcp/tcp_test.go
View file

@ -1,7 +1,6 @@
package tcp package tcp
import ( import (
"strings"
"testing" "testing"
"time" "time"
@ -11,9 +10,8 @@ import (
func TestTCP(t *testing.T) { func TestTCP(t *testing.T) {
go Run("debug", "8081", "8082") go Run("debug", "8081", "8082")
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
c1, banner, ipaddr, err := ConnectToTCPServer("localhost:8081", "testRoom") c1, banner, _, err := ConnectToTCPServer("localhost:8081", "testRoom")
assert.Equal(t, banner, "8082") assert.Equal(t, banner, "8082")
assert.True(t, strings.HasPrefix(ipaddr, "127.0.0.1"))
assert.Nil(t, err) assert.Nil(t, err)
c2, _, _, err := ConnectToTCPServer("localhost:8081", "testRoom") c2, _, _, err := ConnectToTCPServer("localhost:8081", "testRoom")
assert.Nil(t, err) assert.Nil(t, err)