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 }