-
-
Notifications
You must be signed in to change notification settings - Fork 114
/
inactivitymonitor.go
72 lines (58 loc) · 1.31 KB
/
inactivitymonitor.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package inactivity
import (
"context"
"sync/atomic"
"time"
)
type Monitor = interface {
CheckInactivity(now time.Time, cc ClientConn)
Notify()
}
type OnInactiveFunc = func(cc ClientConn)
type ClientConn = interface {
Context() context.Context
Close() error
}
type inactivityMonitor struct {
lastActivity atomic.Value
duration time.Duration
onInactive OnInactiveFunc
}
func (m *inactivityMonitor) Notify() {
m.lastActivity.Store(time.Now())
}
func (m *inactivityMonitor) LastActivity() time.Time {
if t, ok := m.lastActivity.Load().(time.Time); ok {
return t
}
return time.Time{}
}
func CloseClientConn(cc ClientConn) {
// call cc.Close() directly to check and handle error if necessary
_ = cc.Close()
}
func NewInactivityMonitor(duration time.Duration, onInactive OnInactiveFunc) Monitor {
m := &inactivityMonitor{
duration: duration,
onInactive: onInactive,
}
m.Notify()
return m
}
func (m *inactivityMonitor) CheckInactivity(now time.Time, cc ClientConn) {
if m.onInactive == nil || m.duration == time.Duration(0) {
return
}
if now.After(m.LastActivity().Add(m.duration)) {
m.onInactive(cc)
}
}
type nilMonitor struct {
}
func (m *nilMonitor) CheckInactivity(now time.Time, cc ClientConn) {
}
func (m *nilMonitor) Notify() {
}
func NewNilMonitor() Monitor {
return &nilMonitor{}
}