FrameTour-BE/src/main/java/com/ycwl/basic/biz/TemplateBiz.java

package com.ycwl.basic.biz;

import com.ycwl.basic.mapper.SourceMapper;
import com.ycwl.basic.model.pc.face.entity.FaceEntity;
import com.ycwl.basic.model.pc.face.resp.FaceRespVO;
import com.ycwl.basic.model.pc.faceSample.entity.FaceSampleEntity;
import com.ycwl.basic.model.pc.faceSample.resp.FaceSampleRespVO;
import com.ycwl.basic.model.pc.source.entity.SourceEntity;
import com.ycwl.basic.model.pc.template.entity.TemplateConfigEntity;
import com.ycwl.basic.repository.FaceRepository;
import com.ycwl.basic.repository.SourceRepository;
import com.ycwl.basic.repository.TemplateRepository;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;

@Slf4j
@Component
public class TemplateBiz {
    @Autowired
    private TemplateRepository templateRepository;
    @Autowired
    private FaceRepository faceRepository;
    @Autowired
    private SourceMapper sourceMapper;
    @Autowired
    private SourceRepository sourceRepository;

    public boolean determineTemplateCanGenerate(Long templateId, Long faceId) {
        return determineTemplateCanGenerate(templateId, faceId, true);
    }

    public boolean determineTemplateCanGenerate(Long templateId, Long faceId, boolean scanSource) {
        List<String> placeholderList = templateRepository.getTemplatePlaceholder(templateId);
        TemplateConfigEntity templateConfig = templateRepository.getTemplateConfig(templateId);
        int minimalPlaceholderFill = 1;
        if (null != templateConfig) {
            if (null != templateConfig.getMinimalPlaceholderFill()) {
                minimalPlaceholderFill = templateConfig.getMinimalPlaceholderFill();
            }
        }
        if (minimalPlaceholderFill <= 0) {
            return true;
        }
        FaceEntity face = faceRepository.getFace(faceId);
        long count;
        if (scanSource) {
            List<SourceEntity> sourceEntities = sourceMapper.listVideoByScenicFaceRelation(face.getScenicId(), faceId);
            if (sourceEntities == null || sourceEntities.isEmpty()) {
                return false;
            }
            count = sourceEntities.stream()
                    .map(SourceEntity::getDeviceId)
                    .filter(Objects::nonNull) // 添加对 null 的检查
                    .distinct()
                    .filter(deviceId -> placeholderList.contains(deviceId.toString()))
                    .count();
        } else {
            List<FaceSampleEntity> faceSampleList = faceRepository.getFaceSampleList(faceId);
            if (faceSampleList == null || faceSampleList.isEmpty()) {
                return false;
            }
            count = faceSampleList.stream()
                    .map(FaceSampleEntity::getDeviceId)
                    .filter(Objects::nonNull) // 添加对 null 的检查
                    .distinct()
                    .filter(deviceId -> placeholderList.contains(deviceId.toString()))
                    .count();
        }
        return count >= minimalPlaceholderFill;
    }

    public boolean determineTemplateCanAutoGenerate(Long templateId, Long faceId) {
        return determineTemplateCanAutoGenerate(templateId, faceId, true);
    }

    public boolean determineTemplateCanAutoGenerate(Long templateId, Long faceId, boolean scanSource) {
        List<String> placeholderList = templateRepository.getTemplatePlaceholder(templateId);
        TemplateConfigEntity templateConfig = templateRepository.getTemplateConfig(templateId);
        Integer minimalPlaceholderFill = null;
        if (null != templateConfig) {
            if (null != templateConfig.getAutomaticPlaceholderFill()) {
                minimalPlaceholderFill = templateConfig.getAutomaticPlaceholderFill();
            }
        }
        if (minimalPlaceholderFill == null) {
            // 未开启
            log.info("模板：{}，未配置最小自动生成功能，默认不生成", templateId);
            return false;
        }
        if (minimalPlaceholderFill <= 0) {
            return true;
        }
        FaceEntity face = faceRepository.getFace(faceId);
        List<FaceSampleEntity> faceSampleList = faceRepository.getFaceSampleList(faceId);
        if (faceSampleList.isEmpty()) {
            return false;
        }
        long count;
        if (scanSource) {
            List<SourceEntity> sourceEntities = sourceMapper.listVideoByScenicFaceRelation(face.getScenicId(), faceId);
            count = sourceEntities.stream()
                    .map(SourceEntity::getDeviceId)
                    .filter(Objects::nonNull) // 添加对 null 的检查
                    .distinct()
                    .filter(deviceId -> placeholderList.contains(deviceId.toString()))
                    .count();
        } else {
            count = faceSampleList.stream()
                    .map(FaceSampleEntity::getDeviceId)
                    .distinct()
                    .filter(deviceId -> placeholderList.contains(deviceId.toString()))
                    .count();
        }
        return count >= minimalPlaceholderFill;
    }

}