diff --git a/service.go b/service.go index 4723dd6d..4d814f9d 100644 --- a/service.go +++ b/service.go @@ -164,14 +164,6 @@ func (s *service) Start() error { } } - /* - if s.opts.Transport != nil { - if err := s.opts.Transport.Connect(s.opts.Context); err != nil { - return err - } - } - */ - if s.opts.Store != nil { if err := s.opts.Store.Connect(s.opts.Context); err != nil { return err @@ -213,6 +205,24 @@ func (s *service) Stop() error { } } + if s.opts.Registry != nil { + if err := s.opts.Registry.Disconnect(s.opts.Context); err != nil { + return err + } + } + + if s.opts.Broker != nil { + if err := s.opts.Broker.Disconnect(s.opts.Context); err != nil { + return err + } + } + + if s.opts.Store != nil { + if err := s.opts.Store.Disconnect(s.opts.Context); err != nil { + return err + } + } + return nil }