mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
package fcm
|
||
|
||
import (
|
||
"Open_IM/internal/push"
|
||
"Open_IM/pkg/common/config"
|
||
"Open_IM/pkg/common/db"
|
||
"Open_IM/pkg/common/log"
|
||
"context"
|
||
firebase "firebase.google.com/go"
|
||
"firebase.google.com/go/messaging"
|
||
"google.golang.org/api/option"
|
||
"path/filepath"
|
||
"strconv"
|
||
)
|
||
|
||
type Fcm struct {
|
||
FcmMsgCli *messaging.Client
|
||
}
|
||
|
||
func NewFcm() *Fcm {
|
||
return newFcmClient()
|
||
}
|
||
func newFcmClient() *Fcm {
|
||
opt := option.WithCredentialsFile(filepath.Join(config.Root, "config", config.Config.Push.Fcm.ServiceAccount))
|
||
fcmApp, err := firebase.NewApp(context.Background(), nil, opt)
|
||
if err != nil {
|
||
log.Debug("", "error initializing app: ", err.Error())
|
||
return nil
|
||
}
|
||
//授权
|
||
// fcmClient, err := fcmApp.Auth(context.Background())
|
||
// if err != nil {
|
||
// log.Println("error getting Auth client: %v\n", err)
|
||
// return
|
||
// }
|
||
// log.Printf("%#v\r\n", fcmClient)
|
||
ctx := context.Background()
|
||
fcmMsgClient, err := fcmApp.Messaging(ctx)
|
||
if err != nil {
|
||
panic(err.Error())
|
||
return nil
|
||
}
|
||
return &Fcm{FcmMsgCli: fcmMsgClient}
|
||
}
|
||
|
||
func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string, opts push.PushOpts) (string, error) {
|
||
//需要一个客户端的Token
|
||
// accounts->registrationToken
|
||
Tokens := make([]string, 0)
|
||
for _, account := range accounts {
|
||
IosfcmToken, IosErr := db.DB.GetFcmToken(account, 1)
|
||
AndroidfcmToken, AndroidErr := db.DB.GetFcmToken(account, 2)
|
||
|
||
if IosErr == nil {
|
||
Tokens = append(Tokens, IosfcmToken)
|
||
}
|
||
if AndroidErr == nil {
|
||
Tokens = append(Tokens, AndroidfcmToken)
|
||
}
|
||
}
|
||
tokenlen := len(Tokens)
|
||
// 500组为一个推送，我们用400好了
|
||
limit := 400
|
||
pages := int((tokenlen-1)/limit + 1)
|
||
Success := 0
|
||
Fail := 0
|
||
for i := 0; i < pages; i++ {
|
||
Msg := new(messaging.MulticastMessage)
|
||
Msg.Notification = &messaging.Notification{}
|
||
Msg.Notification.Body = detailContent
|
||
Msg.Notification.Title = alert
|
||
ctx := context.Background()
|
||
max := (i+1)*limit - 1
|
||
if max >= tokenlen {
|
||
max = tokenlen - 1
|
||
}
|
||
Msg.Tokens = Tokens[i*limit : max]
|
||
//SendMulticast sends the given multicast message to all the FCM registration tokens specified.
|
||
//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.
|
||
//The responses list obtained from the return value corresponds to the order of the input tokens.
|
||
//An error from SendMulticast indicates a total failure -- i.e.
|
||
//the message could not be sent to any of the recipients.
|
||
//Partial failures are indicated by a `BatchResponse` return value.
|
||
response, err := f.FcmMsgCli.SendMulticast(ctx, Msg)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
Success = Success + response.SuccessCount
|
||
Fail = Fail + response.FailureCount
|
||
}
|
||
return strconv.Itoa(Success) + " Success," + strconv.Itoa(Fail) + " Fail", nil
|
||
}
|