You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
815 B
Go
55 lines
815 B
Go
package lifetime
|
|
|
|
import (
|
|
"git.noahlan.cn/noahlan/nnet/connection"
|
|
)
|
|
|
|
type (
|
|
Handler func(conn *connection.Connection)
|
|
|
|
Lifetime interface {
|
|
OnClosed(h Handler)
|
|
OnOpen(h Handler)
|
|
}
|
|
|
|
Mgr struct {
|
|
onOpen []Handler
|
|
onClosed []Handler
|
|
}
|
|
)
|
|
|
|
func NewLifetime() *Mgr {
|
|
return &Mgr{
|
|
onOpen: make([]Handler, 0),
|
|
onClosed: make([]Handler, 0),
|
|
}
|
|
}
|
|
|
|
func (lt *Mgr) OnClosed(h Handler) {
|
|
lt.onClosed = append(lt.onClosed, h)
|
|
}
|
|
|
|
func (lt *Mgr) OnOpen(h Handler) {
|
|
lt.onOpen = append(lt.onOpen, h)
|
|
}
|
|
|
|
func (lt *Mgr) Open(conn *connection.Connection) {
|
|
if len(lt.onOpen) <= 0 {
|
|
return
|
|
}
|
|
|
|
for _, handler := range lt.onOpen {
|
|
handler(conn)
|
|
}
|
|
}
|
|
|
|
func (lt *Mgr) Close(conn *connection.Connection) {
|
|
if len(lt.onClosed) <= 0 {
|
|
return
|
|
}
|
|
|
|
for _, handler := range lt.onClosed {
|
|
handler(conn)
|
|
}
|
|
}
|