crazy-max_diun/internal/grpc/client.go

package grpc

import (
	"net"

	"github.com/crazy-max/diun/v4/internal/db"
	grpclogger "github.com/crazy-max/diun/v4/internal/grpc/logger"
	"github.com/crazy-max/diun/v4/internal/notif"
	"github.com/crazy-max/diun/v4/pb"
	"github.com/pkg/errors"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"google.golang.org/grpc"
)

// Client represents an active grpc object
type Client struct {
	server    *grpc.Server
	authority string
	db        *db.Client
	notif     *notif.Client
	pb.UnimplementedImageServiceServer
	pb.UnimplementedNotifServiceServer
}

// New creates a new grpc instance
func New(authority string, db *db.Client, notif *notif.Client) (*Client, error) {
	grpclogger.SetGrpcLogger(log.Level(zerolog.ErrorLevel))

	c := &Client{
		authority: authority,
		db:        db,
		notif:     notif,
	}

	c.server = grpc.NewServer()
	pb.RegisterImageServiceServer(c.server, c)
	pb.RegisterNotifServiceServer(c.server, c)

	return c, nil
}

// Start runs the grpc server
func (c *Client) Start() error {
	var err error

	lis, err := net.Listen("tcp", c.authority)
	if err != nil {
		return errors.Wrap(err, "cannot create gRPC listener")
	}

	return c.server.Serve(lis)
}

// Stop stops the grpc server
func (c *Client) Stop() {
	c.server.GracefulStop()
}