java/SmartPlatform.git

			@@ -1,11 +1,13 @@
			package com.ruoyi.system.service.impl;

			import cn.hutool.core.bean.BeanUtil;
			import cn.hutool.core.collection.CollUtil;
			import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
			import com.baomidou.mybatisplus.core.metadata.IPage;
			import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
			import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
			import com.ruoyi.common.utils.SecurityUtils;
			import com.ruoyi.system.constants.AssetDeptConstant;
			import com.ruoyi.system.dto.asset.AssetAdDTO;
			import com.ruoyi.system.mapper.AssetAdMapper;
			import com.ruoyi.system.mapper.AssetAdRentalRecordMapper;
			@@ -14,18 +16,17 @@
			import com.ruoyi.system.query.AssetAdQuery;
			import com.ruoyi.system.service.AssetAdService;
			import com.ruoyi.system.service.AssetTypeService;
			import com.ruoyi.system.service.ISysUserService;
			import com.ruoyi.system.vo.asset.AssetAdDetailVO;
			import com.ruoyi.system.vo.asset.AssetAdRentalRecordVO;
			import com.ruoyi.system.vo.asset.AssetAdVO;
			import lombok.RequiredArgsConstructor;
			import org.apache.commons.lang3.StringUtils;
			import org.springframework.context.annotation.Lazy;
			import org.springframework.stereotype.Service;

			import java.time.LocalDate;
			import java.time.format.DateTimeFormatter;
			import java.time.temporal.ChronoUnit;
			import java.util.ArrayList;
			import java.util.List;
			import java.util.Map;
			import java.util.function.Function;
			@@ -44,6 +45,7 @@
			public class AssetAdServiceImpl extends ServiceImpl<AssetAdMapper, AssetAd> implements AssetAdService {
			private final AssetTypeService assetTypeService;
			private final AssetAdRentalRecordMapper rentalRecordMapper;
			private final ISysUserService sysUserService;

			@Override
			public void addAssetAd(AssetAdDTO dto) {
			@@ -51,6 +53,7 @@
			assetAd.setAssetTypeId(2);
			assetAd.setAssetCode(generateCode());
			assetAd.setCreateBy(SecurityUtils.getLoginUser().getUser().getNickName());
			assetAd.setDeptId(Integer.parseInt(SecurityUtils.getLoginUser().getUser().getDeptId()));
			save(assetAd);
			}

			@@ -83,15 +86,30 @@

			@Override
			public IPage<AssetAdVO> getPageList(AssetAdQuery query) {
			// 1. 分页查询 AssetAd
			Page<AssetAd> page = new Page<>(query.getPageNum(), query.getPageSize());
			LambdaQueryWrapper<AssetAd> queryWrapper = new LambdaQueryWrapper<AssetAd>()
			.like(StringUtils.isNotBlank(query.getKeyword()), AssetAd::getAssetName, query.getKeyword())
			.or()
			.like(StringUtils.isNotBlank(query.getKeyword()), AssetAd::getAssetCode, query.getKeyword())
			.orderByDesc(AssetAd::getCreateTime);
			// 数据权限：超级管理员/资产管理部查看所有数据，其他部门查看当前及下级部门的数据
			Long userId = SecurityUtils.getUserId();
			boolean isAdmin = SecurityUtils.isAdmin(userId);

			Page<AssetAd> assetPage = this.page(page, queryWrapper);
			if (!isAdmin) {
			try {
			// 获取当前用户的部门名称
			String deptName = sysUserService.selectUserById(userId).getDeptName();

			// 非超级管理员且非资产管理部，设置部门权限
			if (!AssetDeptConstant.ASSET_INVENTORY_DEPT_NAMES.contains(deptName)) {
			query.setDeptId(Integer.valueOf(SecurityUtils.getLoginUser().getDeptId()));
			}
			} catch (Exception e) {
			// 如果获取部门信息失败，默认设置部门权限
			try {
			query.setDeptId(Integer.valueOf(SecurityUtils.getLoginUser().getDeptId()));
			} catch (Exception ex) {
			// ignore parse, leave null if cannot parse
			}
			}
			}
			// 1. 分页查询 AssetAd
			Page<AssetAdVO> assetPage = this.baseMapper.getPageList(new Page<>(query.getPageNum(), query.getPageSize()), query);

			// 2. 如果没有数据，返回空分页
			if (assetPage.getRecords().isEmpty()) {
			@@ -100,7 +118,7 @@

			// 3. 获取所有资产ID
			List<Integer> assetAdIds = assetPage.getRecords().stream()
			.map(AssetAd::getId)
			.map(AssetAdVO::getId)
			.collect(Collectors.toList());

			// 4. 查询相关租赁记录
			@@ -118,30 +136,18 @@
			.collect(Collectors.toMap(AssetAdRentalRecord::getAssetAdId, Function.identity()));

			// 6. 组装VO数据
			List<AssetAdVO> voList = new ArrayList<>();
			for (AssetAd asset : assetPage.getRecords()) {
			AssetAdVO vo = BeanUtil.copyProperties(asset, AssetAdVO.class);

			for (AssetAdVO vo : assetPage.getRecords()) {
			// 设置租赁相关信息
			AssetAdRentalRecord currentRental = currentRentalMap.get(asset.getId());
			AssetAdRentalRecord currentRental = currentRentalMap.get(vo.getId());
			if (currentRental != null) {
			vo.setCurrentUseBy(currentRental.getUseBy());
			vo.setRentalPrice(currentRental.getRentalPrice());
			vo.setUseDuration(calculateUseDuration(currentRental.getStartDate(), currentRental.getEndDate()));
			}

			voList.add(vo);
			}

			// 7. 返回分页结果
			Page<AssetAdVO> resultPage = new Page<>(query.getPageNum(), query.getPageSize());
			resultPage.setRecords(voList);
			resultPage.setTotal(assetPage.getTotal());
			resultPage.setSize(assetPage.getSize());
			resultPage.setCurrent(assetPage.getCurrent());
			resultPage.setPages(assetPage.getPages());

			return resultPage;
			return assetPage;
			}

			/**
			@@ -184,13 +190,15 @@

			@Override
			public void importAssetAd(List<AssetAdDTO> list) {
			List<AssetAd> collect = list.stream().map(item -> {
			AssetAd assetAd = BeanUtil.copyProperties(item, AssetAd.class);
			assetAd.setAssetTypeId(2);
			assetAd.setAssetCode(generateCode());
			assetAd.setCreateBy(SecurityUtils.getLoginUser().getUser().getNickName());
			return assetAd;
			}).collect(Collectors.toList());
			saveBatch(collect);
			if (CollUtil.isNotEmpty(list)) {
			List<AssetAd> collect = list.stream().map(item -> {
			AssetAd assetAd = BeanUtil.copyProperties(item, AssetAd.class);
			assetAd.setAssetTypeId(2);
			assetAd.setAssetCode(generateCode());
			assetAd.setCreateBy(SecurityUtils.getLoginUser().getUser().getNickName());
			return assetAd;
			}).collect(Collectors.toList());
			saveBatch(collect);
			}
			}
			}