adding a OnRebind call back to notify users when rebind was successful

linxGnu · Nov 11, 2023 · 917fb50 · 917fb50
1 parent 7bfd2f8
commit 917fb50
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/pkg.go b/pkg.go
@@ -72,5 +72,8 @@ type Settings struct {
 	// manual response/handling is needed
 	OnAllPDU func(pdu pdu.PDU) pdu.PDU
 
+	// OnRebind notifies `rebind` event due to State.
+	OnRebind RebindCallback
+
 	response func(pdu.PDU)
 }
diff --git a/session.go b/session.go
@@ -119,6 +119,9 @@ func (s *Session) rebind() {
 
 				// reset rebinding state
 				atomic.StoreInt32(&s.rebinding, 0)
+				if s.settings.OnRebind != nil {
+					s.settings.OnRebind()
+				}
 
 				return
 			}

diff --git a/types.go b/types.go
@@ -13,3 +13,6 @@ type ErrorCallback func(error)
 
 // ClosedCallback notifies closed event due to State.
 type ClosedCallback func(State)
+
+// RebindCallback notifies rebind event due to State.
+type RebindCallback func()