task handler

internal/task/handler.go

@@ -0,0 +1,56 @@
+package task
+
+import (
+	log "github.com/sirupsen/logrus"
+	"google.golang.org/grpc"
+	"merch-api/internal/common"
+	taskTransportV1 "merch-api/pkg/taskTransport/v1"
+	"net"
+)
+
+const pkgLogHeader string = "Tasker |"
+
+type Handler struct {
+	addr string
+	srv  *grpc.Server
+	*service
+}
+
+type Deps struct {
+	Addr          string
+	MerchProvider common.MerchProvider
+}
+
+func New(deps Deps) *Handler {
+	srv := grpc.NewServer()
+
+	handler := &Handler{
+		addr:    deps.Addr,
+		service: newService(deps.MerchProvider),
+	}
+
+	taskTransportV1.RegisterTaskProcessorServer(srv, handler)
+
+	return handler
+}
+
+func (h *Handler) Serve() error {
+	listener, err := net.Listen("tcp", h.addr)
+	if err != nil {
+		log.WithError(err).Errorf("%v Listner failure", pkgLogHeader)
+		return err
+	}
+
+	if err = h.srv.Serve(listener); err != nil {
+		log.WithError(err).Errorf("%v Serve failure", pkgLogHeader)
+		return err
+	}
+
+	log.Infof("%v gRPC server started on: %v", pkgLogHeader, h.addr)
+	return nil
+}
+
+func (h *Handler) Shutdown() {
+	h.srv.GracefulStop()
+	log.Infof("%v gRPC server shutdown", pkgLogHeader)
+}

internal/task/service.go

@@ -0,0 +1,44 @@
+package task
+
+import (
+	log "github.com/sirupsen/logrus"
+	"google.golang.org/protobuf/types/known/emptypb"
+	"merch-api/internal/common"
+	tt "merch-api/pkg/taskTransport/v1"
+)
+
+type service struct {
+	tt.UnimplementedTaskProcessorServer
+	merchProvider common.MerchProvider
+}
+
+func newService(mp common.MerchProvider) *service {
+	return &service{
+		merchProvider: mp,
+	}
+}
+
+func (s *service) RequestTasks(_ *emptypb.Empty, stream tt.TaskProcessor_RequestTasksServer) error {
+	tasks, err := s.merchProvider.GetTasks(stream.Context())
+	if err != nil {
+		log.WithError(err).Errorf("%v Failed to get tasks", pkgLogHeader)
+		return err
+	}
+
+	for _, task := range tasks {
+		if err = stream.Send(&tt.Task{
+			MerchUuid: task.MerchUuid,
+			Origins:   task.Origins,
+		}); err != nil {
+			log.WithError(err).Errorf("%v Failed to send tasks", pkgLogHeader)
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (s *service) SendResults(stream tt.TaskProcessor_SendResultsServer) error {
+	//TODO
+	return nil
+}