59 lines
1.2 KiB
Go
59 lines
1.2 KiB
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()
|
|
}
|