add session api

metagogs · Oct 13, 2022 · d011142 · d011142
1 parent c0b3342
commit d011142
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/session/session.go b/session/session.go
@@ -78,6 +78,10 @@ func (sess *Session) SetOnCloseCallbacks(callbacks []func(id int64)) {
 	sess.OnCloseCallbacks = callbacks
 }
 
+func (sess *Session) SetOnCloseCallback(c func(id int64)) {
+	sess.OnCloseCallbacks = []func(id int64){c}
+}
+
 func (sess *Session) GetLatency() int64 {
 	return sess.agent.GetLatency()
 }