cn.edu.zjnu.acm.judge.problem.ProblemStatusController.java Source code

Introduction

Here is the source code for cn.edu.zjnu.acm.judge.problem.ProblemStatusController.java

Source

/*
 * Copyright 2016 ZJNU ACM.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package cn.edu.zjnu.acm.judge.problem;

import cn.edu.zjnu.acm.judge.domain.Problem;
import cn.edu.zjnu.acm.judge.domain.ScoreCount;
import cn.edu.zjnu.acm.judge.domain.Submission;
import cn.edu.zjnu.acm.judge.exception.MessageException;
import cn.edu.zjnu.acm.judge.mapper.ProblemMapper;
import cn.edu.zjnu.acm.judge.mapper.SubmissionMapper;
import cn.edu.zjnu.acm.judge.service.UserDetailService;
import cn.edu.zjnu.acm.judge.util.ResultType;
import cn.edu.zjnu.acm.judge.util.URLBuilder;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.web.PageableDefault;
import org.springframework.http.HttpStatus;
import org.springframework.security.core.Authentication;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

/**
 *
 * @author zhanhb
 */
@Controller
@Slf4j
public class ProblemStatusController {

    @Autowired
    private ProblemMapper problemMapper;
    @Autowired
    private SubmissionMapper submissionMapper;

    @GetMapping("/gotoproblem")
    public String gotoProblem(@RequestParam(value = "pid", required = false) String pid,
            RedirectAttributes redirectAttributes) {
        try {
            redirectAttributes.addAttribute("problem_id", Long.parseLong(pid));
            return "redirect:/showproblem";
        } catch (NumberFormatException ex) {
            redirectAttributes.addAttribute("sstr", pid);
            return "redirect:/searchproblem";
        }
    }

    @GetMapping("/problemstatus")
    @SuppressWarnings("AssignmentToMethodParameter")
    public String status(HttpServletRequest request, @RequestParam("problem_id") long id,
            @PageableDefault(size = 20, sort = { "time", "memory", "code_length" }) Pageable pageable,
            Authentication authentication) {
        log.debug("{}", pageable);
        if (pageable.getPageSize() > 500) {
            pageable = new PageRequest(pageable.getPageNumber(), 500, pageable.getSort());
        }
        Problem problem = problemMapper.findOneNoI18n(id);
        if (problem == null) {
            throw new MessageException("No such problem", HttpStatus.NOT_FOUND);
        }
        final Long contestId = problem.getContest();
        request.setAttribute("contestId", contestId);
        List<ScoreCount> list = problemMapper.groupByScore(id);
        ArrayList<String> scores = new ArrayList<>(list.size());
        ArrayList<Long> counts = new ArrayList<>(list.size());
        ArrayList<String> urls = new ArrayList<>(list.size());
        for (ScoreCount scoreCount : list) {
            int score = scoreCount.getScore();
            scores.add(ResultType.getShowsourceString(score));
            counts.add(scoreCount.getCount());
            urls.add(request.getContextPath() + "/status?problem_id=" + id + "&score=" + score);
        }
        List<Submission> bestSubmissions = submissionMapper.bestSubmission(id, pageable);
        long total = pageable.getOffset() + bestSubmissions.size()
                + (pageable.getPageSize() == bestSubmissions.size() ? 1 : 0);
        PageImpl<Submission> page = new PageImpl<>(bestSubmissions, pageable, total);
        boolean isAdmin = UserDetailService.isAdminLoginned(request);
        boolean isSourceBrowser = UserDetailService.isSourceBrowser(request);
        boolean canView = isAdmin || isSourceBrowser;

        request.setAttribute("page", page);
        request.setAttribute("sa", Arrays.asList(counts, scores, urls));
        request.setAttribute("problem", problem);
        request.setAttribute("url", URLBuilder.fromRequest(request).replaceQueryParam("page").toString());
        request.setAttribute("contestId", contestId);
        request.setAttribute("canView", canView);
        request.setAttribute("authentication", authentication);

        return "problems/status";
    }

}