管道基础大数据平台系统开发-【后端】-Server
1
13693261870
2022-10-11 7fd2c877f44b3835fbcc90e420f52090787f0249
src/main/java/com/lf/server/controller/sys/TokenController.java
@@ -4,6 +4,7 @@
import com.lf.server.controller.all.BaseController;
import com.lf.server.entity.all.ResponseMsg;
import com.lf.server.entity.sys.TokenEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.service.sys.TokenService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
@@ -12,6 +13,7 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import java.util.List;
/**
@@ -90,12 +92,18 @@
    @SysLog()
    @ApiOperation(value = "插入一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "tokenEntity", value = "字典实体类", dataType = "com.lf.server.entity.sys.TokenEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "com.lf.server.entity.sys.TokenEntity", paramType = "body", example = "")
    })
    @PostMapping(value = "/insertToken", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insertToken(@RequestBody TokenEntity tokenEntity) {
    public ResponseMsg<Integer> insertToken(@RequestBody TokenEntity entity, HttpServletRequest req) {
        try {
            int count = tokenService.insertToken(tokenEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setCreateUser(ue.getId());
            }
            int count = tokenService.insertToken(entity);
            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
@@ -105,12 +113,19 @@
    @SysLog()
    @ApiOperation(value = "插入多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "tokenEntity", value = "实体类集合", dataType = "List<TokenEntity>", paramType = "body", example = "")
            @ApiImplicitParam(name = "list", value = "实体类集合", dataType = "List<TokenEntity>", paramType = "body", example = "")
    })
    @PostMapping(value = "/insertTokens", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insertTokens(@RequestBody List<TokenEntity> tokenEntity) {
    public ResponseMsg<Integer> insertTokens(@RequestBody List<TokenEntity> list, HttpServletRequest req) {
        try {
            int count = tokenService.insertTokens(tokenEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                for (TokenEntity entity : list) {
                    entity.setCreateUser(ue.getId());
                }
            }
            int count = tokenService.insertTokens(list);
            return success(count);
        } catch (Exception ex) {
@@ -137,7 +152,7 @@
    @SysLog()
    @ApiOperation(value = "删除多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "ids", value = "ID集合", dataType = "List<Integer>", paramType = "query", example = "1,2")
            @ApiImplicitParam(name = "ids", value = "ID数组", dataType = "List<Integer>", paramType = "query", example = "1,2")
    })
    @GetMapping(value = "/deleteTokens")
    public ResponseMsg<Integer> deleteTokens(@RequestParam List<Integer> ids) {
@@ -155,13 +170,19 @@
    @SysLog()
    @ApiOperation(value = "更新一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "tokenEntity", value = "ID集合", dataType = "TokenEntity", paramType = "body", example = "")
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "TokenEntity", paramType = "body", example = "")
    })
    @ResponseBody
    @PostMapping(value = "/updateToken", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updateToken(@RequestBody TokenEntity tokenEntity) {
    public ResponseMsg<Integer> updateToken(@RequestBody TokenEntity entity, HttpServletRequest req) {
        try {
            int count = tokenService.updateToken(tokenEntity);
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setUpdateUser(ue.getId());
            }
            int count = tokenService.updateToken(entity);
            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);