feat(fxgcppubsub): Added schemas support

ankorstore · Jun 5, 2024 · abb6ea5 · abb6ea5
1 parent 57aec98
commit abb6ea5
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/fxgcppubsub/module.go b/fxgcppubsub/module.go
@@ -25,7 +25,7 @@ var FxGcpPubSubModule = fx.Module(
 	fx.Provide(
 		NewFxGcpPubSubClient,
 		NewFxGcpPubSubSchemaClient,
-		schema.NewSchemaRegistry,
+		NewFxGcpPubSubSchemaRegistry,
 	),
 )
 
@@ -53,7 +53,7 @@ func createClient(p FxGcpPubSubClientParam) (*pubsub.Client, error) {
 	}
 
 	p.LifeCycle.Append(fx.Hook{
-		OnStop: func(ctx context.Context) error {
+		OnStop: func(context.Context) error {
 			return client.Close()
 		},
 	})
@@ -79,7 +79,7 @@ func createTestClient(p FxGcpPubSubClientParam) (*pubsub.Client, error) {
 	}
 
 	p.LifeCycle.Append(fx.Hook{
-		OnStop: func(ctx context.Context) error {
+		OnStop: func(context.Context) error {
 			err = client.Close()
 			if err != nil {
 				return err
@@ -104,3 +104,15 @@ func NewFxGcpPubSubSchemaClient(p FxGcpPubSubSchemaClientParam) (*pubsub.SchemaC
 	return pubsub.NewSchemaClient(p.Context, p.Config.GetString("modules.gcppubsub.project.id"))
 
 }
+
+// FxGcpPubSubSchemaRegistryParam allows injection of the required dependencies in [NewFxGcpPubSubSchemaRegistry].
+type FxGcpPubSubSchemaRegistryParam struct {
+	fx.In
+	Client *pubsub.SchemaClient
+}
+
+// NewFxGcpPubSubSchemaRegistry returns a [schema.SchemaRegistry].
+func NewFxGcpPubSubSchemaRegistry(p FxGcpPubSubSchemaRegistryParam) *schema.SchemaRegistry {
+	return schema.NewSchemaRegistry(p.Client)
+
+}