Working with External gRPC Services

Oftentimes, you will want to include in your gRPC server, methods that are not automatically generated from your Ent schema. To achieve this result, define the methods in an additional service in an additional .proto file in your entpb directory.

info

Find the changes described in this section in this pull request.

For example, suppose you want to add a method named TopUser which will return the user with the highest ID number. To do this, create a new .proto file in your entpb directory, and define a new service:

ent/proto/entpb/ext.proto

syntax = "proto3";

package entpb;

option go_package = "github.com/rotemtam/ent-grpc-example/ent/proto/entpb";

import "entpb/entpb.proto";

import "google/protobuf/empty.proto";


service ExtService {
  rpc TopUser ( google.protobuf.Empty ) returns ( User );
}

Next, update entpb/generate.go to include the new file in the protoc command input:

ent/proto/entpb/generate.go

- //go:generate protoc -I=.. --go_out=.. --go-grpc_out=.. --go_opt=paths=source_relative --go-grpc_opt=paths=source_relative --entgrpc_out=.. --entgrpc_opt=paths=source_relative,schema_path=../../schema entpb/entpb.proto 
+ //go:generate protoc -I=.. --go_out=.. --go-grpc_out=.. --go_opt=paths=source_relative --go-grpc_opt=paths=source_relative --entgrpc_out=.. --entgrpc_opt=paths=source_relative,schema_path=../../schema entpb/entpb.proto entpb/ext.proto

Next, re-run code generation:

go generate ./...

Observe some new files were generated in the ent/proto/entpb directory:

tree
.
|-- entpb.pb.go
|-- entpb.proto
|-- entpb_grpc.pb.go
|-- entpb_user_service.go
|-- ext.pb.go
|-- ext.proto
|-- ext_grpc.pb.go
`-- generate.go

0 directories, 9 files

Now, you can implement the TopUser method in ent/proto/entpb/ext.go:

ent/proto/entpb/ext.go

package entpb

import (
    "context"

    "github.com/rotemtam/ent-grpc-example/ent"
    "github.com/rotemtam/ent-grpc-example/ent/user"
    "google.golang.org/protobuf/types/known/emptypb"
)

// ExtService implements ExtServiceServer.
type ExtService struct {
    client *ent.Client
    UnimplementedExtServiceServer
}

// TopUser returns the user with the highest ID.
func (s *ExtService) TopUser(ctx context.Context, _ *emptypb.Empty) (*User, error) {
    id := s.client.User.Query().Aggregate(ent.Max(user.FieldID)).IntX(ctx)
    user := s.client.User.GetX(ctx, id)
    return toProtoUser(user)
}

// NewExtService returns a new ExtService.
func NewExtService(client *ent.Client) *ExtService {
    return &ExtService{
        client: client,
    }
}

Adding the New Service to the gRPC Server

Finally, update cmd/server.go to include the new service:

cmd/server.go

package main

import (
    "context"
    "log"
    "net"

    _ "github.com/mattn/go-sqlite3"
    "github.com/rotemtam/ent-grpc-example/ent"
    "github.com/rotemtam/ent-grpc-example/ent/proto/entpb"
    "google.golang.org/grpc"
)

func main() {
    // Initialize an ent client.
    client, err := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
    if err != nil {
        log.Fatalf("failed opening connection to sqlite: %v", err)
    }
    defer client.Close()

    // Run the migration tool (creating tables, etc).
    if err := client.Schema.Create(context.Background()); err != nil {
        log.Fatalf("failed creating schema resources: %v", err)
    }

    // Initialize the generated User service.
    svc := entpb.NewUserService(client)

    // Create a new gRPC server (you can wire multiple services to a single server).
    server := grpc.NewServer()

    // Register the User service with the server.
    entpb.RegisterUserServiceServer(server, svc)

    // Register the external ExtService service with the server.
    entpb.RegisterExtServiceServer(server, entpb.NewExtService(client))

    // Open port 5000 for listening to traffic.
    lis, err := net.Listen("tcp", ":5000")
    if err != nil {
        log.Fatalf("failed listening: %s", err)
    }

    // Listen for traffic indefinitely.
    if err := server.Serve(lis); err != nil {
        log.Fatalf("server ended: %s", err)
    }
}