From 56efd0aa54e2c5cea81ff9b28855bca520dee475 Mon Sep 17 00:00:00 2001 From: mitao <2763622819@qq.com> Date: 星期五, 17 十月 2025 16:42:43 +0800 Subject: [PATCH] 无形广告资产接口 --- ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetAdServiceImpl.java | 164 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 164 insertions(+), 0 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 00de1bd..97ed50b 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,10 +1,35 @@ package com.ruoyi.system.service.impl; +import cn.hutool.core.bean.BeanUtil; +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.dto.asset.AssetAdDTO; import com.ruoyi.system.mapper.AssetAdMapper; +import com.ruoyi.system.mapper.AssetAdRentalRecordMapper; import com.ruoyi.system.model.AssetAd; +import com.ruoyi.system.model.AssetAdRentalRecord; +import com.ruoyi.system.query.AssetAdQuery; import com.ruoyi.system.service.AssetAdService; +import com.ruoyi.system.service.AssetTypeService; +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; +import java.util.stream.Collectors; /** * <p> @@ -15,6 +40,145 @@ * @since 2025-09-15 */ @Service +@RequiredArgsConstructor(onConstructor_ = {@Lazy}) public class AssetAdServiceImpl extends ServiceImpl<AssetAdMapper, AssetAd> implements AssetAdService { + private final AssetTypeService assetTypeService; + private final AssetAdRentalRecordMapper rentalRecordMapper; + @Override + public void addAssetAd(AssetAdDTO dto) { + AssetAd assetAd = BeanUtil.copyProperties(dto, AssetAd.class); + assetAd.setAssetTypeId(2); + assetAd.setAssetCode(generateCode()); + assetAd.setCreateBy(SecurityUtils.getLoginUser().getUser().getNickName()); + save(assetAd); + } + + private String generateCode() { + // 根据资产类型ID获取资产编码前缀 + String typeCodePrefix = assetTypeService.getAssetCodePrefix(2); + LocalDate today = LocalDate.now(); + String dateStr = today.format(DateTimeFormatter.ofPattern("yyyyMMdd")); + String prefix = typeCodePrefix + dateStr + "-"; + + // 查询当天该类型资产的数量 + Long count = this.lambdaQuery() + .like(AssetAd::getAssetCode, prefix) + .ge(AssetAd::getCreateTime, today.atStartOfDay()) + .lt(AssetAd::getCreateTime, today.plusDays(1).atStartOfDay()) + .count(); + + int sequence = (count != null ? count.intValue() : 0) + 1; + String sequenceStr = String.format("%04d", sequence); + + return prefix + sequenceStr; + } + + @Override + public void editAssetAd(AssetAdDTO dto) { + AssetAd assetAd = BeanUtil.copyProperties(dto, AssetAd.class); + assetAd.setUpdateBy(SecurityUtils.getLoginUser().getUser().getNickName()); + updateById(assetAd); + } + + @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); + + Page<AssetAd> assetPage = this.page(page, queryWrapper); + + // 2. 如果没有数据,返回空分页 + if (assetPage.getRecords().isEmpty()) { + return new Page<>(query.getPageNum(), query.getPageSize()); + } + + // 3. 获取所有资产ID + List<Integer> assetAdIds = assetPage.getRecords().stream() + .map(AssetAd::getId) + .collect(Collectors.toList()); + + // 4. 查询相关租赁记录 + LocalDate today = LocalDate.now(); + List<AssetAdRentalRecord> allRentalRecords = rentalRecordMapper.selectList( + new LambdaQueryWrapper<AssetAdRentalRecord>() + .in(AssetAdRentalRecord::getAssetAdId, assetAdIds) + .ge(AssetAdRentalRecord::getStartDate, today) + .le(AssetAdRentalRecord::getEndDate,today) + .orderByDesc(AssetAdRentalRecord::getCreateTime) + ); + + // 5. 转换为Map + Map<Integer, AssetAdRentalRecord> currentRentalMap = allRentalRecords.stream() + .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); + + // 设置租赁相关信息 + AssetAdRentalRecord currentRental = currentRentalMap.get(asset.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; + } + + /** + * 计算使用期限(月) + */ + private Integer calculateUseDuration(LocalDate startDate, LocalDate endDate) { + if (startDate == null || endDate == null) { + return null; + } + // 计算月份差值 + long months = ChronoUnit.MONTHS.between(startDate, endDate); + return (int) months; + } + + @Override + public AssetAdDetailVO getDetail(Integer id) { + // 1. 查询资产基础信息 + AssetAd assetAd = getById(id); + if (assetAd == null) { + return null; + } + + // 2. 查询租赁记录 + List<AssetAdRentalRecord> rentalRecords = rentalRecordMapper.selectList( + new LambdaQueryWrapper<AssetAdRentalRecord>() + .eq(AssetAdRentalRecord::getAssetAdId, id) + .orderByDesc(AssetAdRentalRecord::getStartDate) + ); + + // 3. 转换为AssetAdDetailVO + AssetAdDetailVO detailVO = BeanUtil.copyProperties(assetAd, AssetAdDetailVO.class); + + // 4. 转换租赁记录为VO列表 + List<AssetAdRentalRecordVO> rentalRecordVOList = BeanUtil.copyToList(rentalRecords, AssetAdRentalRecordVO.class); + + detailVO.setRentalRecordVOList(rentalRecordVOList); + + return detailVO; + } } -- Gitblit v1.7.1