linwe
2024-08-09 8b7258c9427882bb1798f1502eaa35184c6e374e
src/main/java/org/springblade/modules/discuss/controller/UserTopicsController.java
@@ -19,10 +19,13 @@
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import com.xxl.job.core.biz.impl.ExecutorBizImpl;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springblade.common.constant.CommonConstant;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.excel.util.ExcelUtil;
@@ -45,6 +48,8 @@
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.List;
/**
@@ -58,6 +63,8 @@
@RequestMapping("blade-userTopics/userTopics")
@Api(value = "用户议题报表", tags = "用户议题报表接口")
public class UserTopicsController extends BladeController {
   private static Logger logger = LoggerFactory.getLogger(UserTopicsController.class);
   private final IUserTopicsService userTopicsService;
@@ -121,28 +128,7 @@
   @ApiOperationSupport(order = 6)
   @ApiOperation(value = "新增或修改", notes = "传入userTopics")
   public R submit(@Valid @RequestBody UserTopicsEntity userTopics) {
      userTopics.setUserId(AuthUtil.getUserId());
      // 判断是否一户一票 还是一人一票
      IPublicDiscussService bean = SpringUtil.getBean(IPublicDiscussService.class);
      PublicDiscussEntity one = bean.getOne(Wrappers.<PublicDiscussEntity>lambdaQuery().eq(PublicDiscussEntity::getId, userTopics.getPublicDiscussId()));
      // 一户一票
      if (one.getVoteRestrictions().equals(CommonConstant.NUMBER_ONE)) {
         long count = userTopicsService.count(Wrappers.<UserTopicsEntity>lambdaQuery()
            .eq(UserTopicsEntity::getPublicDiscussId, userTopics.getPublicDiscussId())
            .eq(UserTopicsEntity::getHouseCode, userTopics.getHouseCode()));
         if (count > 0) {
            return R.fail("您房屋已经投票,不能重复投票!");
         }
         // 一人一票
      } else {
         long count = userTopicsService.count(Wrappers.<UserTopicsEntity>lambdaQuery()
            .eq(UserTopicsEntity::getPublicDiscussId, userTopics.getPublicDiscussId())
            .eq(UserTopicsEntity::getUserId, userTopics.getUserId()));
         if (count > 0) {
            return R.fail("您已投票,不能重复投票!");
         }
      }
      return R.status(userTopicsService.save(userTopics));
      return R.status(userTopicsService.saveUserTopicsEntity(userTopics));
   }
   /**
@@ -164,10 +150,10 @@
   @PostMapping("/updateBath")
   @ApiOperationSupport(order = 5)
   @ApiOperation(value = "批量更新", notes = "传入topics")
   public R updateBath(@Valid @RequestBody List<TopicsVO> topics) throws Exception {
   public R updateBath(@Valid @RequestBody TopicsVO topics) throws Exception {
      String result = userTopicsService.batchSave(topics);
      if (result.equals("200")) {
         R.success("操作成功");
         return R.data("操作成功");
      }
      return R.fail(result);
   }
@@ -183,5 +169,34 @@
      ExcelUtil.export(response, "投票人员" + DateUtil.time(), "投票人员数据表", list, UserTopicsExcel.class);
   }
   /**
    * 导出Excel文件的方法。
    * 该方法用于将用户主题数据导出为Excel格式,供用户下载。
    *
    * @param response 用于设置HTTP响应,包括内容类型和文件名等信息。
    * @param userTopics 包含要导出的数据的用户主题视图对象。
    * @return 该方法不返回任何内容,而是将数据直接输出到HTTP响应的输出流中。
    */
   @GetMapping("/exportDataIndex")
   public void exportExcel(HttpServletResponse response, UserTopicsVO userTopics) {
       try (OutputStream out = response.getOutputStream()) {
           // 设置响应的内容类型为Excel文件
           response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
           // 设置响应的字符编码为UTF-8
           response.setCharacterEncoding("utf-8");
           // 编码文件名,并替换+号为%20,以兼容URL下载
           String fileName = URLEncoder.encode("投票人员", "UTF-8").replaceAll("\\+", "%20");
           // 设置响应头,指示浏览器以附件形式下载文件,并设置文件名
           response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
           // 调用服务层处理Excel数据的写入
           userTopicsService.handleExcel(out, userTopics);
           // 刷新输出流
           out.flush();
       } catch (Exception e) {
           // 记录异常信息
           logger.error(e.getMessage(), e);
       }
   }
}