1

13693261870

2024-12-18 0c8cd957630d43aa71183d34ecb1faf60689f8a4

 src/main/java/com/lf/server/controller/data/DirController.java

@@ -1,10 +1,13 @@
package com.lf.server.controller.data;

import com.lf.server.aspect.SysLog;
import com.lf.server.annotation.SysLog;
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.WebHelper;
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;
@@ -12,6 +15,8 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;

/**
@@ -26,42 +31,55 @@
    @Autowired
    DirService dirService;

    @Autowired
    TokenService tokenService;

    @SysLog()
    @ApiOperation(value = "插入数据")
    @ApiOperation(value = "插入一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "DirEntity", value = "目录实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "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());
            }

            return success(count);
            dirService.insert(entity);

            return success(entity.getId());
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "插入多条数据")
    @ApiOperation(value = "插入多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "DirEntity", value = "目录实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "list", value = "实体类集合", dataType = "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.inserts(list);

            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "删除一条数据")
    @ApiOperation(value = "删除一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "主键ID", dataType = "Integer", paramType = "query", example = "1")
            @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/deleteDir")
    public ResponseMsg<Integer> deleteDir(int id) {
@@ -70,14 +88,14 @@

            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "删除多条数据")
    @ApiOperation(value = "删除多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "ids", value = "主键ID集合", dataType = "List<Integer>", paramType = "query", example = "1,2")
            @ApiImplicitParam(name = "ids", value = "ID数组", dataType = "DirEntity", paramType = "body", example = "")
    })
    @GetMapping(value = "/deleteDirs")
    public ResponseMsg<Integer> deleteDirs(@RequestParam List<Integer> ids) {
@@ -90,31 +108,60 @@

            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "更新一条数据")
    @ApiOperation(value = "更新一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "dirEntity", value = "主键ID集合", dataType = "DictEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "entity", value = "实体类", 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.update(entity);

            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "更新多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "list", value = "实体类集合", dataType = "DirEntity", paramType = "body", example = "")
    })
    @ResponseBody
    @PostMapping(value = "/updateDirs", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updateDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                for (DirEntity entity : list) {
                    entity.setUpdateUser(ue.getId());
                }
            }

            int count = dirService.updates(list);

            return success(count);
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "根据ID查询数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "主键ID", dataType = "Integer", paramType = "query", example = "1")
            @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/selectDir")
    public ResponseMsg<DirEntity> selectDir(int id) {
@@ -123,7 +170,7 @@

            return success(dirEntity);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
            return fail(ex, null);
        }
    }

@@ -135,7 +182,63 @@
            List<DirEntity> list = dirService.selectDirAll();
            return success(list);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
            return fail(ex, null);
        }
    }

    @SysLog()
    @ApiOperation(value = "下载目录结构")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "目录ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/downloadDir")
    public void downloadDir(int id, HttpServletResponse res) {
        try {
            DirEntity entity = dirService.selectDir(id);
            if (null == entity || entity.getPid() > 0) {
                return;
            }

            String zipFile = dirService.createDirs(id);
            if (null == zipFile) {
                return;
            }

            WebHelper.download(zipFile, entity.getName() + ".zip", res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }
    }

    @SysLog()
    @ApiOperation(value = "查询项目")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "name", value = "名称", dataType = "String", paramType = "query", example = "西")
    })
    @GetMapping(value = "/selectProject")
    public ResponseMsg<List<DirEntity>> selectProject(String name) {
        try {
            List<DirEntity> list = dirService.selectProject(name);

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @ApiOperation(value = "递归查询")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "pid", value = "父ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/selectByPid")
    public ResponseMsg<List<DirEntity>> selectByPid(int pid) {
        try {
            List<DirEntity> list = dirService.selectByPid(pid);

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
}