[Schemas] Connector SDK: Create and get a schema from Conduit

conduit/schema/client/client.go

@@ -0,0 +1,45 @@
+// Copyright © 2024 Meroxa, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package client
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/conduitio/conduit-connector-protocol/conduit/schema"
+	v1 "github.com/conduitio/conduit-connector-protocol/conduit/schema/v1"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+type schemaServiceKey struct{}
+
+func WithSchemaService(ctx context.Context, s schema.Service) context.Context {
+	return context.WithValue(ctx, schemaServiceKey{}, s)
+}
+
+func New(ctx context.Context) (schema.Service, error) {
+	service := ctx.Value(schemaServiceKey{})
+	if service != nil {
+		return service.(schema.Service), nil
+	}
+
+	conn, err := grpc.NewClient("localhost:8184", grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		return nil, fmt.Errorf("failed creating gRPC client: %w", err)
+	}
+
+	return v1.NewClient(conn)
+}

conduit/schema/client/client_test.go

@@ -0,0 +1,45 @@
+// Copyright © 2024 Meroxa, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package client
+
+import (
+	"context"
+	"testing"
+
+	"github.com/conduitio/conduit-connector-protocol/conduit/schema/mock"
+	v1 "github.com/conduitio/conduit-connector-protocol/conduit/schema/v1"
+	"github.com/matryer/is"
+	"go.uber.org/mock/gomock"
+)
+
+func TestClient_NewFromContext(t *testing.T) {
+	is := is.New(t)
+	service := mock.NewService(gomock.NewController(t))
+	ctx := WithSchemaService(context.Background(), service)
+
+	s, err := New(ctx)
+	is.NoErr(err)
+	is.Equal(s, service)
+}
+
+func TestClient_NewGRPC(t *testing.T) {
+	is := is.New(t)
+	ctx := context.Background()
+
+	s, err := New(ctx)
+	is.NoErr(err)
+	_, ok := s.(*v1.Client)
+	is.True(ok)
+}

conduit/schema/schema.go

@@ -12,6 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+//go:generate mockgen -destination=mock/schema.go -package=mock -mock_names=Service=Service . Service
+
 package schema
 
 import (
@@ -41,7 +43,8 @@ type CreateResponse struct {
 }
 
 type GetRequest struct {
-	ID string
+	Name    string
+	Version int
 }
 type GetResponse struct {
 	schema.Instance

conduit/schema/v1/fromproto/schema.go

@@ -49,3 +49,15 @@ func CreateResponse(resp *conduitv1.CreateSchemaResponse) schema.CreateResponse
 		},
 	}
 }
+
+func GetResponse(resp *conduitv1.GetSchemaResponse) schema.GetResponse {
+	return schema.GetResponse{
+		Instance: cschema.Instance{
+			ID:      resp.Schema.Id,
+			Name:    resp.Schema.Name,
+			Version: resp.Schema.Version,
+			Type:    cschema.Type(resp.Schema.Type),
+			Bytes:   resp.Schema.Bytes,
+		},
+	}
+}

conduit/schema/v1/grpc_client.go

@@ -0,0 +1,56 @@
+// Copyright © 2024 Meroxa, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package v1
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/conduitio/conduit-connector-protocol/conduit/schema"
+	"github.com/conduitio/conduit-connector-protocol/conduit/schema/v1/fromproto"
+	"github.com/conduitio/conduit-connector-protocol/conduit/schema/v1/toproto"
+	conduitv1 "github.com/conduitio/conduit-connector-protocol/proto/conduit/v1"
+	"google.golang.org/grpc"
+)
+
+var _ schema.Service = (*Client)(nil)
+
+type Client struct {
+	grpcClient conduitv1.SchemaServiceClient
+}
+
+func NewClient(conn *grpc.ClientConn) (*Client, error) {
+	return &Client{
+		grpcClient: conduitv1.NewSchemaServiceClient(conn),
+	}, nil
+}
+
+func (c *Client) Create(ctx context.Context, request schema.CreateRequest) (schema.CreateResponse, error) {
+	resp, err := c.grpcClient.Create(ctx, toproto.CreateSchemaRequest(request))
+	if err != nil {
+		return schema.CreateResponse{}, fmt.Errorf("failed creating schema: %w", err)
+	}
+
+	return fromproto.CreateResponse(resp), nil
+}
+
+func (c *Client) Get(ctx context.Context, request schema.GetRequest) (schema.GetResponse, error) {
+	resp, err := c.grpcClient.Get(ctx, toproto.GetSchemaRequest(request))
+	if err != nil {
+		return schema.GetResponse{}, fmt.Errorf("failed creating schema: %w", err)
+	}
+
+	return fromproto.GetResponse(resp), nil
+}

conduit/schema/v1/toproto/schema.go

@@ -20,9 +20,10 @@
 	conduitv1 "github.com/conduitio/conduit-connector-protocol/proto/conduit/v1"
 )
 
-func GetSchemaRequest(id string) *conduitv1.GetSchemaRequest {
+func GetSchemaRequest(request v1.GetRequest) *conduitv1.GetSchemaRequest {
 	return &conduitv1.GetSchemaRequest{
-		Id: id,
+		Name:    request.Name,
+		Version: int32(request.Version),
 	}
 }
 
@@ -31,7 +32,7 @@
 		Schema: &conduitv1.Schema{
 			Id:      inst.ID,
 			Name:    inst.Name,
-			Version: inst.Version,
+			Version: int32(inst.Version),
 			Type:    conduitv1.Schema_Type(inst.Type),
 			Bytes:   inst.Bytes,
 		},
@@ -51,7 +52,7 @@
 		Schema: &conduitv1.Schema{
 			Id:      inst.ID,
 			Name:    inst.Name,
-			Version: inst.Version,
+			Version: int32(inst.Version),
 			Type:    conduitv1.Schema_Type(inst.Type),
 			Bytes:   inst.Bytes,
 		},

go.mod

@@ -7,6 +7,7 @@ require (
 	github.com/conduitio/conduit-commons v0.2.1-0.20240529171657-31b922714201
 	github.com/golangci/golangci-lint v1.59.0
 	github.com/hashicorp/go-plugin v1.6.1
+	github.com/matryer/is v1.4.1
 	go.uber.org/mock v0.4.0
 	google.golang.org/grpc v1.64.0
 	google.golang.org/protobuf v1.34.1