管道基础大数据平台系统开发-【后端】-Server
1
13693261870
2022-10-11 9c5c93a36294b3f14536689d548928196c367de8
src/main/java/com/lf/server/controller/data/DirController.java
@@ -4,16 +4,18 @@
import com.lf.server.controller.all.BaseController;
import com.lf.server.entity.all.ResponseMsg;
import com.lf.server.entity.data.DirEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.StringHelper;
import com.lf.server.service.data.DirService;
import com.lf.server.service.sys.TokenService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import java.util.List;
/**
@@ -28,15 +30,23 @@
    @Autowired
    DirService dirService;
    @Autowired
    TokenService tokenService;
    @SysLog()
    @ApiOperation(value = "插入数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "DirEntity", value = "目录实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "entity", value = "目录实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
    })
    @PostMapping(value = "/insertDir", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insertDir(@RequestBody DirEntity dirEntity) {
    public ResponseMsg<Integer> insertDir(@RequestBody DirEntity entity, HttpServletRequest req) {
        try {
            int count = dirService.insertDir(dirEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setCreateUser(ue.getId());
            }
            int count = dirService.insertDir(entity);
            return success(count);
        } catch (Exception ex) {
@@ -47,12 +57,17 @@
    @SysLog()
    @ApiOperation(value = "插入多条数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "DirEntity", value = "目录实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "list", value = "目录实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
    })
    @PostMapping(value = "/insertDirs", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insertDirs(@RequestBody List<DirEntity> dirEntity) {
    public ResponseMsg<Integer> insertDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
        try {
            int count = dirService.insertDirs(dirEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            for (DirEntity entity : list) {
                entity.setCreateUser(ue.getId());
            }
            int count = dirService.insertDirs(list);
            return success(count);
        } catch (Exception ex) {
@@ -99,13 +114,18 @@
    @SysLog()
    @ApiOperation(value = "更新一条数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "dirEntity", value = "主键ID集合", dataType = "DictEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "entity", value = "主键ID集合", dataType = "DictEntity", paramType = "body", example = "")
    })
    @ResponseBody
    @PostMapping(value = "/updateDir", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updateDir(@RequestBody DirEntity dirEntity) {
    public ResponseMsg<Integer> updateDir(@RequestBody DirEntity entity, HttpServletRequest req) {
        try {
            int count = dirService.updateDir(dirEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setUpdateUser(ue.getId());
            }
            int count = dirService.updateDir(entity);
            return success(count);
        } catch (Exception ex) {
@@ -116,13 +136,20 @@
    @SysLog()
    @ApiOperation(value = "更新多条数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "DirEntity", value = "主键ID集合", dataType = "DirEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "list", value = "主键ID集合", dataType = "DirEntity", paramType = "body", example = "")
    })
    @ResponseBody
    @PostMapping(value = "/updateDirs", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updateDirs(@RequestBody List<DirEntity> dirEntity) {
    public ResponseMsg<Integer> updateDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
        try {
            int count = dirService.updateDirs(dirEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                for (DirEntity entity : list) {
                    entity.setUpdateUser(ue.getId());
                }
            }
            int count = dirService.updateDirs(list);
            return success(count);
        } catch (Exception ex) {