You've already forked FrameTour-BE
task缓存
This commit is contained in:
@ -97,38 +97,6 @@ public class AppScenicServiceImpl implements AppScenicService {
|
||||
return ApiResponse.success(dataList);
|
||||
}
|
||||
|
||||
@Override
|
||||
public ApiResponse<TopStateResp> topState() {
|
||||
TopStateResp topStateResp = new TopStateResp();
|
||||
String userId = BaseContextHandler.getUserId();
|
||||
if (StringUtils.isBlank(userId)) {
|
||||
return ApiResponse.success(topStateResp);
|
||||
}
|
||||
// 查询是否有人脸
|
||||
FaceReqQuery faceReqQuery = new FaceReqQuery();
|
||||
faceReqQuery.setMemberId(Long.parseLong(userId));
|
||||
List<FaceRespVO> list = faceMapper.list(faceReqQuery);
|
||||
for (FaceRespVO faceRespVO : list) {
|
||||
if (faceRespVO.getScore().compareTo(faceScore) >= NumberConstant.ZERO) {
|
||||
topStateResp.setIsFace(true);
|
||||
}
|
||||
}
|
||||
|
||||
// 查询是否有专属视频待制作
|
||||
int sourceNum = sourceMapper.countByMemberId(userId);
|
||||
topStateResp.setMadeVideos(sourceNum > 0);
|
||||
|
||||
// 是否有专属视频合成中
|
||||
int taskSynthesizingNum = taskMapper.countByMemberIdStau(userId);
|
||||
topStateResp.setCompositingVideo(taskSynthesizingNum > 0);
|
||||
|
||||
// AI合成了多少个视频 完成
|
||||
int taskFinishNum = taskMapper.countByMemberIdStauFinish(userId);
|
||||
topStateResp.setAiVideoNum(taskFinishNum);
|
||||
|
||||
return ApiResponse.success(topStateResp);
|
||||
}
|
||||
|
||||
@Override
|
||||
public ApiResponse<List<ContentPageVO>> contentList(Long faceId) {
|
||||
JwtInfo worker = JwtTokenUtil.getWorker();
|
||||
|
@ -26,6 +26,7 @@ import com.ycwl.basic.model.task.req.ClientStatusReqVo;
|
||||
import com.ycwl.basic.model.task.req.TaskReqVo;
|
||||
import com.ycwl.basic.model.task.req.WorkerAuthReqVo;
|
||||
import com.ycwl.basic.model.task.resp.TaskSyncRespVo;
|
||||
import com.ycwl.basic.repository.VideoTaskRepository;
|
||||
import com.ycwl.basic.service.task.TaskService;
|
||||
import com.ycwl.basic.storage.StorageFactory;
|
||||
import com.ycwl.basic.storage.adapters.IStorageAdapter;
|
||||
@ -68,6 +69,8 @@ public class TaskTaskServiceImpl implements TaskService {
|
||||
private VideoMapper videoMapper;
|
||||
@Autowired
|
||||
private TemplateRepository templateRepository;
|
||||
@Autowired
|
||||
private VideoTaskRepository videoTaskRepository;
|
||||
|
||||
private RenderWorkerEntity getWorker(@NonNull WorkerAuthReqVo req) {
|
||||
String accessKey = req.getAccessKey();
|
||||
@ -110,7 +113,10 @@ public class TaskTaskServiceImpl implements TaskService {
|
||||
// Template
|
||||
List<TaskRespVO> taskList = taskMapper.selectNotRunning();
|
||||
resp.setTasks(taskList);
|
||||
taskList.forEach(task -> taskMapper.assignToWorker(task.getId(), worker.getId()));
|
||||
taskList.forEach(task -> {
|
||||
taskMapper.assignToWorker(task.getId(), worker.getId());
|
||||
videoTaskRepository.clearTaskCache(task.getId());
|
||||
});
|
||||
// return Task
|
||||
return resp;
|
||||
}
|
||||
@ -174,6 +180,7 @@ public class TaskTaskServiceImpl implements TaskService {
|
||||
taskUpdate.setId(taskId);
|
||||
taskUpdate.setWorkerId(worker.getId());
|
||||
taskMapper.setStart(taskUpdate);
|
||||
videoTaskRepository.clearTaskCache(taskUpdate.getId());
|
||||
}
|
||||
|
||||
@Override
|
||||
@ -284,6 +291,7 @@ public class TaskTaskServiceImpl implements TaskService {
|
||||
taskUpdate.setStatus(1);
|
||||
taskUpdate.setWorkerId(worker.getId());
|
||||
taskMapper.setSuccess(taskUpdate);
|
||||
videoTaskRepository.clearTaskCache(taskUpdate.getId());
|
||||
VideoEntity video = videoMapper.findByTaskId(taskId);
|
||||
if (video != null) {
|
||||
video.setVideoUrl(task.getVideoUrl());
|
||||
@ -319,6 +327,7 @@ public class TaskTaskServiceImpl implements TaskService {
|
||||
taskUpdate.setWorkerId(worker.getId());
|
||||
taskMapper.setFail(taskUpdate);
|
||||
taskMapper.deassign(taskId);
|
||||
videoTaskRepository.clearTaskCache(taskUpdate.getId());
|
||||
}
|
||||
|
||||
@Override
|
||||
@ -336,6 +345,7 @@ public class TaskTaskServiceImpl implements TaskService {
|
||||
updateTask.setId(taskId);
|
||||
updateTask.setVideoUrl(url);
|
||||
taskMapper.update(updateTask);
|
||||
videoTaskRepository.clearTaskCache(updateTask.getId());
|
||||
}
|
||||
return adapter.getUrlForUpload(filename);
|
||||
}
|
||||
|
@ -33,12 +33,6 @@ public interface AppScenicService {
|
||||
*/
|
||||
ApiResponse<List<ScenicAppVO>> scenicList(ScenicIndexVO scenicIndexVO);
|
||||
|
||||
/**
|
||||
* 首页顶部状态
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
ApiResponse<TopStateResp> topState();
|
||||
|
||||
ApiResponse<List<ContentPageVO>> contentList(Long scenicId);
|
||||
ApiResponse<List<ContentPageVO>> faceContentList(Long userId, Long faceId);
|
||||
|
Reference in New Issue
Block a user