Use context everywhere

internal/domains/deconnector/deconnector.go

@@ -1,6 +1,7 @@
 package deconnector
 
 import (
+	"context"
 	"fmt"
 
 	"source.hodakov.me/hdkv/deconnect/internal/application"
@@ -24,7 +25,7 @@ func New(app *application.App) *Deconnector {
 	}
 }
 
-func (d *Deconnector) ConnectDependencies() error {
+func (d *Deconnector) ConnectDependencies(_ context.Context) error {
 	dialer, ok := d.app.RetrieveDomain(domains.DomainNameDialer).(domains.Dialer)
 	if !ok {
 		return fmt.Errorf(
@@ -38,6 +39,6 @@ func (d *Deconnector) ConnectDependencies() error {
 	return nil
 }
 
-func (d *Deconnector) Start() error {
+func (d *Deconnector) Start(_ context.Context) error {
 	return nil
 }

internal/domains/dialer/dialer.go

@@ -1,6 +1,7 @@
 package dialer
 
 import (
+	"context"
 	"fmt"
 	"net/url"
 
@@ -25,7 +26,7 @@ func New(app *application.App) *Dialer {
 	}
 }
 
-func (d *Dialer) ConnectDependencies() error {
+func (d *Dialer) ConnectDependencies(ctx context.Context) error {
 	dialURL, err := d.UpstreamURL()
 	if err != nil {
 		return fmt.Errorf(
@@ -38,6 +39,6 @@ func (d *Dialer) ConnectDependencies() error {
 	return nil
 }
 
-func (d *Dialer) Start() error {
+func (d *Dialer) Start(_ context.Context) error {
 	return nil
 }

internal/domains/domain.go

@@ -1,6 +1,8 @@
 package domains
 
+import "context"
+
 type Domain interface {
-	ConnectDependencies() error
-	Start() error
+	ConnectDependencies(ctx context.Context) error
+	Start(ctx context.Context) error
 }

internal/domains/listener/listen.go

@@ -1,11 +1,12 @@
 package listener
 
 import (
+	"context"
 	"fmt"
 	"net"
 )
 
-func (l *Listener) Listen() error {
+func (l *Listener) Listen(ctx context.Context) error {
 	ln, err := net.Listen(
 		"tcp",
 		l.app.Config().Deconnect.Host+":"+l.app.Config().Deconnect.Port,
@@ -19,7 +20,7 @@ func (l *Listener) Listen() error {
 		Info("Listening for incoming connections")
 
 	go func() {
-		<-l.app.Context().Done()
+		<-ctx.Done()
 
 		l.app.Logger().Info("Shutting down listener")
 
@@ -31,7 +32,7 @@ func (l *Listener) Listen() error {
 	for {
 		conn, err := ln.Accept()
 		if err != nil {
-			if l.app.Context().Err() != nil {
+			if ctx.Err() != nil {
 				return nil
 			}
 

internal/domains/listener/listener.go

@@ -1,6 +1,7 @@
 package listener
 
 import (
+	"context"
 	"fmt"
 
 	"source.hodakov.me/hdkv/deconnect/internal/application"
@@ -24,7 +25,7 @@ func New(app *application.App) *Listener {
 	}
 }
 
-func (l *Listener) ConnectDependencies() error {
+func (l *Listener) ConnectDependencies(ctx context.Context) error {
 	deconnector, ok := l.app.RetrieveDomain(domains.DomainNameDeconnector).(domains.Deconnector)
 	if !ok {
 		return fmt.Errorf(
@@ -38,15 +39,14 @@ func (l *Listener) ConnectDependencies() error {
 	return nil
 }
 
-func (l *Listener) Start() error {
-	wg := l.app.GetGlobalWaitGroup()
-	if wg == nil {
-		return fmt.Errorf("%w: %w (%w)", ErrListener, ErrStart, ErrFailedToGetWaitGroup)
-	}
-
-	wg.Go(func() {
-		l.Listen()
-	})
+func (l *Listener) Start(ctx context.Context) error {
+	go func() {
+		if err := l.Listen(ctx); err != nil {
+			l.app.Logger().
+				WithError(err).
+				Error("Listener stopped with error")
+		}
+	}()
 
 	return nil
 }