feat: 增加OnSessionCreated回调

chenqinghe · Oct 18, 2022 · 8fc1ade · 8fc1ade
1 parent 7bb4a32
commit 8fc1ade
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -21,6 +21,7 @@ type NewClientOptions struct {
 	KeepalivePeriod        time.Duration
 	HeartbeatPacketFactory func() Packet
 	OnConnected            func(c Conn) error
+	OnSessionCreated       func(s *Session)
 	OnClosed               func(session *Session)
 	NeedReconnect          bool
 	ReconnectPolicy        ReconnectPolicy
@@ -40,6 +41,7 @@ func NewClient(codec Codec, handler Handler, opts *NewClientOptions) *Client {
 		closed: make(chan struct{}),
 	}
 	cli.mgr = NewManager(handler, &NewManagerOptions{
+		OnSessionCreated: opts.OnSessionCreated,
 		AfterSessionClosed: func(s *Session) {
 			cli.reconnect(s)
 		},