client.go

package bertyprotocol

import (
	"context"

	"berty.tech/berty/v2/go/internal/grpcutil"
	"google.golang.org/grpc"
)

type Client interface {
	ProtocolServiceClient

	Close() error
}

type client struct {
	ProtocolServiceClient

	l *grpcutil.PipeListener
}

func (c *client) Close() error {
	return c.l.Close()
}

func NewClient(svc Service, opts ...grpc.ServerOption) (Client, error) {
	return NewClientFromServer(grpc.NewServer(opts...), svc)
}

func NewClientFromServer(s *grpc.Server, svc Service, opts ...grpc.DialOption) (Client, error) {
	pl := grpcutil.NewPipeListener()

	opts = append(opts, grpc.WithInsecure())
	cc, err := pl.NewClientConn(opts...)
	if err != nil {
		return nil, err
	}

	RegisterProtocolServiceServer(s, svc)

	go func() {
		err := s.Serve(pl)
		if err != nil && err != context.Canceled {
			panic(err)
		}
	}()

	c := client{ProtocolServiceClient: NewProtocolServiceClient(cc), l: pl}
	return &c, nil
}