1

13693261870

2022-10-11 9c5c93a36294b3f14536689d548928196c367de8

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

@@ -3,8 +3,11 @@
import com.lf.server.aspect.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.data.DownloadEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.service.data.DownloadService;
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,7 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

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

/**
@@ -24,6 +28,9 @@
public class DownloadController extends BaseController {
    @Autowired
    DownloadService downloadService;

    @Autowired
    TokenService tokenService;

    @SysLog()
    @ApiOperation(value = "查询记录数")
@@ -123,8 +130,13 @@
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "DownloadEntity", paramType = "body")
    })
    @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insert(@RequestBody DownloadEntity entity) {
    public ResponseMsg<Integer> insert(@RequestBody DownloadEntity entity, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setCreateUser(ue.getId());
            }

            int count = downloadService.insert(entity);

            return success(count);
@@ -139,8 +151,15 @@
            @ApiImplicitParam(name = "list", value = "实体类集合", dataType = "DownloadEntity", paramType = "body")
    })
    @PostMapping(value = "/inserts", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> inserts(@RequestBody List<DownloadEntity> list) {
    public ResponseMsg<Integer> inserts(@RequestBody List<DownloadEntity> list, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                for (DownloadEntity entity : list) {
                    entity.setCreateUser(ue.getId());
                }
            }

            int count = downloadService.inserts(list);

            return success(count);
@@ -192,8 +211,13 @@
    })
    @ResponseBody
    @PostMapping(value = "/update", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> update(@RequestBody DownloadEntity entity) {
    public ResponseMsg<Integer> update(@RequestBody DownloadEntity entity, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setDownloadUser(ue.getId());
            }

            int count = downloadService.update(entity);

            return success(count);
@@ -209,8 +233,15 @@
    })
    @ResponseBody
    @PostMapping(value = "/updates", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updates(@RequestBody List<DownloadEntity> list) {
    public ResponseMsg<Integer> updates(@RequestBody List<DownloadEntity> list, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                for (DownloadEntity entity : list) {
                    entity.setDownloadUser(ue.getId());
                }
            }

            int count = downloadService.updates(list);

            return success(count);