service: add Register method to builder

lithdew · Jun 11, 2020 · 6396f5a · 6396f5a
1 parent 6a2b7f5
commit 6396f5a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/service.go b/service.go
@@ -17,6 +17,13 @@ type Service struct {
 	Services   map[string]Handler
 }
 
+func (s *Service) Register(service string, handler Handler) {
+	if s.Services == nil {
+		s.Services = make(map[string]Handler, 1)
+	}
+	s.Services[service] = handler
+}
+
 func (s Service) Start() error {
 	addr := s.Addr
 	if addr == "" {