feat 添加SetErrorHandle方法

2024-08-31 16:49:48 +08:00 · 2024-08-31 16:49:48 +08:00 · 35b6608cc1
commit 35b6608cc1
parent adcb9ed31b
1 changed files with 5 additions and 0 deletions
--- a/pkg/subscribe/server.go
+++ b/pkg/subscribe/server.go
@ -57,6 +57,11 @@ func (s *Server) Unsubscribe(channel string) error {
 	return s.pubSub.Unsubscribe(s.ctx, channel)
 }

+func (s *Server) SetErrorHandle(f func(err error)) *Server {
+	s.errHandle = f
+	return s
+}
+
 func (s *Server) Run() {
 	ch := s.pubSub.Channel()
 	defer s.pubSub.Close()