data-analysis/server/src/main/java/edp/davinci/controller/DataRulesController.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.DataRules;
import edp.davinci.model.User;
import edp.davinci.service.DataRulesService;
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 = "/dataRules", tags = "dataRules", produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
@ApiResponses(@ApiResponse(code = 404, message = "dataRules not found"))
@Slf4j
@RestController
@RequestMapping(value = Constants.BASE_API_PATH + "/dataRules", produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
public class DataRulesController extends BaseController
{
    @Autowired
    private DataRulesService dataRulesService;

    /**
     * 获取 数据规则
     *
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "get dataRules")
    @GetMapping(value = "/getDataRules")
    public ResponseEntity getDataRules(@ApiIgnore @CurrentUser User user,
                                         @RequestParam(value = "subjectId", required = false) Long subjectId,
                                         HttpServletRequest request) {
        List<DataRules> dataRules = null;
        if(null == subjectId){
            dataRules = dataRulesService.getDataRules();
        }else{
            dataRules = dataRulesService.getDataRulesListBySubjectId(subjectId);
        }

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

    /**
     * 增加数据规则
     *
     * @param dataRules
     * @param bindingResult
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "create dataRules")
    @PostMapping(value = "/createDataRules",consumes = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity createDataRules(@Valid @RequestBody DataRules dataRules,
                                          @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);
        }

        dataRules = dataRulesService.createDataRules(dataRules);

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

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


        if (invalidId(id) || !id.equals(dataRules.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);
        }

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


    /**
     * 删除deleteDataRules
     *
     * @param id
     * @param user
     * @param request
     * @return
     */
    @ApiOperation(value = "delete dataRules")
    @DeleteMapping("/deleteDataRules/{id}")
    public ResponseEntity deleteDataRules(@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);
        }

        dataRulesService.deleteDataRules(id);

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