From fd03e31f173ad9c52b15a30a9127e2b6a468538d Mon Sep 17 00:00:00 2001 From: shikeying <shikeying@163.com> Date: 星期一, 26 九月 2022 18:50:33 +0800 Subject: [PATCH] 视频相似度分析2 --- recommend-video/src/main/java/com/iplatform/recvideo/service/VideoExecutorServiceImpl.java | 90 +++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 90 insertions(+), 0 deletions(-) diff --git a/recommend-video/src/main/java/com/iplatform/recvideo/service/VideoExecutorServiceImpl.java b/recommend-video/src/main/java/com/iplatform/recvideo/service/VideoExecutorServiceImpl.java index bd4408f..6fe6174 100644 --- a/recommend-video/src/main/java/com/iplatform/recvideo/service/VideoExecutorServiceImpl.java +++ b/recommend-video/src/main/java/com/iplatform/recvideo/service/VideoExecutorServiceImpl.java @@ -1,9 +1,99 @@ package com.iplatform.recvideo.service; +import com.iplatform.model.po.Rc_video_t1; +import com.iplatform.model.po.Rc_video_t2; +import com.iplatform.model.po.Rc_video_user; +import com.walker.infrastructure.utils.DateUtils; +import com.walker.infrastructure.utils.StringUtils; import com.walker.jdbc.service.BaseServiceImpl; import org.springframework.stereotype.Service; + +import java.util.List; +import java.util.Map; @Service public class VideoExecutorServiceImpl extends BaseServiceImpl { + private static final String SQL_CHECK_VIDEO_STATUS = "select * from milvus_video_status where id=?"; + private static final String SQL_CLEAR_VIDEO_T1 = "delete from rc_video_t1 where src_img=?"; + private static final String SQL_CLEAR_VIDEO_T2 = "delete from rc_video_t2 where src_video_id=?"; + + private static final String SQL_GET_BATCH_VIDEO = "select user_id, src_video_id from rc_video_batch where batch_id=?"; + + private static final String SQL_UPDATE_TASK_STATUS_LOAD = "update rc_task_status set status='1', end_time=? where last_value=? and status='0'"; + + /** + * 鍐欏叆瑙嗛鐩镐技搴︾涓�绾т复鏃舵暟鎹紝姣忎釜鍥惧儚鍖呭惈澶氫釜鐩镐技瑙嗛璁板綍銆� + * @param list + * @param srcImageId 鍘熷鍥惧儚ID + */ + public void execBatchInsertVideoT1(List<Rc_video_t1> list, String srcImageId){ + // 鍐欏叆鏂版暟鎹箣鍓嶏紝鍏堟竻闄よ�佽棰戞暟鎹紝鍚庣画鍒嗘瀽閬垮厤鏁版嵁閲嶅(姣忎釜瑙嗛) + this.execute(SQL_CLEAR_VIDEO_T1, new Object[]{srcImageId}); + this.insert(list); + } + + /** + * 鍐欏叆瑙嗛鐩镐技搴︾浜屼复鏃舵暟鎹紝姣忎釜鍘熷瑙嗛瀵瑰簲澶氫釜鐩镐技瑙嗛锛屽凡鎺掑簭瀛樺偍銆� + * @param list + * @param srcVideoId + */ + public void execBatchInsertVideoT2(List<Rc_video_t2> list, String srcVideoId){ + this.execute(SQL_CLEAR_VIDEO_T2, new Object[]{srcVideoId}); + this.insert(list); + } + + /** + * 鍐欏叆鐢ㄦ埛鎺ㄨ崘瑙嗛璁板綍闆嗗悎锛屽苟鏇存柊鎵规浠诲姟鐘舵�佷负(宸插畬鎴�) + * @param videoUserList + * @param batchId + */ + public void execBatchInsertVideoUser(List<Rc_video_user> videoUserList, String batchId){ + this.insert(videoUserList); + Object[] param = new Object[2]; + param[0] = Long.parseLong(DateUtils.getDateTimeSecondForShow()); + param[1] = Long.parseLong(batchId); + this.execute(SQL_UPDATE_TASK_STATUS_LOAD, param); + } + + + /** + * 杩斿洖涓�涓壒娆$敤鎴峰搴旇棰戣褰曢泦鍚堬紝鐢ㄤ簬鏈�鍚庢洿鏂扮敤鎴锋帹鑽愯棰戞暟鎹�� + * @param batchId + * @return + * @date 2022-09-26 + */ + public List<Map<String, Object>> queryBatchUserVideoList(String batchId){ + return this.select(SQL_GET_BATCH_VIDEO, new Object[]{batchId}); + } + + public List<Rc_video_t2> queryVideoT_2List(String batchId){ + return this.select(new Rc_video_t2(), "where batch_id=?", new Object[]{batchId}); + } + + /** + * 鏍规嵁鍘熷瑙嗛ID杩斿洖鐩镐技璁板綍闆嗗悎銆� + * @param srcVideoId + * @return + */ + public List<Rc_video_t1> queryVideoT_1List(String srcVideoId){ + return this.select(new Rc_video_t1(), "where src_video_id=?", new Object[]{srcVideoId}); + } + + /** + * 鏌ヨ瑙嗛鍔犺浇鐘舵�佽〃锛屾槸鍚﹀凡缁忓畬鎴愬姞杞�(绗竴姝�)锛屽鏋滆褰曚笉瀛樺湪璇存槑杩樻病鏈夊紑濮嬪姞杞姐�� + * @param videoStatusId + * @return + */ + public boolean queryLoadVideoDone(String videoStatusId){ + Map<String, Object> map = this.get(SQL_CHECK_VIDEO_STATUS, new Object[]{videoStatusId}); + if(map == null){ + return false; + } + int status = Integer.parseInt(map.get("status").toString()); + if(status == 1){ + return true; + } + return false; + } } -- Gitblit v1.9.1