ms-tester/internal/transport/problem.go

package transport

import (
	"context"
	"git.sch9.ru/new_gate/ms-tester/internal/lib"
	"git.sch9.ru/new_gate/ms-tester/internal/models"
	problemv1 "git.sch9.ru/new_gate/ms-tester/pkg/go/gen/proto/problem/v1"
	sessionv1 "git.sch9.ru/new_gate/ms-tester/pkg/go/gen/proto/session/v1"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
	"google.golang.org/protobuf/types/known/emptypb"
	"io"
)

func (s *TesterServer) CreateProblem(server problemv1.ProblemService_CreateProblemServer) error {
	ctx := server.Context()

	req, err := server.Recv() // receive token
	if err != nil {
		return err // FIXME
	}

	token := req.GetToken()
	userId, err := s.sessionClient.Read(ctx, &sessionv1.ReadSessionRequest{
		Token: token,
	})
	if err != nil {
		return err // FIXME
	}

	ctx = context.WithValue(ctx, "user_id", userId.GetUserId())

	req, err = server.Recv() // receive problem
	if err != nil {
		return err // FIXME
	}
	problem := req.GetProblem()
	if problem == nil {
		return status.Errorf(codes.Unknown, "") // FIXME
	}

	p := &models.Problem{
		Name:        lib.AsStringP(problem.Name),
		Description: lib.AsStringP(problem.Description),
		TimeLimit:   lib.AsInt32P(problem.TimeLimit),
		MemoryLimit: lib.AsInt32P(problem.MemoryLimit),
	}

	ch := readChunks(ctx, server)

	id, err := s.problemService.CreateProblem(ctx, p, ch)
	if err != nil {
		return status.Errorf(codes.Unknown, "") // FIXME
	}

	err = server.SendAndClose(&problemv1.CreateProblemResponse{
		Id: id,
	})
	if err != nil {
		return err // FIXME
	}

	return nil
}

func readChunks(ctx context.Context, server problemv1.ProblemService_CreateProblemServer) <-chan []byte {
	ch := make(chan []byte)

	go func() {
		defer close(ch)
		for {
			select {
			case <-ctx.Done():
				return // FIXME
			default:
				req, err := server.Recv()
				if err != nil {
					if err == io.EOF {
						return // FIXME
					}
					if status.Code(err) == codes.Canceled {
						return // FIXME
					}
					continue
				}

				test := req.GetTest()
				if test == nil {
					return // FIXME
				}

				ch <- test.Chunk
			}
		}
	}()
	return ch
}

func (s *TesterServer) ReadProblem(ctx context.Context, req *problemv1.ReadProblemRequest) (*problemv1.ReadProblemResponse, error) {
	problem, err := s.problemService.ReadProblemById(ctx, req.GetId())
	if err != nil {
		return nil, status.Errorf(codes.Unknown, err.Error()) // FIXME
	}
	return &problemv1.ReadProblemResponse{
		Problem: &problemv1.ReadProblemResponse_Problem{
			Id:          *problem.Id,
			Name:        *problem.Name,
			Description: *problem.Description,
			TimeLimit:   *problem.TimeLimit,
			MemoryLimit: *problem.MemoryLimit,
			CreatedAt:   AsTimestampP(problem.CreatedAt),
			UpdatedAt:   AsTimestampP(problem.UpdatedAt),
		},
	}, nil
}

//func (s *TesterServer) UpdateProblem(ctx context.Context, req *problemv1.UpdateProblemRequest) (*emptypb.Empty, error) {
//	problem := req.GetProblem()
//	if problem == nil {
//		return nil, status.Errorf(codes.Unknown, "") // FIXME
//	}
//	err := s.problemService.UpdateProblem(
//		ctx,
//		&models.Problem{
//			Id:          lib.AsInt32P(problem.Id),
//			Name:        problem.Name,
//			Description: problem.Description,
//			TimeLimit:   problem.TimeLimit,
//			MemoryLimit: problem.MemoryLimit,
//		},
//	)
//	if err != nil {
//		return nil, status.Errorf(codes.Unknown, err.Error()) // FIXME
//	}
//
//	return &emptypb.Empty{}, nil
//}

func (s *TesterServer) DeleteProblem(ctx context.Context, req *problemv1.DeleteProblemRequest) (*emptypb.Empty, error) {
	err := s.problemService.DeleteProblem(ctx, req.GetId())
	if err != nil {
		return nil, status.Errorf(codes.Unknown, err.Error()) // FIXME
	}
	return &emptypb.Empty{}, nil
}