4dkankan-center-scene/src/main/java/com/fdkankan/scene/service/impl/SceneResourceServiceImpl.java

package com.fdkankan.scene.service.impl;

import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.scene.entity.SceneResource;
import com.fdkankan.scene.entity.SceneResourceCooperation;
import com.fdkankan.scene.mapper.ISceneResourceMapper;
import com.fdkankan.scene.service.ISceneResourceCooperationService;
import com.fdkankan.scene.service.ISceneResourceService;
import java.util.List;
import java.util.stream.Collectors;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

/**
 * <p>
 * 场景资源表 服务实现类
 * </p>
 *
 * @author 
 * @since 2022-07-18
 */
@Service
public class SceneResourceServiceImpl extends ServiceImpl<ISceneResourceMapper, SceneResource> implements ISceneResourceService {

    @Autowired
    private ISceneResourceCooperationService sceneResourceCooperationService;

    @Override
    public List<SceneResource> findByCooperationId(Long sceneCooperationId) {

        List<SceneResourceCooperation> sceneResourceCooperationList
            = sceneResourceCooperationService.list(
            new LambdaQueryWrapper<SceneResourceCooperation>()
                .eq(SceneResourceCooperation::getSceneCooperationId, sceneCooperationId));

        List<Long> sceneResourceIdList = null;
        if(CollUtil.isNotEmpty(sceneResourceCooperationList)){
            sceneResourceIdList = sceneResourceCooperationList.stream()
                .map(sceneResourceCooperation -> sceneResourceCooperation.getSceneResourceId())
                .collect(Collectors.toList());
        }
        if(CollUtil.isEmpty(sceneResourceIdList)){
            return null;
        }

        return this.listByIds(sceneResourceIdList);
    }
}