ziin/vvDataAdmin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1.添加员工导出主播功能

Browse Source
This commit is contained in:
ziin
2025-09-04 19:28:57 +08:00
parent 55c585bddc
commit ac29f38ae7
1 changed files with 27 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/employeehosts/EmployeeHostsServiceImpl.java
View File

@@ -1,10 +1,17 @@
package cn.iocoder.yudao.module.tkdata.service.employeehosts; package cn.iocoder.yudao.module.tkdata.service.employeehosts;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.security.core.LoginUser; import cn.iocoder.yudao.framework.security.core.LoginUser;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils; import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder; import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
import cn.iocoder.yudao.module.tkdata.controller.admin.employeehosts.vo.CompletedRateVO;
import cn.iocoder.yudao.module.tkdata.controller.admin.employeehosts.vo.EmployeeHostsPageReqVO;
import cn.iocoder.yudao.module.tkdata.controller.admin.employeehosts.vo.EmployeeHostsSaveReqVO;
import cn.iocoder.yudao.module.tkdata.dal.dataobject.employeehosts.EmployeeHostsDO;
import cn.iocoder.yudao.module.tkdata.dal.dataobject.newhosts.NewHostsDO; import cn.iocoder.yudao.module.tkdata.dal.dataobject.newhosts.NewHostsDO;
import cn.iocoder.yudao.module.tkdata.dal.mysql.employeehosts.EmployeeHostsMapper;
import cn.iocoder.yudao.module.tkdata.dal.mysql.newhosts.NewHostsMapper; import cn.iocoder.yudao.module.tkdata.dal.mysql.newhosts.NewHostsMapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
@@ -12,26 +19,15 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.util.StopWatch;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;
import java.util.*; import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors; import java.util.stream.Collectors;
import cn.iocoder.yudao.module.tkdata.controller.admin.employeehosts.vo.*;
import cn.iocoder.yudao.module.tkdata.dal.dataobject.employeehosts.EmployeeHostsDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.tkdata.dal.mysql.employeehosts.EmployeeHostsMapper;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.diffList;
import static cn.iocoder.yudao.module.tkdata.enums.ErrorCodeConstants.EMPLOYEE_HOSTS_NOT_EXISTS; import static cn.iocoder.yudao.module.tkdata.enums.ErrorCodeConstants.EMPLOYEE_HOSTS_NOT_EXISTS;
@@ -71,7 +67,7 @@ public class EmployeeHostsServiceImpl implements EmployeeHostsService {
for (EmployeeHostsDO hostsDO : employeeHostsDOS) { for (EmployeeHostsDO hostsDO : employeeHostsDOS) {
hostsDO.setOperationStatus(updateReqVO.getOperationStatus()); hostsDO.setOperationStatus(updateReqVO.getOperationStatus());
hostsDO.setRemake(updateReqVO.getRemake()); hostsDO.setRemake(updateReqVO.getRemake());
if (updateReqVO.getFlag() != null){ if (updateReqVO.getFlag() != null) {
hostsDO.setFlag(updateReqVO.getFlag()); hostsDO.setFlag(updateReqVO.getFlag());
} }
} }
@@ -114,10 +110,15 @@ public class EmployeeHostsServiceImpl implements EmployeeHostsService {
@Override @Override
public PageResult<EmployeeHostsDO> getEmployeeHostsPage(EmployeeHostsPageReqVO pageReqVO) { public PageResult<EmployeeHostsDO> getEmployeeHostsPage(EmployeeHostsPageReqVO pageReqVO) {
LoginUser loginUser = SecurityFrameworkUtils.getLoginUser();
pageReqVO.setTenantId(TenantContextHolder.getTenantId()); pageReqVO.setTenantId(TenantContextHolder.getTenantId());
IPage<EmployeeHostsDO> iPage = new Page<>(pageReqVO.getPageNo(),pageReqVO.getPageSize()); if (loginUser != null) {
IPage<EmployeeHostsDO> newHostsDOIPage = employeeHostsMapper.selectPageWithXML(iPage, pageReqVO); pageReqVO.setUserId(loginUser.getId());
return new PageResult<>(newHostsDOIPage.getRecords(),newHostsDOIPage.getTotal()); IPage<EmployeeHostsDO> iPage = new Page<>(pageReqVO.getPageNo(), pageReqVO.getPageSize());
IPage<EmployeeHostsDO> newHostsDOIPage = employeeHostsMapper.selectPageWithXMLWithSelf(iPage, pageReqVO);
return new PageResult<>(newHostsDOIPage.getRecords(), newHostsDOIPage.getTotal());
}
return null;
} }
@Override @Override
@@ -144,10 +145,10 @@ public class EmployeeHostsServiceImpl implements EmployeeHostsService {
.map(NewHostsDO::getHostsId) .map(NewHostsDO::getHostsId)
.collect(Collectors.toList()); .collect(Collectors.toList());
int update = newHostsMapper.update(lambdaUpdateWrapper int update = newHostsMapper.update(lambdaUpdateWrapper
.in(NewHostsDO::getHostsId,hostsIds) .in(NewHostsDO::getHostsId, hostsIds)
.eq(NewHostsDO::getTenantId, tenantId) .eq(NewHostsDO::getTenantId, tenantId)
.set(NewHostsDO::getIsAssigned, 1)); .set(NewHostsDO::getIsAssigned, 1));
log.info("修改租户:{}下 {} 个主播分配状态，共更新 {} 条记录",tenantId,hostsIds.size(),update); log.info("修改租户:{}下 {} 个主播分配状态，共更新 {} 条记录", tenantId, hostsIds.size(), update);
// newHostsMapper.updateBatch(newHostsDOArrayList); // newHostsMapper.updateBatch(newHostsDOArrayList);
return i; return i;
@@ -159,9 +160,9 @@ public class EmployeeHostsServiceImpl implements EmployeeHostsService {
pageReqVO.setTenantId(TenantContextHolder.getTenantId()); pageReqVO.setTenantId(TenantContextHolder.getTenantId());
if (loginUser != null) { if (loginUser != null) {
pageReqVO.setUserId(loginUser.getId()); pageReqVO.setUserId(loginUser.getId());
IPage<EmployeeHostsDO> iPage = new Page<>(pageReqVO.getPageNo(),pageReqVO.getPageSize()); IPage<EmployeeHostsDO> iPage = new Page<>(pageReqVO.getPageNo(), pageReqVO.getPageSize());
IPage<EmployeeHostsDO> newHostsDOIPage = employeeHostsMapper.selectPageWithXMLWithSelf(iPage, pageReqVO); IPage<EmployeeHostsDO> newHostsDOIPage = employeeHostsMapper.selectPageWithXMLWithSelf(iPage, pageReqVO);
return new PageResult<>(newHostsDOIPage.getRecords(),newHostsDOIPage.getTotal()); return new PageResult<>(newHostsDOIPage.getRecords(), newHostsDOIPage.getTotal());
} }
return null; return null;
} }