diff --git a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/employeehosts/EmployeeHostsServiceImpl.java b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/employeehosts/EmployeeHostsServiceImpl.java index f6cd346..0498d65 100644 --- a/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/employeehosts/EmployeeHostsServiceImpl.java +++ b/tkdata-model-server/src/main/java/cn/iocoder/yudao/module/tkdata/service/employeehosts/EmployeeHostsServiceImpl.java @@ -10,12 +10,17 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import lombok.extern.slf4j.Slf4j; 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 java.util.*; +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; @@ -37,6 +42,7 @@ import static cn.iocoder.yudao.module.tkdata.enums.ErrorCodeConstants.EMPLOYEE_H */ @Service @Validated +@Slf4j public class EmployeeHostsServiceImpl implements EmployeeHostsService { @Resource @@ -119,6 +125,7 @@ public class EmployeeHostsServiceImpl implements EmployeeHostsService { for (EmployeeHostsSaveReqVO employeeHostsSaveReqVO : hostsList) { NewHostsDO newHostsDO = new NewHostsDO(); newHostsDO.setId(employeeHostsSaveReqVO.getId()); + newHostsDO.setHostsId(employeeHostsSaveReqVO.getHostsId()); newHostsDO.setUserId(employeeHostsSaveReqVO.getUserId()); newHostsDO.setIsAssigned(1); newHostsDOArrayList.add(newHostsDO); @@ -126,8 +133,20 @@ public class EmployeeHostsServiceImpl implements EmployeeHostsService { employeeHostsDO.setOperationStatus(0); employeeHostsDOS.add(employeeHostsDO); } + Long tenantId = TenantContextHolder.getTenantId(); int i = employeeHostsMapper.batchInsertIgnore(employeeHostsDOS); - newHostsMapper.updateBatch(newHostsDOArrayList); + LambdaUpdateWrapper lambdaUpdateWrapper = new LambdaUpdateWrapper<>(); + + List hostsIds = newHostsDOArrayList.stream() + .map(NewHostsDO::getHostsId) + .collect(Collectors.toList()); + int update = newHostsMapper.update(lambdaUpdateWrapper + .in(NewHostsDO::getHostsId,hostsIds) + .eq(NewHostsDO::getTenantId, tenantId) + .set(NewHostsDO::getIsAssigned, 1)); + log.info("修改租户:{}下 {} 个主播分配状态,共更新 {} 条记录",tenantId,hostsIds.size(),update); + +// newHostsMapper.updateBatch(newHostsDOArrayList); return i; }