From cff854df2740e862a9255df58c7629e5c559d973 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期二, 21 十月 2025 19:54:04 +0800
Subject: [PATCH] 广告物料单价接口
---
ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetAdServiceImpl.java | 76 +++++++++++++++++++++-----------------
1 files changed, 42 insertions(+), 34 deletions(-)
diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetAdServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetAdServiceImpl.java
index 29d748e..44dd518 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetAdServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetAdServiceImpl.java
@@ -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);
+ }
}
}
--
Gitblit v1.7.1