bug修复

This commit is contained in:
Jerry Yan 2025-02-05 11:28:50 +08:00
parent 0b861f0e21
commit 7892c0f5cc
21 changed files with 324 additions and 95 deletions

View File

@ -233,6 +233,8 @@ public class ViidController {
} }
if (shotTime == null) { if (shotTime == null) {
shotTime = new Date(); shotTime = new Date();
} else if (Math.abs(shotTime.getTime() - System.currentTimeMillis()) > + 24 * 60 * 60 * 1000) {
shotTime = new Date();
} }
Long scenicId = device.getScenicId(); Long scenicId = device.getScenicId();
if (scenicId == null) { if (scenicId == null) {
@ -269,8 +271,10 @@ public class ViidController {
String url = adapter.uploadFile(file, "user-face", UUID.randomUUID() + "." + ext); String url = adapter.uploadFile(file, "user-face", UUID.randomUUID() + "." + ext);
faceSample.setFaceUrl(url); faceSample.setFaceUrl(url);
faceSampleMapper.add(faceSample); faceSampleMapper.add(faceSample);
DynamicTaskGenerator.addTask(faceSample.getId()); new Thread(() -> {
taskFaceService.addFaceSample(faceSample.getId()); taskFaceService.addFaceSample(faceSample.getId());
DynamicTaskGenerator.addTask(faceSample.getId());
}).start();
for (SubImageInfoObject _subImage : type14ImageList) { for (SubImageInfoObject _subImage : type14ImageList) {
facePosition.setImgHeight(_subImage.getHeight()); facePosition.setImgHeight(_subImage.getHeight());
facePosition.setImgWidth(_subImage.getWidth()); facePosition.setImgWidth(_subImage.getWidth());

View File

@ -18,6 +18,7 @@ import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController; import org.springframework.web.bind.annotation.RestController;
import java.util.Date;
import java.util.List; import java.util.List;
@Slf4j @Slf4j

View File

@ -62,6 +62,12 @@ public class CustomExceptionHandle {
return ApiResponse.buildResponse(bizException.getCode(), bizException.getMsg()); return ApiResponse.buildResponse(bizException.getCode(), bizException.getMsg());
} }
@ExceptionHandler(value =IOException.class)
public ApiResponse<String> handle(IOException e) {
LOGGER.error("系统异常 -> {}", e.getMessage(), e);
return ApiResponse.buildResult(BizCodeEnum.SERVER_UNKONWN_ERROR);
}
/** /**
* 异常统一返回处理 * 异常统一返回处理
*/ */

View File

@ -6,6 +6,7 @@ import com.ycwl.basic.model.pc.faceSample.resp.FaceSampleRespVO;
import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param; import org.apache.ibatis.annotations.Param;
import java.util.Date;
import java.util.List; import java.util.List;
/** /**
@ -25,4 +26,5 @@ public interface FaceSampleMapper {
List<FaceSampleEntity> listByIds(List<Long> list); List<FaceSampleEntity> listByIds(List<Long> list);
FaceSampleEntity getEntity(Long faceSampleId); FaceSampleEntity getEntity(Long faceSampleId);
List<FaceSampleEntity> listEntity(Long scenicId, Date endDate);
} }

View File

@ -56,4 +56,6 @@ public class ScenicConfigEntity {
private Integer disableSourceImage; private Integer disableSourceImage;
private Integer templateNewVideoType; private Integer templateNewVideoType;
private Integer antiScreenRecordType; private Integer antiScreenRecordType;
private Integer videoSourceStoreDay;
private Integer imageSourceStoreDay;
} }

View File

@ -0,0 +1,32 @@
package com.ycwl.basic.ratelimiter;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
public class FixedRateLimiter {
private final Semaphore semaphore = new Semaphore(1);
private final ScheduledExecutorService scheduler = new ScheduledThreadPoolExecutor(1);
public FixedRateLimiter(int rate, TimeUnit timeUnit) {
// 启动一个线程每0.5秒释放一个许可
scheduler.scheduleAtFixedRate(() -> {
synchronized (semaphore) {
if (semaphore.availablePermits() < 1) {
semaphore.release(1);
}
}
}, rate, rate, timeUnit);
}
public void acquire() throws InterruptedException {
synchronized (semaphore) {
semaphore.acquire();
}
}
public void shutdown() {
scheduler.shutdown();
}
}

View File

@ -12,14 +12,15 @@ public class SlidingWindowRateLimiter {
public SlidingWindowRateLimiter(int maxRequestsPerSecond) { public SlidingWindowRateLimiter(int maxRequestsPerSecond) {
this.semaphore = new Semaphore(maxRequestsPerSecond); this.semaphore = new Semaphore(maxRequestsPerSecond);
int rate = 1000000 / maxRequestsPerSecond;
scheduler.scheduleAtFixedRate(() -> { scheduler.scheduleAtFixedRate(() -> {
if (semaphore.availablePermits() < maxRequestsPerSecond) { if (semaphore.availablePermits() < maxRequestsPerSecond) {
semaphore.release(1); semaphore.release(1);
} }
}, 0, (1000 / maxRequestsPerSecond), TimeUnit.MILLISECONDS); }, rate, rate, TimeUnit.MICROSECONDS);
} }
public void aquire() throws InterruptedException { public void acquire() throws InterruptedException {
semaphore.acquire(); semaphore.acquire();
} }

View File

@ -73,6 +73,9 @@ public class TemplateRepository {
return JSONObject.parseObject(redisTemplate.opsForValue().get(String.format(TEMPLATE_CACHE_KEY, templateId)), TemplateRespVO.class); return JSONObject.parseObject(redisTemplate.opsForValue().get(String.format(TEMPLATE_CACHE_KEY, templateId)), TemplateRespVO.class);
} }
TemplateRespVO template = templateMapper.getById(templateId); TemplateRespVO template = templateMapper.getById(templateId);
if (template == null) {
return null;
}
if (null == template.getPid() || template.getPid() == 0) { if (null == template.getPid() || template.getPid() == 0) {
template.setChildren(templateMapper.getByPid(templateId)); template.setChildren(templateMapper.getByPid(templateId));
redisTemplate.opsForValue().set(String.format(TEMPLATE_CACHE_KEY, templateId), JSONObject.toJSONString(template)); redisTemplate.opsForValue().set(String.format(TEMPLATE_CACHE_KEY, templateId), JSONObject.toJSONString(template));

View File

@ -358,14 +358,17 @@ public class AppStatisticsServiceImpl implements AppStatisticsService {
//上一个周期的支付订单金额 //上一个周期的支付订单金额
vo.setPreviousOrderAmount(orderAmountDf.format(orderAmount)); vo.setPreviousOrderAmount(orderAmountDf.format(orderAmount));
// 计算预览_支付转化率 // 计算预览_支付转化率
if(pay==0){ if(preview==0){
vo.setPreviousPreviewPay("0.00"); vo.setPreviousPreviewPay("0.00");
}else {
BigDecimal previewPay = new BigDecimal(pay).divide(new BigDecimal(preview), 4, RoundingMode.HALF_UP).multiply(new BigDecimal(100));
vo.setPreviousPreviewPay(df.format(previewPay));
}
if(scanCode==0){
vo.setPreviousScanCodePay("0.00"); vo.setPreviousScanCodePay("0.00");
}else { }else {
BigDecimal previewPay = new BigDecimal(preview).divide(new BigDecimal(pay), 4, RoundingMode.HALF_UP).multiply(new BigDecimal(100)); BigDecimal scanCodePay = new BigDecimal(pay).divide(new BigDecimal(scanCode), 4, RoundingMode.HALF_UP).multiply(new BigDecimal(100));
vo.setNowPreviewPay(df.format(previewPay)); vo.setPreviousScanCodePay(df.format(scanCodePay));
BigDecimal scanCodePay = new BigDecimal(scanCode).divide(new BigDecimal(pay), 4, RoundingMode.HALF_UP).multiply(new BigDecimal(100));
vo.setNowScanCodePay(df.format(scanCodePay));
} }
} }
} }

View File

@ -25,6 +25,8 @@ public interface TaskService {
void taskStart(Long taskId, WorkerAuthReqVo req); void taskStart(Long taskId, WorkerAuthReqVo req);
void forceCreateTaskByFaceIdAndTempalteId(Long faceId, Long templateId);
void autoCreateTaskByFaceId(Long id); void autoCreateTaskByFaceId(Long id);
Date getTaskShotDate(Long taskId); Date getTaskShotDate(Long taskId);

View File

@ -3,6 +3,7 @@ package com.ycwl.basic.service.task.impl;
import cn.hutool.core.date.DateUtil; import cn.hutool.core.date.DateUtil;
import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSON;
import com.aliyuncs.exceptions.ClientException; import com.aliyuncs.exceptions.ClientException;
import com.aliyuncs.exceptions.ServerException;
import com.aliyuncs.facebody.model.v20191230.AddFaceEntityRequest; import com.aliyuncs.facebody.model.v20191230.AddFaceEntityRequest;
import com.aliyuncs.facebody.model.v20191230.AddFaceRequest; import com.aliyuncs.facebody.model.v20191230.AddFaceRequest;
import com.aliyuncs.facebody.model.v20191230.AddFaceResponse; import com.aliyuncs.facebody.model.v20191230.AddFaceResponse;
@ -15,6 +16,7 @@ import com.aliyuncs.facebody.model.v20191230.ListFaceEntitiesRequest;
import com.aliyuncs.facebody.model.v20191230.ListFaceEntitiesResponse; import com.aliyuncs.facebody.model.v20191230.ListFaceEntitiesResponse;
import com.aliyuncs.facebody.model.v20191230.SearchFaceRequest; import com.aliyuncs.facebody.model.v20191230.SearchFaceRequest;
import com.aliyuncs.facebody.model.v20191230.SearchFaceResponse; import com.aliyuncs.facebody.model.v20191230.SearchFaceResponse;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ycwl.basic.biz.OrderBiz; import com.ycwl.basic.biz.OrderBiz;
import com.ycwl.basic.config.FaceDetectConfig; import com.ycwl.basic.config.FaceDetectConfig;
import com.ycwl.basic.constant.FaceConstant; import com.ycwl.basic.constant.FaceConstant;
@ -33,12 +35,16 @@ import com.ycwl.basic.model.pc.faceSample.entity.FaceSampleEntity;
import com.ycwl.basic.model.pc.faceSample.req.FaceSampleReqQuery; import com.ycwl.basic.model.pc.faceSample.req.FaceSampleReqQuery;
import com.ycwl.basic.model.pc.faceSample.resp.FaceSampleRespVO; import com.ycwl.basic.model.pc.faceSample.resp.FaceSampleRespVO;
import com.ycwl.basic.model.pc.scenic.entity.ScenicConfigEntity; import com.ycwl.basic.model.pc.scenic.entity.ScenicConfigEntity;
import com.ycwl.basic.model.pc.scenic.req.ScenicReqQuery;
import com.ycwl.basic.model.pc.scenic.resp.ScenicRespVO;
import com.ycwl.basic.model.pc.source.entity.MemberSourceEntity; import com.ycwl.basic.model.pc.source.entity.MemberSourceEntity;
import com.ycwl.basic.model.pc.source.entity.SourceEntity; import com.ycwl.basic.model.pc.source.entity.SourceEntity;
import com.ycwl.basic.model.task.resp.AddFaceSampleRespVo; import com.ycwl.basic.model.task.resp.AddFaceSampleRespVo;
import com.ycwl.basic.model.task.resp.SearchFaceRespVo; import com.ycwl.basic.model.task.resp.SearchFaceRespVo;
import com.ycwl.basic.ratelimiter.FixedRateLimiter;
import com.ycwl.basic.ratelimiter.SlidingWindowRateLimiter; import com.ycwl.basic.ratelimiter.SlidingWindowRateLimiter;
import com.ycwl.basic.repository.FaceRepository; import com.ycwl.basic.repository.FaceRepository;
import com.ycwl.basic.repository.ScenicRepository;
import com.ycwl.basic.service.task.TaskFaceService; import com.ycwl.basic.service.task.TaskFaceService;
import com.ycwl.basic.storage.StorageFactory; import com.ycwl.basic.storage.StorageFactory;
import com.ycwl.basic.storage.adapters.IStorageAdapter; import com.ycwl.basic.storage.adapters.IStorageAdapter;
@ -49,6 +55,7 @@ import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate; import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import com.aliyuncs.DefaultAcsClient; import com.aliyuncs.DefaultAcsClient;
import com.aliyuncs.IAcsClient; import com.aliyuncs.IAcsClient;
@ -63,6 +70,7 @@ import java.util.Date;
import java.util.List; import java.util.List;
import java.util.Random; import java.util.Random;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@Slf4j @Slf4j
@ -87,11 +95,14 @@ public class TaskFaceServiceImpl implements TaskFaceService {
@Autowired @Autowired
private OrderBiz orderBiz; private OrderBiz orderBiz;
// 阿里云人脸检索限制qps=2 // 阿里云人脸检索限制qps=2
private final SlidingWindowRateLimiter addEntityLimiter = new SlidingWindowRateLimiter(1); private final FixedRateLimiter addEntityLimiter = new FixedRateLimiter(600, TimeUnit.MILLISECONDS);
// 阿里云人脸检索限制qps=5 // 阿里云人脸检索限制qps=5
private final SlidingWindowRateLimiter searchFaceLimiter = new SlidingWindowRateLimiter(4); private final FixedRateLimiter searchFaceLimiter = new FixedRateLimiter(200, TimeUnit.MILLISECONDS);
private final SlidingWindowRateLimiter deleteDbLimiter = new SlidingWindowRateLimiter(1); private final FixedRateLimiter deleteDbLimiter = new FixedRateLimiter(600, TimeUnit.MILLISECONDS);
private final SlidingWindowRateLimiter deleteEntityLimiter = new SlidingWindowRateLimiter(1); private final FixedRateLimiter deleteEntityLimiter = new FixedRateLimiter(600, TimeUnit.MILLISECONDS);
@Autowired
private ScenicRepository scenicRepository;
private IAcsClient getClient() { private IAcsClient getClient() {
DefaultProfile profile = DefaultProfile.getProfile( DefaultProfile profile = DefaultProfile.getProfile(
@ -168,7 +179,7 @@ public class TaskFaceServiceImpl implements TaskFaceService {
// request.setQualityScoreThreshold(60f); // request.setQualityScoreThreshold(60f);
FaceDetectLog log = FaceDetectLog.quickCreate("预留字段", request); FaceDetectLog log = FaceDetectLog.quickCreate("预留字段", request);
try { try {
searchFaceLimiter.aquire(); searchFaceLimiter.acquire();
} catch (InterruptedException ignored) { } catch (InterruptedException ignored) {
} }
try { try {
@ -243,7 +254,7 @@ public class TaskFaceServiceImpl implements TaskFaceService {
request.setEntityId(entityId); request.setEntityId(entityId);
IAcsClient client = getClient(); IAcsClient client = getClient();
try { try {
addEntityLimiter.aquire(); addEntityLimiter.acquire();
} catch (InterruptedException ignored) { } catch (InterruptedException ignored) {
} }
try { try {
@ -270,38 +281,67 @@ public class TaskFaceServiceImpl implements TaskFaceService {
@Override @Override
public void batchDeleteExpiredFace(Long scenicId) { public void batchDeleteExpiredFace(Long scenicId) {
FaceSampleReqQuery query = new FaceSampleReqQuery(); log.info("当前景区{},开始删除人脸样本", scenicId);
query.setDeviceId(scenicId); ScenicConfigEntity scenicConfig = scenicRepository.getScenicConfig(scenicId);
faceSampleMapper.list(query);
ScenicConfigEntity scenicConfig = scenicMapper.getConfig(scenicId);
if (scenicConfig == null) {
return;
}
Integer sampleStoreDay = scenicConfig.getSampleStoreDay(); Integer sampleStoreDay = scenicConfig.getSampleStoreDay();
if (sampleStoreDay == null) { if (sampleStoreDay == null) {
sampleStoreDay = 3; log.info("当前景区{}人脸样本保存天数未设置默认7天", scenicId);
sampleStoreDay = 7;
} }
Date thatDay = DateUtil.offsetDay(new Date(), -sampleStoreDay); Date endDate = DateUtils.addDateDays(new Date(), -(sampleStoreDay + 1));
Date dayEnd = DateUtil.endOfDay(thatDay); List<FaceSampleEntity> faceSampleList = faceSampleMapper.listEntity(scenicId, endDate);
query.setEndTime(dayEnd); if (faceSampleList.isEmpty()) {
IAcsClient client = getClient(); log.info("当前景区{},人脸样本为空", scenicId);
faceSampleMapper.list(query).forEach(faceSampleEntity -> { return;
String entityId = generateEntityId(faceSampleEntity); }
DeleteFaceEntityRequest request = new DeleteFaceEntityRequest(); faceSampleList.forEach(faceSample -> {
request.setDbName(scenicId.toString()); boolean success = deleteFaceSample(String.valueOf(scenicId), generateEntityId(faceSample));
request.setEntityId(entityId); if (success) {
try { log.info("当前景区{}人脸样本ID{},删除成功", scenicId, faceSample.getId());
deleteEntityLimiter.aquire(); faceSampleMapper.deleteById(faceSample.getId());
} catch (InterruptedException ignored) { } else {
} log.info("当前景区{}人脸样本ID{},删除失败", scenicId, faceSample.getId());
try {
client.getAcsResponse(request);
} catch (ClientException e) {
return;
} finally {
faceSampleMapper.deleteById(faceSampleEntity.getId());
} }
}); });
ListFaceEntitiesRequest listFaceEntitiesRequest = new ListFaceEntitiesRequest();
listFaceEntitiesRequest.setDbName(String.valueOf(scenicId));
listFaceEntitiesRequest.setOrder("asc");
try {
IAcsClient client = getClient();
while (true) {
AtomicBoolean flag = new AtomicBoolean(false);
ListFaceEntitiesResponse listFaceEntitiesResponse = client.getAcsResponse(listFaceEntitiesRequest);
if (listFaceEntitiesResponse == null || listFaceEntitiesResponse.getData() == null || listFaceEntitiesResponse.getData().getEntities() == null || listFaceEntitiesResponse.getData().getEntities().isEmpty()) {
break;
}
listFaceEntitiesResponse.getData().getEntities().forEach(entity -> {
String entityId = entity.getEntityId();
String[] split = entityId.split("_");
if (split.length != 2) {
return;
}
String deviceId = split[0];
if (StringUtils.isBlank(deviceId)) {
return;
}
String dateString = split[1];
if (StringUtils.isBlank(dateString)) {
return;
}
if (DateUtils.parse(dateString, DATE_FORMAT).before(endDate)) {
flag.set(true);
log.info("当前景区{},开始删除人脸样本:{}", scenicId, entity.getEntityId());
deleteFaceSample(String.valueOf(scenicId), entity.getEntityId());
} else {
log.info("当前景区{},人脸样本:{}未过期", scenicId, entity.getEntityId());
}
});
if (!flag.get()) {
break;
}
}
} catch (Exception ignored) {
}
} }
@Override @Override
@ -321,7 +361,7 @@ public class TaskFaceServiceImpl implements TaskFaceService {
deleteFaceEntityRequest.setDbName(entity.getDbName()); deleteFaceEntityRequest.setDbName(entity.getDbName());
deleteFaceEntityRequest.setEntityId(entity.getEntityId()); deleteFaceEntityRequest.setEntityId(entity.getEntityId());
try { try {
deleteEntityLimiter.aquire(); deleteEntityLimiter.acquire();
} catch (InterruptedException ignored) { } catch (InterruptedException ignored) {
} }
try { try {
@ -334,7 +374,7 @@ public class TaskFaceServiceImpl implements TaskFaceService {
DeleteFaceDbRequest deleteFaceDbRequest = new DeleteFaceDbRequest(); DeleteFaceDbRequest deleteFaceDbRequest = new DeleteFaceDbRequest();
deleteFaceDbRequest.setName(dbName); deleteFaceDbRequest.setName(dbName);
try { try {
deleteDbLimiter.aquire(); deleteDbLimiter.acquire();
} catch (InterruptedException ignored) { } catch (InterruptedException ignored) {
} }
client.getAcsResponse(deleteFaceDbRequest); client.getAcsResponse(deleteFaceDbRequest);
@ -426,10 +466,12 @@ public class TaskFaceServiceImpl implements TaskFaceService {
deleteFaceEntityRequest.setDbName(dbName); deleteFaceEntityRequest.setDbName(dbName);
deleteFaceEntityRequest.setEntityId(entityId); deleteFaceEntityRequest.setEntityId(entityId);
try { try {
deleteEntityLimiter.acquire();
log.info("删除{}人脸实体:{}", dbName, entityId);
IAcsClient client = getClient(); IAcsClient client = getClient();
client.getAcsResponse(deleteFaceEntityRequest); client.getAcsResponse(deleteFaceEntityRequest);
return true; return true;
} catch (ClientException e) { } catch (Exception e) {
log.error("删除人脸样本失败!", e); log.error("删除人脸样本失败!", e);
return false; return false;
} }

View File

@ -16,7 +16,6 @@ import com.ycwl.basic.mapper.TaskMapper;
import com.ycwl.basic.mapper.TemplateMapper; import com.ycwl.basic.mapper.TemplateMapper;
import com.ycwl.basic.mapper.VideoMapper; import com.ycwl.basic.mapper.VideoMapper;
import com.ycwl.basic.model.mobile.order.IsBuyRespVO; import com.ycwl.basic.model.mobile.order.IsBuyRespVO;
import com.ycwl.basic.model.mobile.order.PriceObj;
import com.ycwl.basic.model.pc.face.entity.FaceEntity; import com.ycwl.basic.model.pc.face.entity.FaceEntity;
import com.ycwl.basic.model.pc.face.resp.FaceRespVO; 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.entity.FaceSampleEntity;
@ -26,11 +25,9 @@ import com.ycwl.basic.model.pc.renderWorker.entity.RenderWorkerEntity;
import com.ycwl.basic.model.pc.scenic.entity.ScenicConfigEntity; 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.entity.ScenicEntity;
import com.ycwl.basic.model.pc.source.entity.SourceEntity; import com.ycwl.basic.model.pc.source.entity.SourceEntity;
import com.ycwl.basic.model.pc.source.resp.SourceRespVO;
import com.ycwl.basic.model.pc.task.entity.TaskEntity; import com.ycwl.basic.model.pc.task.entity.TaskEntity;
import com.ycwl.basic.model.pc.task.req.TaskReqQuery; import com.ycwl.basic.model.pc.task.req.TaskReqQuery;
import com.ycwl.basic.model.pc.task.resp.TaskRespVO; import com.ycwl.basic.model.pc.task.resp.TaskRespVO;
import com.ycwl.basic.model.pc.template.entity.TemplateConfigEntity;
import com.ycwl.basic.model.pc.template.resp.TemplateRespVO; import com.ycwl.basic.model.pc.template.resp.TemplateRespVO;
import com.ycwl.basic.model.pc.video.entity.MemberVideoEntity; import com.ycwl.basic.model.pc.video.entity.MemberVideoEntity;
import com.ycwl.basic.model.pc.video.entity.VideoEntity; import com.ycwl.basic.model.pc.video.entity.VideoEntity;
@ -69,7 +66,6 @@ import java.util.Date;
import java.util.HashMap; import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Objects;
import java.util.Optional; import java.util.Optional;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.ReentrantLock; import java.util.concurrent.locks.ReentrantLock;
@ -161,6 +157,9 @@ public class TaskTaskServiceImpl implements TaskService {
req.getTemplateList().forEach(template -> { req.getTemplateList().forEach(template -> {
if (StringUtils.isNumeric(template.getId())) { if (StringUtils.isNumeric(template.getId())) {
TemplateRespVO dbTemplate = templateRepository.getTemplate(Long.parseLong(template.getId())); TemplateRespVO dbTemplate = templateRepository.getTemplate(Long.parseLong(template.getId()));
if (dbTemplate == null) {
return;
}
if (!dbTemplate.getUpdateTime().equals(template.getUpdateTime())) { if (!dbTemplate.getUpdateTime().equals(template.getUpdateTime())) {
updTemplateList.add(dbTemplate); updTemplateList.add(dbTemplate);
} }
@ -171,13 +170,17 @@ public class TaskTaskServiceImpl implements TaskService {
} }
try { try {
if (lock.tryLock(2, TimeUnit.SECONDS)) { if (lock.tryLock(2, TimeUnit.SECONDS)) {
List<TaskRespVO> taskList = taskMapper.selectNotRunning(); try {
resp.setTasks(taskList); List<TaskRespVO> taskList = taskMapper.selectNotRunning();
resp.setTemplates(updTemplateList); resp.setTasks(taskList);
taskList.forEach(task -> { resp.setTemplates(updTemplateList);
taskMapper.assignToWorker(task.getId(), worker.getId()); taskList.forEach(task -> {
videoTaskRepository.clearTaskCache(task.getId()); taskMapper.assignToWorker(task.getId(), worker.getId());
}); videoTaskRepository.clearTaskCache(task.getId());
});
} finally {
lock.unlock();
}
} }
} catch (InterruptedException ignored) { } catch (InterruptedException ignored) {
} }
@ -214,6 +217,113 @@ public class TaskTaskServiceImpl implements TaskService {
videoTaskRepository.clearTaskCache(taskUpdate.getId()); videoTaskRepository.clearTaskCache(taskUpdate.getId());
} }
@Override
public void forceCreateTaskByFaceIdAndTempalteId(Long faceId, Long templateId) {
FaceEntity face = faceRepository.getFace(faceId);
if (face == null) {
log.info("faceId:{} is not exist", faceId);
return;
}
List<FaceSampleEntity> faceSampleList = faceRepository.getFaceSampleList(faceId);
if (faceSampleList.isEmpty()) {
log.info("faceId:{} sample list not exist", faceId);
}
List<Long> faceSampleIds = faceSampleList.stream().map(FaceSampleEntity::getId).collect(Collectors.toList());
List<SourceEntity> sourceList = sourceMapper.listVideoByScenicFaceRelation(face.getScenicId(), faceId);
VideoPieceGetter.Task task = new VideoPieceGetter.Task();
task.faceId = faceId;
task.faceSampleIds = faceSampleIds;
task.templateId = templateId;
task.memberId = face.getMemberId();
task.callback = () -> {
log.info("task callback");
List<SourceEntity> videoSourceList = sourceMapper.listVideoByScenicFaceRelation(face.getScenicId(), faceId);
Map<String, List<SourceEntity>> sourcesMap = videoSourceList.stream()
.peek(item -> item.setUrl(item.getVideoUrl()))
.collect(Collectors.groupingBy(item -> item.getDeviceId().toString()));
if (sourcesMap.isEmpty()) {
// 主动禁止没有视频源视频生成
log.info("task callback: 没有视频源");
return;
}
sourcesMap.forEach((key, value) -> {
// 每个value只保留第一个
value.removeIf(item -> !value.get(0).equals(item));
});
TaskReqQuery taskReqQuery = new TaskReqQuery();
taskReqQuery.setFaceId(faceId);
taskReqQuery.setTemplateId(templateId);
taskReqQuery.setTaskParams(JSON.toJSONString(sourcesMap));
List<TaskRespVO> list = taskMapper.list(taskReqQuery);
MemberVideoEntity memberVideoEntity = new MemberVideoEntity();
memberVideoEntity.setMemberId(face.getMemberId());
memberVideoEntity.setScenicId(face.getScenicId());
memberVideoEntity.setFaceId(faceId);
memberVideoEntity.setTemplateId(templateId);
memberVideoEntity.setIsBuy(0);
if (list.isEmpty()) {
log.info("创建任务! faceId:{},templateId:{},taskParams:{}", faceId, templateId, sourcesMap);
ScenicConfigEntity scenicConfig = scenicRepository.getScenicConfig(face.getScenicId());
TaskEntity taskEntity = null;
if (Integer.valueOf(0).equals(scenicConfig.getTemplateNewVideoType())) {
log.info("景区{}启用templateNewVideoType全新视频原位替换", face.getScenicId());
taskReqQuery.setTemplateId(templateId);
List<TaskEntity> templateTaskList = taskMapper.listEntity(taskReqQuery);
if (!templateTaskList.isEmpty()) {
taskEntity = templateTaskList.get(0);
log.info("已有旧生成的视频:{}", taskEntity);
MemberVideoEntity taskVideoRelation = videoMapper.queryRelationByMemberTask(face.getMemberId(), taskEntity.getId());
if (taskVideoRelation != null) {
log.info("已有旧关联记录的视频:{}", taskVideoRelation);
memberVideoEntity.setIsBuy(taskVideoRelation.getIsBuy());
memberVideoEntity.setOrderId(taskVideoRelation.getOrderId());
}
taskMapper.deleteById(taskEntity.getId());
}
}
if (taskEntity == null) {
taskEntity = new TaskEntity();
taskEntity.setId(SnowFlakeUtil.getLongId());
taskEntity.setScenicId(face.getScenicId());
taskEntity.setFaceId(faceId);
taskEntity.setTemplateId(templateId);
taskEntity.setAutomatic(0);
}
taskEntity.setWorkerId(null);
taskEntity.setStatus(0);
taskEntity.setTaskParams(JSON.toJSONString(sourcesMap));
taskMapper.add(taskEntity);
memberVideoEntity.setTaskId(taskEntity.getId());
} else {
log.info("重复task! faceId:{},templateId:{},taskParams:{}", faceId, templateId, sourcesMap);
memberVideoEntity.setTaskId(list.get(0).getId());
VideoEntity video = videoMapper.findByTaskId(list.get(0).getId());
if (video != null) {
IsBuyRespVO isBuy = orderBiz.isBuy(face.getMemberId(), list.get(0).getScenicId(), 0, video.getId());
if (isBuy.isBuy()) {
memberVideoEntity.setIsBuy(1);
memberVideoEntity.setOrderId(isBuy.getOrderId());
}
if (isBuy.isFree()) {
memberVideoEntity.setIsBuy(1);
}
memberVideoEntity.setVideoId(video.getId());
}
}
videoMapper.addRelation(memberVideoEntity);
};
if (faceSampleIds.isEmpty()) {
// 没有人脸样本
if (!sourceList.isEmpty()) {
// 但是有原片
task.callback.onInvoke();
}
} else {
VideoPieceGetter.addTask(task);
}
}
@Override @Override
public void autoCreateTaskByFaceId(Long faceId) { public void autoCreateTaskByFaceId(Long faceId) {
FaceRespVO faceRespVO = faceMapper.getById(faceId); FaceRespVO faceRespVO = faceMapper.getById(faceId);
@ -410,6 +520,7 @@ public class TaskTaskServiceImpl implements TaskService {
taskUpdate.setWorkerId(worker.getId()); taskUpdate.setWorkerId(worker.getId());
taskMapper.setSuccess(taskUpdate); taskMapper.setSuccess(taskUpdate);
videoTaskRepository.clearTaskCache(taskUpdate.getId()); videoTaskRepository.clearTaskCache(taskUpdate.getId());
VideoEntity video = videoMapper.findByTaskId(taskId); VideoEntity video = videoMapper.findByTaskId(taskId);
if (video != null) { if (video != null) {
video.setVideoUrl(task.getVideoUrl()); video.setVideoUrl(task.getVideoUrl());
@ -503,7 +614,7 @@ public class TaskTaskServiceImpl implements TaskService {
taskMapper.update(updateTask); taskMapper.update(updateTask);
videoTaskRepository.clearTaskCache(updateTask.getId()); videoTaskRepository.clearTaskCache(updateTask.getId());
} }
return adapter.getUrlForUpload(filename); return adapter.getUrlForUpload(new Date(System.currentTimeMillis() + 1000 * 60 * 60), "video/mp4", filename);
} }
public void sendVideoGeneratedServiceNotification(Long taskId) { public void sendVideoGeneratedServiceNotification(Long taskId) {

View File

@ -96,26 +96,26 @@ public class DynamicTaskGenerator {
if (task == null) { if (task == null) {
return; return;
} }
log.info("开始执行任务:{}", task);
// 根据人脸照片获取人脸样本ID // 根据人脸照片获取人脸样本ID
FaceSampleRespVO faceSample = faceSampleMapper.getById(task.getFaceSampleId()); FaceSampleRespVO faceSample = faceSampleMapper.getById(task.getFaceSampleId());
if (faceSample == null) { if (faceSample == null) {
log.info("人脸样本ID{}不存在", task.getFaceSampleId()); log.debug("人脸样本ID{}不存在", task.getFaceSampleId());
return; return;
} }
if (faceSample.getScore() == null) { if (faceSample.getScore() == null) {
log.info("人脸样本ID{}人脸质量为空", task.getFaceSampleId()); log.debug("人脸样本ID{}人脸质量为空", task.getFaceSampleId());
return; return;
} }
ScenicConfigEntity scenicConfig = scenicRepository.getScenicConfig(faceSample.getScenicId()); ScenicConfigEntity scenicConfig = scenicRepository.getScenicConfig(faceSample.getScenicId());
if (scenicConfig == null) { if (scenicConfig == null) {
log.info("当前景区{},无配置", faceSample.getScenicId()); log.debug("当前景区{},无配置", faceSample.getScenicId());
return; return;
} }
if (!Integer.valueOf(1).equals(scenicConfig.getBookRoutine()) && !Integer.valueOf(3).equals(scenicConfig.getBookRoutine())) { if (!Integer.valueOf(1).equals(scenicConfig.getBookRoutine()) && !Integer.valueOf(3).equals(scenicConfig.getBookRoutine())) {
log.info("当前景区{}未启用预约流程,跳过", faceSample.getScenicId()); log.debug("当前景区{}未启用预约流程,跳过", faceSample.getScenicId());
return; return;
} }
log.info("开始执行任务:{}", task);
SearchFaceRespVo userDbSearchResult = faceService.searchFace(USER_FACE_DB_NAME+faceSample.getScenicId(), faceSample.getFaceUrl()); SearchFaceRespVo userDbSearchResult = faceService.searchFace(USER_FACE_DB_NAME+faceSample.getScenicId(), faceSample.getFaceUrl());
// 如果人脸样本ID在人脸样本库中则创建任务 // 如果人脸样本ID在人脸样本库中则创建任务
if (!userDbSearchResult.getSampleListIds().isEmpty()) { if (!userDbSearchResult.getSampleListIds().isEmpty()) {

View File

@ -1,9 +1,16 @@
package com.ycwl.basic.task; package com.ycwl.basic.task;
import com.ycwl.basic.mapper.FaceSampleMapper;
import com.ycwl.basic.mapper.ScenicMapper; import com.ycwl.basic.mapper.ScenicMapper;
import com.ycwl.basic.mapper.SourceMapper;
import com.ycwl.basic.model.pc.faceSample.req.FaceSampleReqQuery;
import com.ycwl.basic.model.pc.faceSample.resp.FaceSampleRespVO;
import com.ycwl.basic.model.pc.scenic.req.ScenicReqQuery; import com.ycwl.basic.model.pc.scenic.req.ScenicReqQuery;
import com.ycwl.basic.model.pc.scenic.resp.ScenicRespVO; import com.ycwl.basic.model.pc.scenic.resp.ScenicRespVO;
import com.ycwl.basic.service.task.TaskFaceService; import com.ycwl.basic.service.task.TaskFaceService;
import com.ycwl.basic.storage.StorageFactory;
import com.ycwl.basic.storage.adapters.IStorageAdapter;
import com.ycwl.basic.storage.entity.StorageFileObject;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.EnableScheduling; import org.springframework.scheduling.annotation.EnableScheduling;
@ -20,7 +27,10 @@ public class FaceCleaner {
private ScenicMapper scenicMapper; private ScenicMapper scenicMapper;
@Autowired @Autowired
private TaskFaceService faceService; private TaskFaceService faceService;
@Autowired
private FaceSampleMapper faceSampleMapper;
@Autowired
private SourceMapper sourceMapper;
@Scheduled(cron = "0 0 4 * * ?") @Scheduled(cron = "0 0 4 * * ?")
public void clean(){ public void clean(){
@ -31,4 +41,23 @@ public class FaceCleaner {
faceService.batchDeleteExpiredFace(scenic.getId()); faceService.batchDeleteExpiredFace(scenic.getId());
}); });
} }
@Scheduled(cron = "0 0 3 * * ?")
public void deleteExpiredSource(){
}
@Scheduled(cron = "0 0 5 * * ?")
public void clear(){
log.info("开始清理人脸文件");
List<FaceSampleRespVO> faceSampleRespVOS = faceSampleMapper.list(new FaceSampleReqQuery());
IStorageAdapter adapter = StorageFactory.use("faces");
List<StorageFileObject> fileObjectList = adapter.listDir("user-face");
fileObjectList.parallelStream().forEach(fileObject -> {
if(faceSampleRespVOS.parallelStream().noneMatch(faceSampleRespVO -> faceSampleRespVO.getFaceUrl().contains(fileObject.getFullPath()))){
log.info("删除人脸文件:{}", fileObject);
adapter.deleteFile(fileObject.getFullPath());
}
});
}
} }

View File

@ -13,7 +13,9 @@ import com.ycwl.basic.repository.ScenicRepository;
import com.ycwl.basic.repository.TemplateRepository; import com.ycwl.basic.repository.TemplateRepository;
import com.ycwl.basic.service.task.TaskFaceService; import com.ycwl.basic.service.task.TaskFaceService;
import com.ycwl.basic.service.task.impl.TaskTaskServiceImpl; import com.ycwl.basic.service.task.impl.TaskTaskServiceImpl;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile;
import org.springframework.scheduling.annotation.EnableScheduling; import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled; import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@ -21,8 +23,10 @@ import org.springframework.stereotype.Component;
import java.util.Date; import java.util.Date;
import java.util.List; import java.util.List;
@Slf4j
@EnableScheduling @EnableScheduling
@Component @Component
@Profile("prod")
public class VideoTaskGenerator { public class VideoTaskGenerator {
@Autowired @Autowired
private FaceMapper faceMapper; private FaceMapper faceMapper;
@ -55,31 +59,10 @@ public class VideoTaskGenerator {
taskFaceService.searchFace(face.getId()); taskFaceService.searchFace(face.getId());
boolean canAutoGenerate = templateBiz.determineTemplateCanAutoGenerate(templateId, face.getId(), false); boolean canAutoGenerate = templateBiz.determineTemplateCanAutoGenerate(templateId, face.getId(), false);
if (canAutoGenerate) { if (canAutoGenerate) {
taskTaskService.autoCreateTaskByFaceId(face.getId()); log.info("task callback: 自动生成");
} taskTaskService.forceCreateTaskByFaceIdAndTempalteId(face.getId(), templateId);
}); } else {
} log.info("task callback: 不自动生成");
@Scheduled(cron = "0 30 4 * * *")
public void generateVideoTaskZTJQ() {
// 指定获取指定日期的未完成人脸样本并生成任务
Long scenicId = 3930324797233434624L;
List<ContentPageVO> contentList = templateMapper.listFor(scenicId);
if (contentList.isEmpty()) {
return;
}
Long templateId = contentList.get(0).getTemplateId();
FaceReqQuery query = new FaceReqQuery();
query.setScenicId(scenicId);
query.setStartTime(DateUtil.beginOfDay(new Date()));
query.setEndTime(DateUtil.endOfDay(new Date()));
List<FaceRespVO> list = faceMapper.list(query);
list.stream().parallel().forEach(face -> {
taskFaceService.searchFace(face.getId());
boolean canAutoGenerate = templateBiz.determineTemplateCanAutoGenerate(templateId, face.getId(), false);
if (canAutoGenerate) {
taskTaskService.autoCreateTaskByFaceId(face.getId());
} }
}); });
} }

View File

@ -96,7 +96,7 @@ public class JwtTokenUtil {
public static void main(String[] args) throws Exception { public static void main(String[] args) throws Exception {
JwtInfo jwtInfo = new JwtInfo(); JwtInfo jwtInfo = new JwtInfo();
jwtInfo.setUserId(3936121342868459520L); jwtInfo.setUserId(3950649683084447744L);
jwtInfo.setName("微信用户"); jwtInfo.setName("微信用户");
System.out.println(generateToken(jwtInfo, 86400)); System.out.println(generateToken(jwtInfo, 86400));
} }

View File

@ -101,4 +101,9 @@
from face_sample from face_sample
where id = #{id} where id = #{id}
</select> </select>
<select id="listEntity" resultType="com.ycwl.basic.model.pc.faceSample.entity.FaceSampleEntity">
select *
from face_sample
where scenic_id = #{scenicId} and create_at &lt;= #{endDate}
</select>
</mapper> </mapper>

View File

@ -70,7 +70,7 @@
LEFT JOIN video v ON mv.video_id = v.id LEFT JOIN video v ON mv.video_id = v.id
), ),
member_source_data AS ( member_source_data AS (
SELECT ms.member_id, ms.source_id, ms.face_id, f.face_url, s.video_url, s.url SELECT ms.member_id, ms.type, ms.source_id, ms.face_id, f.face_url, s.video_url, s.url
FROM member_source ms FROM member_source ms
LEFT JOIN face f ON ms.face_id = f.id LEFT JOIN face f ON ms.face_id = f.id
LEFT JOIN source s ON ms.source_id = s.id LEFT JOIN source s ON ms.source_id = s.id
@ -79,6 +79,7 @@
oi.id AS oiId, oi.id AS oiId,
oi.order_id AS orderId, oi.order_id AS orderId,
oi.goods_id, oi.goods_id,
msd.source_id,
sc.id AS scenic_id, sc.id AS scenic_id,
sc.name AS scenic_name, sc.name AS scenic_name,
CASE oi.goods_type CASE oi.goods_type
@ -113,7 +114,7 @@
LEFT JOIN `order` o ON oi.order_id = o.id LEFT JOIN `order` o ON oi.order_id = o.id
LEFT JOIN scenic sc ON o.scenic_id = sc.id LEFT JOIN scenic sc ON o.scenic_id = sc.id
LEFT JOIN member_video_data mvd ON o.member_id = mvd.member_id AND oi.goods_id = mvd.video_id LEFT JOIN member_video_data mvd ON o.member_id = mvd.member_id AND oi.goods_id = mvd.video_id
LEFT JOIN member_source_data msd ON o.member_id = msd.member_id AND oi.goods_id = msd.face_id LEFT JOIN member_source_data msd ON o.member_id = msd.member_id AND oi.goods_id = msd.face_id AND msd.type = oi.goods_type
WHERE oi.order_id = #{id}; WHERE oi.order_id = #{id};
</select> </select>

View File

@ -92,7 +92,9 @@
template_new_video_type=#{templateNewVideoType}, template_new_video_type=#{templateNewVideoType},
anti_screen_record_type=#{antiScreenRecordType}, anti_screen_record_type=#{antiScreenRecordType},
disable_source_video=#{disableSourceVideo}, disable_source_video=#{disableSourceVideo},
disable_source_image=#{disableSourceImage} disable_source_image=#{disableSourceImage},
video_source_store_day=#{videoSourceStoreDay},
image_source_store_day=#{imageSourceStoreDay}
</set> </set>
where id = #{id} where id = #{id}
</update> </update>

View File

@ -121,7 +121,7 @@
<if test="isBuy!=null">and ms.is_buy = #{isBuy} </if> <if test="isBuy!=null">and ms.is_buy = #{isBuy} </if>
<if test="type!=null">and ms.type = #{type} </if> <if test="type!=null">and ms.type = #{type} </if>
<if test="faceId!=null">and ms.face_id = #{faceId} </if> <if test="faceId!=null">and ms.face_id = #{faceId} </if>
order by so.create_time desc order by so.create_time asc
</select> </select>
<select id="listUserOne" resultType="com.ycwl.basic.model.pc.source.resp.SourceRespVO"> <select id="listUserOne" resultType="com.ycwl.basic.model.pc.source.resp.SourceRespVO">

View File

@ -40,7 +40,7 @@
FROM ( FROM (
select count(1) as count select count(1) as count
from statistics from statistics
where type=0 and scenic_id = #{scenicId} where type=10 and morph_id in (1047,1048,1049) and scenic_id = #{scenicId}
<if test="startTime!= null"> <if test="startTime!= null">
and create_time >= #{startTime} and create_time >= #{startTime}
</if> </if>
@ -169,7 +169,7 @@
FROM ( FROM (
select count(1) as count select count(1) as count
from statistics from statistics
where type in (0,10) and scenic_id = #{scenicId} where type in (10) and scenic_id = #{scenicId}
<if test="startTime!= null"> <if test="startTime!= null">
and create_time >= #{startTime} and create_time >= #{startTime}
</if> </if>