feat: complete IM P0 and P1 reliability work

server/push/fcm/payload.go

@@ -123,7 +123,7 @@ func PrepareV1Notifications(rcpt *push.Receipt, config *configType) ([]*fcmv1.Me
 				skipDevices[deviceID] = struct{}{}
 			}
 		}
-		devices, count, err = store.Devices.GetAll(rcpt.TenantID, uids...)
+		devices, count, err = pushDevicesForUsers(rcpt.TenantID, uids...)
 		if err != nil {
 			logs.Warn.Println("fcm push: db error", err)
 			return nil, nil
@@ -218,7 +218,7 @@ func TenantChannelName(tenantID t.TenantID, channel string) string {
 
 // DevicesForUser loads device IDs of the given user.
 func DevicesForUser(tenantID t.TenantID, uid t.Uid) []string {
-	ddef, count, err := store.Devices.GetAll(tenantID, uid)
+	ddef, count, err := pushDevicesForUsers(tenantID, uid)
 	if err != nil {
 		logs.Warn.Println("fcm devices for user: db error", err)
 		return nil
@@ -235,6 +235,19 @@ func DevicesForUser(tenantID t.TenantID, uid t.Uid) []string {
 	return devices
 }
 
+func pushDevicesForUsers(tenantID t.TenantID, uids ...t.Uid) (map[t.Uid][]t.DeviceDef, int, error) {
+	if tokens, ok := store.DevicePushTokens.(store.DevicePushTokenQueryInterface); ok {
+		devices, count, err := tokens.GetAll(tenantID, uids...)
+		if err != nil {
+			return nil, 0, err
+		}
+		if count > 0 {
+			return devices, count, nil
+		}
+	}
+	return store.Devices.GetAll(tenantID, uids...)
+}
+
 // ChannelsForUser loads user's channel subscriptions with P permission.
 func ChannelsForUser(tenantID t.TenantID, uid t.Uid) []string {
 	channels, err := store.Users.GetChannels(tenantID, uid)