/* * Copyright (c) 2018-2028, Chill Zhuang All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the dreamlu.net developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: Chill 庄骞 (smallchill@163.com) */ package org.springblade.modules.rotation.service.impl; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import org.apache.commons.lang3.StringUtils; import org.springblade.common.param.CommonParamSet; import org.springblade.common.utils.AuthUtils; import org.springblade.common.utils.RoleUtil; import org.springblade.common.utils.SpringUtils; import org.springblade.core.secure.utils.AuthUtil; import org.springblade.core.tool.utils.SpringUtil; import org.springblade.modules.district.entity.DistrictEntity; import org.springblade.modules.district.service.IDistrictService; import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity; import org.springblade.modules.property.entity.PropertyCompanyEntity; import org.springblade.modules.property.service.IPropertyCompanyDistrictService; import org.springblade.modules.property.service.IPropertyCompanyService; import org.springblade.modules.property.service.IPropertyDistrictUserService; import org.springblade.modules.rotation.entity.RotationEntity; import org.springblade.modules.rotation.mapper.RotationMapper; import org.springblade.modules.rotation.service.IRotationService; import org.springblade.modules.rotation.vo.RotationVO; import org.springblade.modules.system.service.IDeptService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.ArrayList; import java.util.List; import java.util.stream.Collectors; /** * 轮播图 服务实现类 * * @author BladeX * @since 2023-11-16 */ @Service public class RotationServiceImpl extends ServiceImpl implements IRotationService { @Autowired private IDeptService deptService; /** * 自定义分页 * * @param page * @param rotation * @return */ @Override public IPage selectRotationPage(IPage page, RotationVO rotation) { CommonParamSet commonParamSet = new CommonParamSet<>().invoke(RotationVO.class, rotation); String userRole = AuthUtil.getUserRole(); if (RoleUtil.isProperty(userRole)) { // 查询小区id IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class); List districtIds = propertyDistrictUserService.selectPropertyDistrictByUserId(AuthUtil.getUserId()); // 通过用户机构查询用户的物业公司 IPropertyCompanyService bean = SpringUtil.getBean(IPropertyCompanyService.class); PropertyCompanyEntity companyEntity = bean.getOne(Wrappers.lambdaQuery() .eq(PropertyCompanyEntity::getDeptId, AuthUtil.getDeptId()).last("limit 1")); if (companyEntity != null) { IPropertyCompanyDistrictService bean2 = SpringUtils.getBean(IPropertyCompanyDistrictService.class); // 通过物业公司,查询小区 List propertyCompanyDistrictEntityList = bean2.list(Wrappers.lambdaQuery() .eq(PropertyCompanyDistrictEntity::getPropertyCompanyId, companyEntity.getId())); if (propertyCompanyDistrictEntityList.size() > 0) { List collect = propertyCompanyDistrictEntityList.stream().map(i -> i.getDistrictId()).collect(Collectors.toList()); districtIds.addAll(collect); } } rotation.setDistrictIdList(districtIds); if (districtIds.size() == 0) { return page.setRecords(new ArrayList<>()); } rotation.setPropertyFlag(1); } if(AuthUtils.isMj(userRole)){ rotation.setType(1); } if(AuthUtils.isJd(userRole)){ rotation.setType(2); } if(AuthUtils.isAdministratorOrAdmin(userRole)){ rotation.setType(null); } if (StringUtils.isNotBlank(rotation.getAoiCode())) { // 查询房屋的小区 IDistrictService bean = SpringUtils.getBean(IDistrictService.class); DistrictEntity one = bean.getOne(Wrappers.lambdaQuery().eq(DistrictEntity::getAoiCode, rotation.getAoiCode())); List districtIds = new ArrayList<>(); districtIds.add(one.getId()); rotation.setDistrictIdList(districtIds); } List rotationVOList = baseMapper.selectRotationPage(page, rotation, commonParamSet.getIsAdministrator(), null == commonParamSet.getRegionChildCodesList() ? null : String.join("|", commonParamSet.getRegionChildCodesList()), null == commonParamSet.getGridCodeList() ? null : String.join("|", commonParamSet.getGridCodeList())); if (rotationVOList.size() == 0) { // 如果没有对应的轮播图,则采用系统默认的 rotation.setType(3); rotation.setPublicFlag(2); rotation.setDistrictIdList(null); rotationVOList = baseMapper.selectRotationPage(page, rotation, 1, null == commonParamSet.getRegionChildCodesList() ? null : String.join("|", commonParamSet.getRegionChildCodesList()), null == commonParamSet.getGridCodeList() ? null : String.join("|", commonParamSet.getGridCodeList())); } return page.setRecords(rotationVOList); } }