data-analysis/server/src/main/java/edp/davinci/controller/QualityTaskController.java

package edp.davinci.controller;

import edp.core.annotation.CurrentUser;
import edp.davinci.common.controller.BaseController;
import edp.davinci.core.common.Constants;
import edp.davinci.core.common.ResultMap;
import edp.davinci.model.QualityAuditor;
import edp.davinci.model.QualityTask;
import edp.davinci.model.User;
import edp.davinci.service.QualityAuditorService;
import edp.davinci.service.QualityTaskService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiResponse;
import io.swagger.annotations.ApiResponses;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletRequest;
import javax.validation.Valid;
import java.util.List;

@Api(value = "/qualityTask", tags = "qualityTask", produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
@ApiResponses(@ApiResponse(code = 404, message = "qualityTask not found"))
@Slf4j
@RestController
@RequestMapping(value = Constants.BASE_API_PATH + "/qualityTask", produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
public class QualityTaskController extends BaseController
{
    @Autowired
    private QualityTaskService qualityTaskService;

    @Autowired
    private QualityAuditorService qualityAuditorService;

    /**
     * 获取 质量任务
     *
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "get qualityTask")
    @GetMapping(value = "/getQualityTask")
    public ResponseEntity getQualityTask(@ApiIgnore @CurrentUser User user,
                                       HttpServletRequest request) {

        List<QualityTask> qualityTasks = qualityTaskService.getQualityTask();
        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request).payloads(qualityTasks));
    }

    /**
     * 增加 质量任务
     *
     * @param qualityTask
     * @param bindingResult
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "create qualityTask")
    @PostMapping(value = "/createQualityTask",consumes = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity createQualityTask(@Valid @RequestBody QualityTask qualityTask,
                                          @ApiIgnore BindingResult bindingResult,
                                          @ApiIgnore @CurrentUser User user,
                                          HttpServletRequest request) {

        if (bindingResult.hasErrors()) {
            ResultMap resultMap = new ResultMap(tokenUtils).failAndRefreshToken(request).message(bindingResult.getFieldErrors().get(0).getDefaultMessage());
            return ResponseEntity.status(resultMap.getCode()).body(resultMap);
        }

        qualityTask = qualityTaskService.createQualityTask(qualityTask);

        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request).payload(qualityTask));
    }

    /**
     * 修改 qualityTask
     *
     * @param id
     * @param qualityTask
     * @param bindingResult
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "update qualityTask")
    @PutMapping(value = "/updateQualityTask/{id}", consumes = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity updateQualityTask(@PathVariable Long id,
                                          @Valid @RequestBody QualityTask qualityTask,
                                          @ApiIgnore BindingResult bindingResult,
                                          @ApiIgnore @CurrentUser User user,
                                          HttpServletRequest request) {


        if (invalidId(id) || !id.equals(qualityTask.getId())) {
            ResultMap resultMap = new ResultMap(tokenUtils).failAndRefreshToken(request).message("Invalid dataSubject id");
            return ResponseEntity.status(resultMap.getCode()).body(resultMap);
        }

        if (bindingResult.hasErrors()) {
            ResultMap resultMap = new ResultMap(tokenUtils).failAndRefreshToken(request).message(bindingResult.getFieldErrors().get(0).getDefaultMessage());
            return ResponseEntity.status(resultMap.getCode()).body(resultMap);
        }

        qualityTaskService.updateQualityTask(qualityTask);
        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request));
    }


    /**
     * 删除 qualityTask
     *
     * @param id
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "delete qualityTask")
    @DeleteMapping("/deleteQualityTask/{id}")
    public ResponseEntity deleteQualityTask(@PathVariable Long id,
                                          @ApiIgnore @CurrentUser User user,
                                          HttpServletRequest request) {
        if (invalidId(id)) {
            ResultMap resultMap = new ResultMap(tokenUtils).failAndRefreshToken(request).message("Invalid dataSubject id");
            return ResponseEntity.status(resultMap.getCode()).body(resultMap);
        }

        qualityTaskService.deleteQualityTask(id);

        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request));
    }

    /**
     * 立即稽核
     *
     * @param id
     * @param request
     * @return
     */
    @ApiOperation(value = "auditor")
    @PutMapping(value = "/auditor/{id}", consumes = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity auditor(@PathVariable Long id,HttpServletRequest request){
        // 1. 根据任务id，查询质量任务
        qualityTaskService.auditor(id);
        // 2.

        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request));
    }

    /**
     * 设置调度
     *
     * @param id
     * @param request
     * @return
     */
    @ApiOperation(value = "setDispatch")
    @PutMapping(value = "/setDispatch/{id}", consumes = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity setDispatch(@PathVariable Long id,HttpServletRequest request){


        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request));
    }


    /**
     *  质量报告
     *
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "get qualityReport")
    @GetMapping(value = "/qualityReport")
    public ResponseEntity qualityReport(@ApiIgnore @CurrentUser User user,
                                         HttpServletRequest request) {

        List<QualityAuditor> qualityAuditors = qualityAuditorService.getQualityAuditors();
        return ResponseEntity.ok(new ResultMap(tokenUtils).successAndRefreshToken(request).payloads(qualityAuditors));
    }

    public static void main(String[] args)
    {
        int resultLimit = 10000;
        int limit =-1;
        int count = limit > resultLimit ? resultLimit : limit > 0 ? limit : resultLimit;
        System.out.println(count);
    }


}