feat(signal): Improved invocation of asynchronous features (#81)

Signed-off-by: Flc゛ <[email protected]>
go-fries · Jan 30, 2024 · ad0f9e0 · ad0f9e0
1 parent 3168cbf
commit ad0f9e0
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/signal/README.md b/signal/README.md
@@ -44,7 +44,9 @@ func (h *exampleHandler) Handle(sig os.Signal) {
 	println("exampleHandler signal:", sig)
 }
 
-type example2Handler struct{}
+type example2Handler struct{
+	signal.AsyncFeature // async feature
+}
 
 func (h *example2Handler) Listen() []os.Signal {
 	return []os.Signal{syscall.SIGUSR1}

diff --git a/signal/handler.go b/signal/handler.go
@@ -9,6 +9,12 @@ type Handler interface {
 	Handle(os.Signal)
 }
 
-type AsyncFeature interface {
+type asyncFeature interface {
 	Async() bool
 }
+
+type AsyncFeature struct{}
+
+func (*AsyncFeature) Async() bool {
+	return true
+}
diff --git a/signal/server.go b/signal/server.go
@@ -79,8 +79,8 @@ func (s *Server) Start(ctx context.Context) error {
 		case sig := <-ch:
 			if hs, ok := handlers[sig]; ok {
 				for _, h := range hs {
-					// if Support AsyncFeature
-					if async, ok := h.(AsyncFeature); ok && async.Async() {
+					// if Support asyncFeature
+					if async, ok := h.(asyncFeature); ok && async.Async() {
 						go s.handle(sig, h)
 					} else {
 						s.handle(sig, h)