long-article-manage/long-article-server/src/main/java/com/tzld/piaoquan/longarticle/job/NewMatchVideoJob.java

package com.tzld.piaoquan.longarticle.job;

import com.tzld.piaoquan.longarticle.common.enums.NewContentStatusEnum;
import com.tzld.piaoquan.longarticle.common.enums.NewMatchResultStatusEnum;
import com.tzld.piaoquan.longarticle.dao.mapper.longarticle.NewMatchVideoMapper;
import com.tzld.piaoquan.longarticle.model.dto.AuditContentRequest;
import com.tzld.piaoquan.longarticle.model.po.longarticle.NewMatchVideo;
import com.tzld.piaoquan.longarticle.model.po.longarticle.NewMatchVideoExample;
import com.tzld.piaoquan.longarticle.model.vo.MatchContentItem;
import com.tzld.piaoquan.longarticle.model.vo.MatchContentStatusParam;
import com.tzld.piaoquan.longarticle.service.local.NewMatchVideoService;
import com.tzld.piaoquan.longarticle.service.remote.AigcService;
import com.tzld.piaoquan.longarticle.service.remote.MatchService;
import com.tzld.piaoquan.longarticle.utils.LarkRobotUtil;
import com.xxl.job.core.biz.model.ReturnT;
import com.xxl.job.core.handler.annotation.XxlJob;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.util.List;

@Slf4j
@Component
public class NewMatchVideoJob {

    @Autowired
    private NewMatchVideoService newMatchVideoService;

    @Autowired
    private AigcService aigcService;

    @Autowired
    private NewMatchVideoMapper newMatchVideoMapper;

    @Autowired
    private MatchService matchService;

    @XxlJob("matchContentJob")
    public ReturnT<String> matchContentJob(String param) {
        List<MatchContentItem> matchContentItemList = aigcService.getMatchContentItemList();
        if (CollectionUtils.isEmpty(matchContentItemList)) {
            return ReturnT.SUCCESS;
        }
        for (MatchContentItem matchContentItem : matchContentItemList) {
            try {
                NewMatchVideoExample example = new NewMatchVideoExample();
                example.createCriteria().andContentIdEqualTo(matchContentItem.getProduceContentId());
                List<NewMatchVideo> newMatchVideos = newMatchVideoMapper.selectByExample(example);
                if (CollectionUtils.isEmpty(newMatchVideos)) {
                    newMatchVideoService.addMatchVideo(matchContentItem);
                } else {
                    NewMatchVideo newMatchVideo = newMatchVideos.get(0);
                    if (NewContentStatusEnum.isSuccess(newMatchVideo.getContentStatus())) {
                        MatchContentStatusParam matchContentStatusParam = new MatchContentStatusParam();
                        matchContentStatusParam.setProduceContentId(newMatchVideo.getContentId());
                        matchContentStatusParam.setStatus(NewMatchResultStatusEnum.SUCCESS.getStatusCode());
                        boolean res = aigcService.updateMatchContentStatus(matchContentStatusParam);
                        if (res) {
                            AuditContentRequest auditContentRequest = new AuditContentRequest();
                            auditContentRequest.setContentId(newMatchVideo.getContentId());
                            auditContentRequest.setFlowPoolLevel(newMatchVideo.getFlowPoolLevel());
                            matchService.addAudit(auditContentRequest);
                        }

                    }
                    if (NewContentStatusEnum.isFail(newMatchVideo.getContentStatus())) {
                        MatchContentStatusParam matchContentStatusParam = new MatchContentStatusParam();
                        matchContentStatusParam.setProduceContentId(newMatchVideo.getContentId());
                        matchContentStatusParam.setStatus(NewMatchResultStatusEnum.FAIL.getStatusCode());
                        aigcService.updateMatchContentStatus(matchContentStatusParam);
                    }
                }
            } catch (Exception e) {
                LarkRobotUtil.sendMessage("matchContentJob异常，contentId:"+matchContentItem.getProduceContentId()+", 请及时查看，<at user_id=\"g6732afb\">王云鹏</at>");
                log.error("matchContentJob error:{}", e.getMessage());
            }
        }
        return ReturnT.SUCCESS;
    }

    @XxlJob("newKimiContentJob")
    public ReturnT<String> kimiContentJob(String param) {
        try {
            newMatchVideoService.kimiContent();
        } catch (Exception e) {
            LarkRobotUtil.sendMessage("kimiContentJob异常，请及时查看，<at user_id=\"g6732afb\">王云鹏</at>");
            log.error("kimiContentJob error:{}", e.getMessage());
        }
        return ReturnT.SUCCESS;
    }

    @XxlJob("newMatchCrawlerVideoJob")
    public ReturnT<String> matchCrawlerVideoJob(String param) {
        try {
            newMatchVideoService.matchCrawlerVideo();
        } catch (Exception e) {
            LarkRobotUtil.sendMessage("newMatchCrawlerVideoJob异常，请及时查看，<at user_id=\"g6732afb\">王云鹏</at>");
            log.error("newMatchCrawlerVideoJob error:{}", e.getMessage());
        }
        return ReturnT.SUCCESS;
    }

    @XxlJob("newUploadCrawlerVideoJob")
    public ReturnT<String> uploadCrawlerVideoJob(String param) {
        try {
            newMatchVideoService.uploadCrawlerVideo();
        } catch (Exception e) {
            LarkRobotUtil.sendMessage("newUploadCrawlerVideoJob异常，请及时查看，<at user_id=\"g6732afb\">王云鹏</at>");
            log.error("newUploadCrawlerVideoJob error:{}", e.getMessage());
        }
        return ReturnT.SUCCESS;
    }

    @XxlJob("vectorMatchVideoJob")
    public ReturnT<String> vectorMatchVideoJob(String param) {
        try {
            newMatchVideoService.vectorMatchVideo();
        } catch (Exception e) {
            LarkRobotUtil.sendMessage("vectorMatchVideoJob异常，请及时查看，<at user_id=\"g6732afb\">王云鹏</at>");
            log.error("vectorMatchVideoJob error:{}", e.getMessage());
        }
        return ReturnT.SUCCESS;
    }


}