diff --git a/src/test/java/com/ycwl/basic/task/FaceCleanerTest.java b/src/test/java/com/ycwl/basic/task/FaceCleanerTest.java index f0e3d24..d2a6406 100644 --- a/src/test/java/com/ycwl/basic/task/FaceCleanerTest.java +++ b/src/test/java/com/ycwl/basic/task/FaceCleanerTest.java @@ -1,13 +1,20 @@ package com.ycwl.basic.task; +import cn.hutool.core.date.DateUtil; import com.aliyuncs.DefaultAcsClient; import com.aliyuncs.IAcsClient; import com.aliyuncs.profile.DefaultProfile; +import com.ycwl.basic.facebody.adapter.IFaceBodyAdapter; import com.ycwl.basic.mapper.FaceMapper; import com.ycwl.basic.mapper.FaceSampleMapper; import com.ycwl.basic.mapper.ScenicMapper; +import com.ycwl.basic.model.pc.faceSample.entity.FaceSampleEntity; +import com.ycwl.basic.model.pc.scenic.entity.ScenicConfigEntity; +import com.ycwl.basic.model.pc.scenic.entity.ScenicEntity; import com.ycwl.basic.model.pc.scenic.req.ScenicReqQuery; import com.ycwl.basic.model.pc.scenic.resp.ScenicRespVO; +import com.ycwl.basic.repository.ScenicRepository; +import com.ycwl.basic.service.pc.ScenicService; import com.ycwl.basic.service.task.TaskFaceService; import lombok.extern.slf4j.Slf4j; import org.junit.Test; @@ -17,6 +24,7 @@ import org.springframework.boot.test.context.SpringBootTest; import org.springframework.test.context.junit4.SpringRunner; import java.util.Arrays; +import java.util.Date; import java.util.List; import java.util.concurrent.LinkedBlockingQueue; import java.util.concurrent.ThreadPoolExecutor; @@ -36,10 +44,37 @@ public class FaceCleanerTest { private ScenicMapper scenicMapper; @Autowired private FaceCleaner faceCleaner; + @Autowired + private ScenicService scenicService; + @Autowired + private ScenicRepository scenicRepository; + + @Test + public void testA() { + ScenicEntity scenic = scenicMapper.get(3980001650692722688L); + log.info("当前景区{},开始删除人脸样本", scenic.getId()); + IFaceBodyAdapter adapter = scenicService.getScenicFaceBodyAdapter(scenic.getId()); + int sampleStoreDay = 1; + Date sampleEndDate = DateUtil.offsetDay(DateUtil.beginOfDay(new Date()), -sampleStoreDay); + List faceSampleList = faceSampleMapper.listEntityBeforeDate(scenic.getId(), sampleEndDate); + if (faceSampleList.isEmpty()) { + log.info("当前景区{},人脸样本为空", scenic.getId()); + return; + } + faceSampleList.forEach(faceSample -> { + boolean success = adapter.deleteFace(String.valueOf(scenic.getId()), faceSample.getId().toString()); + if (success) { + log.info("当前景区{},人脸样本ID{},删除成功", scenic.getId(), faceSample.getId()); + faceSampleMapper.deleteById(faceSample.getId()); + } else { + log.info("当前景区{},人脸样本ID{},删除失败", scenic.getId(), faceSample.getId()); + } + }); + } @Test public void testC() { - faceCleaner.cleanSourceOss(); + faceCleaner.deleteExpireSample(); } @Test