孙永军 1 рік тому
батько
коміт
77c9467e46

+ 3 - 1
src/main/java/com/sxtvs/open/api/review/service/impl/ReviewImgRecordServiceImpl.java

@@ -77,7 +77,9 @@ public class ReviewImgRecordServiceImpl extends ServiceImpl<ReviewImgRecordMappe
             reviewImgRecord.setResult(JSON.toJSONString(body));
             reviewImgRecord.setReviewResult(body);
             updateById(reviewImgRecord);
-            if(sseService.sendMsg(uid, "图片" + reviewImgRecord.getTitle() + "审核完成")){
+            if(sseService.sendMsg(uid,
+                    ImmutableMap.of("type", "IMG", "msg", "图片" + reviewImgRecord.getTitle() + "审核完成")
+                    )){
                 reviewImgRecord.setPushStatus(1);
                 updateById(reviewImgRecord);
             }

+ 7 - 2
src/main/java/com/sxtvs/open/api/review/service/impl/ReviewVideosRecordServiceImpl.java

@@ -99,7 +99,10 @@ public class ReviewVideosRecordServiceImpl extends ServiceImpl<ReviewVideosRecor
 
         updateById(reviewVideosRecord);
 
-        if(sseService.sendMsg(reviewVideosRecord.getUid(), "视频" + reviewVideosRecord.getTitle() + "审核完成")){
+        if(sseService.sendMsg(reviewVideosRecord.getUid(),
+                ImmutableMap.of("type", "IMG", "msg",
+                        "视频" + reviewVideosRecord.getTitle() + "审核完成")
+        )){
             reviewVideosRecord.setPushStatus(1);
             updateById(reviewVideosRecord);
         }
@@ -108,7 +111,9 @@ public class ReviewVideosRecordServiceImpl extends ServiceImpl<ReviewVideosRecor
 
     public void push(){
         lambdaQuery().eq(ReviewVideosRecord::getStatus, "SUCCESS").eq(ReviewVideosRecord::getPushStatus, 0).list().forEach(x -> {
-            if(sseService.sendMsg(x.getUid(), "视频" + x.getTitle() + "审核完成")){
+            if(sseService.sendMsg(x.getUid(),
+                    ImmutableMap.of("type", "IMG", "msg", "视频" + x.getTitle() + "审核完成")
+            )){
                 x.setPushStatus(1);
                 updateById(x);
             }

+ 1 - 1
src/main/java/com/sxtvs/open/api/review/service/impl/SSEService.java

@@ -30,7 +30,7 @@ public class SSEService {
 
     }
 
-    public Boolean sendMsg(Long uid, String msg) {
+    public Boolean sendMsg(Long uid, Object msg) {
         if (sseMap.containsKey(uid)){
 
             SseEmitter sseEmitter = sseMap.get(uid);