diff --git a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/NewHostsController.java b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/NewHostsController.java index b13b63e..160f6a0 100644 --- a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/NewHostsController.java +++ b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/NewHostsController.java @@ -6,6 +6,7 @@ import cn.iocoder.yudao.module.tkdata.controller.admin.newhosts.vo.NewHostsRespV import cn.iocoder.yudao.module.tkdata.controller.admin.newhosts.vo.NewHostsSaveReqVO; import cn.iocoder.yudao.module.tkdata.dal.dataobject.newhosts.NewHostsDO; import cn.iocoder.yudao.module.tkdata.service.newhosts.NewHostsService; +import lombok.extern.slf4j.Slf4j; import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; import org.springframework.validation.annotation.Validated; @@ -33,6 +34,7 @@ import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*; +@Slf4j @Tag(name = "管理后台 - 主播数据") @RestController @RequestMapping("/server/new-hosts") @@ -108,6 +110,10 @@ public class NewHostsController { HttpServletResponse response) throws IOException { pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE); List list = newHostsService.getNewHostsPage(pageReqVO).getList(); + if (pageReqVO.getDeleteFlag()){ + newHostsService.deleteExportData(list); + log.info("删除导出数据{}",list.size() ); + } // 导出 Excel ExcelUtils.write(response, "主播数据.xls", "数据", NewHostsRespVO.class, BeanUtils.toBean(list, NewHostsRespVO.class)); diff --git a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/vo/NewHostsPageReqVO.java b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/vo/NewHostsPageReqVO.java index 5c2d671..a1f5793 100644 --- a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/vo/NewHostsPageReqVO.java +++ b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/controller/admin/newhosts/vo/NewHostsPageReqVO.java @@ -135,4 +135,7 @@ public class NewHostsPageReqVO extends PageParam { */ @Schema(description = "是否建联", example = "1") private Byte operationStatus; + + @Schema(description = "是否删除原数据") + private Boolean deleteFlag; } \ No newline at end of file diff --git a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsService.java b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsService.java index f48354b..7376c1a 100644 --- a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsService.java +++ b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsService.java @@ -62,4 +62,6 @@ public interface NewHostsService { PageResult getNewHostsPage(NewHostsPageReqVO pageReqVO); void batchUpdateNewHosts(@Valid List updateReqVO); + + void deleteExportData(List list); } \ No newline at end of file diff --git a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsServiceImpl.java b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsServiceImpl.java index a02dcba..7a82fc9 100644 --- a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsServiceImpl.java +++ b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/newhosts/NewHostsServiceImpl.java @@ -106,4 +106,9 @@ public class NewHostsServiceImpl implements NewHostsService { newHostsMapper.updateBatch(newHostsDOS); } + @Override + public void deleteExportData(List list) { + newHostsMapper.deleteBatchIds(list); + } + } \ No newline at end of file