fcm

internal/push/fcm/push.go

@@ -7,11 +7,12 @@ import (
 	"Open_IM/pkg/common/log"
 	"Open_IM/pkg/tools/splitter"
 	"context"
+	"path/filepath"
+	"strconv"
+
 	firebase "firebase.google.com/go"
 	"firebase.google.com/go/messaging"
 	"google.golang.org/api/option"
-	"path/filepath"
-	"strconv"
 )
 
 const SinglePushCountLimit = 400
@@ -67,6 +68,13 @@ func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string,
 	Msg.Notification = &messaging.Notification{}
 	Msg.Notification.Body = detailContent
 	Msg.Notification.Title = alert
+	if opts.IOSBadgeCount {
+		i := 1
+		Msg.APNS.Payload.Aps.Badge = &i
+	}
+	if opts.IOSPushSound != "" {
+		Msg.APNS.Payload.Aps.Sound = opts.IOSPushSound
+	}
 	ctx := context.Background()
 	for _, v := range result {
 		Msg.Tokens = v.Item

internal/push/logic/push_to_client.go

@@ -318,7 +318,8 @@ func GetOfflinePushOpts(pushMsg *pbPush.PushMsgReq) (opts push.PushOpts, err err
 			opts.Signal.ClientMsgID = pushMsg.MsgData.ClientMsgID
 			log.NewDebug(pushMsg.OperationID, opts)
 		}
-
+		opts.IOSBadgeCount = pushMsg.MsgData.OfflinePushInfo.IOSBadgeCount
+		opts.IOSPushSound = pushMsg.MsgData.OfflinePushInfo.IOSPushSound
 	}
 	return opts, nil
 }

internal/push/push_interface.go

@@ -5,7 +5,9 @@ type OfflinePusher interface {
 }
 
 type PushOpts struct {
-	Signal Signal
+	Signal        Signal
+	IOSPushSound  string
+	IOSBadgeCount bool
 }
 
 type Signal struct {