[pubsub] add close on shutdown

Author: capcom6

internal/sms-gateway/pubsub/module.go

@@ -1,6 +1,9 @@
 package pubsub
 
 import (
+	"context"
+
+	"github.com/android-sms-gateway/server/pkg/pubsub"
 	"go.uber.org/fx"
 	"go.uber.org/zap"
 )
@@ -12,5 +15,12 @@ func Module() fx.Option {
 			return log.Named("pubsub")
 		}),
 		fx.Provide(New),
+		fx.Invoke(func(pubsub pubsub.PubSub, lc fx.Lifecycle) {
+			lc.Append(fx.Hook{
+				OnStop: func(_ context.Context) error {
+					return pubsub.Close()
+				},
+			})
+		}),
 	)
 }