zqy
3 天以前 27fff67070616dacd6c5af18d07a988d28a04de7
ruoyi-admin/src/main/java/com/ruoyi/web/controller/zhang/ZIdeaController.java
@@ -4,8 +4,10 @@
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.text.Convert;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.domain.ZIdea;
@@ -13,6 +15,8 @@
import com.ruoyi.service.ZIdeaService;
import com.ruoyi.service.ZIdeaShareService;
import com.ruoyi.service.ZPropertyDownloadService;
import com.ruoyi.system.service.ISysRoleService;
import com.ruoyi.system.service.ISysUserService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
@@ -48,6 +52,9 @@
    @Autowired
    private ZIdeaShareService zIdeaShareService;
    @Autowired
    private ISysUserService sysUserService;
    /**
     * 查询所有记录
@@ -82,6 +89,13 @@
    @PostMapping("/export")
    public void export(HttpServletResponse response)
    {
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Boolean userStatus = sysUserService.findUserStatus(user.getUserId());
        if(userStatus) {
             throw new IllegalArgumentException("您的试用期已经结束,请付费继续使用");
        }
        List<ZIdea> list = zIdeaService.selectByCondition();
        log.info("导出记录为:{}",list);
        ExcelUtil<ZIdea> util = new ExcelUtil<>(ZIdea.class);
@@ -109,7 +123,10 @@
    @PostMapping("/importData")
    public AjaxResult importData(@RequestParam("excelImport") MultipartFile file) throws Exception
    {
        return zIdeaService.importExcel(file);
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Boolean userStatus = sysUserService.findUserStatus(user.getUserId());
        return userStatus?AjaxResult.error("您的试用期已经结束,请付费继续使用"):zIdeaService.importExcel(file);
    }
    /**
@@ -120,7 +137,10 @@
    @PostMapping
    public AjaxResult add(@RequestBody ZIdea zIdea)
    {
        return zIdeaService.mySave(zIdea);
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Boolean userStatus = sysUserService.findUserStatus(user.getUserId());
        return userStatus?AjaxResult.error("您的试用期已经结束,请付费继续使用"):zIdeaService.mySave(zIdea);
    }
    /**
@@ -131,7 +151,9 @@
    @PutMapping
    public AjaxResult edit(@RequestBody ZIdea zIdea)
    {
        return toAjax(zIdeaService.updateById(zIdea));
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Boolean userStatus = sysUserService.findUserStatus(user.getUserId());
        return userStatus?AjaxResult.error("您的试用期已经结束,请付费继续使用"):toAjax(zIdeaService.updateById(zIdea));
    }
//
    /**
@@ -146,7 +168,9 @@
        if (! (zIdeaDownloadService.deleteData(ids) && zIdeaShareService.deleteByContentId(ids)  ) ){
            return AjaxResult.error();
        }
        return toAjax(zIdeaService.removeByIds(Arrays.asList(ids)));
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Boolean userStatus = sysUserService.findUserStatus(user.getUserId());
        return userStatus?AjaxResult.error("您的试用期已经结束,请付费继续使用"): toAjax(zIdeaService.removeByIds(Arrays.asList(ids)));
    }
}