blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

add test file

Browse Source
This commit is contained in:
Gordon 2022-07-29 17:03:26 +08:00
parent cd3778a620
commit 1b92774778
1 changed files with 12 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
internal/push/fcm/push.go
View File

@ -5,10 +5,10 @@ import (
"Open_IM/pkg/common/config" "Open_IM/pkg/common/config"
"Open_IM/pkg/common/db" "Open_IM/pkg/common/db"
"Open_IM/pkg/common/log" "Open_IM/pkg/common/log"
"Open_IM/pkg/tools"
"context" "context"
firebase "firebase.google.com/go" firebase "firebase.google.com/go"
"firebase.google.com/go/messaging" "firebase.google.com/go/messaging"
"fmt"
"google.golang.org/api/option" "google.golang.org/api/option"
"path/filepath" "path/filepath"
"strconv" "strconv"
@ -47,7 +47,6 @@ func newFcmClient() *Fcm {
} }
func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string, opts push.PushOpts) (string, error) { func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string, opts push.PushOpts) (string, error) {
//需要一个客户端的Token
// accounts->registrationToken // accounts->registrationToken
Tokens := make([]string, 0) Tokens := make([]string, 0)
for _, account := range accounts { for _, account := range accounts {
@ -61,23 +60,16 @@ func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string,
Tokens = append(Tokens, AndroidfcmToken) Tokens = append(Tokens, AndroidfcmToken)
} }
} }
tokenlen := len(Tokens)
// 500组为一个推送，我们用400好了
pages := int((tokenlen-1)/SinglePushCountLimit + 1)
Success := 0 Success := 0
Fail := 0 Fail := 0
fmt.Println(operationID, "fcm args", tokenlen, pages) result := tools.NewSplitter(SinglePushCountLimit, Tokens).GetSplitResult()
for i := 0; i < pages; i++ { Msg := new(messaging.MulticastMessage)
Msg := new(messaging.MulticastMessage) Msg.Notification = &messaging.Notification{}
Msg.Notification = &messaging.Notification{} Msg.Notification.Body = detailContent
Msg.Notification.Body = detailContent Msg.Notification.Title = alert
Msg.Notification.Title = alert ctx := context.Background()
ctx := context.Background() for _, v := range result {
max := (i+1)*SinglePushCountLimit - 1 Msg.Tokens = v.Item
if max >= tokenlen {
max = tokenlen - 1
}
Msg.Tokens = Tokens[i*SinglePushCountLimit : max]
//SendMulticast sends the given multicast message to all the FCM registration tokens specified. //SendMulticast sends the given multicast message to all the FCM registration tokens specified.
//The tokens array in MulticastMessage may contain up to 500 tokens. //The tokens array in MulticastMessage may contain up to 500 tokens.
//SendMulticast uses the `SendAll()` function to send the given message to all the target recipients. //SendMulticast uses the `SendAll()` function to send the given message to all the target recipients.
@ -87,7 +79,9 @@ func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string,
//Partial failures are indicated by a `BatchResponse` return value. //Partial failures are indicated by a `BatchResponse` return value.
response, err := f.FcmMsgCli.SendMulticast(ctx, Msg) response, err := f.FcmMsgCli.SendMulticast(ctx, Msg)
if err != nil { if err != nil {
return "", err Fail = Fail + len(v.Item)
log.Info(operationID, "some token push err", err.Error(), len(v.Item))
continue
} }
Success = Success + response.SuccessCount Success = Success + response.SuccessCount
Fail = Fail + response.FailureCount Fail = Fail + response.FailureCount