优化

zhanghua

2023-09-08 7ef4892f9f24f941aca37e6b3991b808a0aca619

 ycl-platform/src/main/java/com/ycl/service/resources/impl/ImageResourcesServiceImpl.java

@@ -1,6 +1,9 @@
package com.ycl.service.resources.impl;


import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ycl.entity.resources.ImageResources;
@@ -10,6 +13,11 @@
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * <p>
@@ -26,10 +34,22 @@
    ImageResourcesMapper imageResourcesMapper;

    @Override
    public Page<MediaVO> selectImages(Integer type, String startTime, String endTime, Long size, Long current) {
    public Page<MediaVO> selectImages(String type, String startTime, String endTime, Long size, Long current) {
        Page<MediaVO> imagesVOPage = new Page<>();
        imagesVOPage.setCurrent(current);
        imagesVOPage.setSize(size);
        return imageResourcesMapper.selectImagePage(imagesVOPage,type,startTime,endTime);
        return imageResourcesMapper.selectImagePage(imagesVOPage, type, startTime, endTime);
    }

    @Override
    public Map<Long, String> getUrlMap(Collection<Long> ids, String type) {
        LambdaQueryWrapper<ImageResources> wrapper = new QueryWrapper<ImageResources>().lambda().in(ImageResources::getBelongToId, ids).eq(ImageResources::getType, type);
        List<ImageResources> list = imageResourcesMapper.selectList(wrapper);
        if (CollUtil.isEmpty(list)) {
            return new HashMap<>();

        }
        return list.stream().collect(Collectors.toMap(ImageResources::getBelongToId, ImageResources::getUrl, (v1, v2) -> v1));
    }

}