app极光推送

business-service/src/main/java/com/ozs/service/mapper/UserRegistrationidMapper.java

@@ -4,6 +4,9 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.ozs.service.entity.SvcAddress;
 import com.ozs.service.entity.UserRegistrationid;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+
+import java.util.List;
 
 /**
  * @Author : sunhh
@@ -14,4 +17,6 @@ public interface UserRegistrationidMapper extends BaseMapper<UserRegistrationid>
     int updateRegistrationID(UserRegistrationid userRegistrationid);
 
     int addRegistrationID(UserRegistrationid userRegistrationid);
+
+    List<String> selectRegistrationID(@Param("userIdList") List<String> userIdList);
 }

business-service/src/main/java/com/ozs/service/service/UserRegistrationidService.java

@@ -3,10 +3,14 @@ package com.ozs.service.service;
 import com.baomidou.mybatisplus.extension.service.IService;
 import com.ozs.service.entity.UserRegistrationid;
 
+import java.util.List;
+
 /**
  * @Author : sunhh
  * @create 2023/3/14 17:07
  */
 public interface UserRegistrationidService extends IService<UserRegistrationid> {
     Integer addRegistrationID(UserRegistrationid userRegistrationid);
+
+    List<String> selectRegistrationID(List<String> userIdList);
 }

business-service/src/main/java/com/ozs/service/service/impl/UserRegistrationidServiceImpl.java

@@ -35,4 +35,9 @@ public class UserRegistrationidServiceImpl extends ServiceImpl<UserRegistrationi
         }
         return add;
     }
+
+    @Override
+    public List<String> selectRegistrationID(List<String> userIdList) {
+        return userRegistrationidMapper.selectRegistrationID(userIdList);
+    }
 }

business-service/src/main/resources/mapper/service/UserRegistrationidMapper.xml

@@ -11,4 +11,11 @@
         </set>
         where user_id = #{userId}
     </update>
+
+    <select id="selectRegistrationID" resultType="java.lang.String" parameterType="java.util.List">
+        select registration_id from user_registrationid where user_id in
+        <foreach item="item" collection="userIdList" separator="," open="(" close=")" index="">
+            #{item}
+        </foreach>
+    </select>
 </mapper>

hazard-admin/src/main/java/com/ozs/web/controller/websocket/WebSocketConteoller.java

@@ -10,6 +10,7 @@ import com.ozs.service.entity.vo.MsgAlarmResp;
 import com.ozs.service.service.BaseCameraManagementService;
 import com.ozs.service.service.BaseRailwayManagementService;
 import com.ozs.service.service.BaseUserService;
+import com.ozs.service.service.UserRegistrationidService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Controller;
@@ -17,6 +18,7 @@ import org.springframework.util.ObjectUtils;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
+import java.text.SimpleDateFormat;
 import java.util.*;
 
 /**
@@ -36,6 +38,8 @@ public class WebSocketConteoller extends BaseController {
     private BaseRailwayManagementService baseRailwayManagementService;
     @Autowired
     private BaseCameraManagementService baseCameraManagementService;
+    @Autowired
+    private UserRegistrationidService userRegistrationidService;
 
     @ResponseBody
     @PostMapping("/publish")
@@ -64,9 +68,18 @@ public class WebSocketConteoller extends BaseController {
         webSocketServer.sendMoreMessage(list, objStr(msgAlarmResp));
 
         // app 消息推送
-        List<String> alias = new ArrayList<String>();
-        alias.add("sunhh");
-        JPushUtil.sendToRegistrationId(alias, "中奖通知!!!!", "中奖通知!!!!!", "恭喜你中了100000000个亿!!!！", "https://www.baidu.com");
+        List<String> userIdList = new ArrayList<>(set);
+        List<String> alias = userRegistrationidService.selectRegistrationID(userIdList);
+        // List<String> alias = new ArrayList<String>();
+        // alias.add("sunhh");
+        Date alarmTime = msgAlarm.getAlarmTime();
+        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        String formatDate = sdf.format(alarmTime);
+        JPushUtil.sendToRegistrationId(
+                alias,
+                "报警通知！", cameraName,
+                formatDate + "     " + ("1".equals(msgAlarmResp.getAlarmType()) ? "泥石流" : msgAlarmResp.getAlarmType()) + "     " + mile + "     " + endMile,
+                "");
         HashMap<String, Object> map = new HashMap<>();
         map.put("code", 200);
         return map;