coopgo-platform
/
gorush
7
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

refactor: remove config from notification struct. (#617) 

* refactor: remove config from notification struct.

* chore: update

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu 2021-08-01 19:48:57 +08:00 committed by GitHub
parent 132e1791cd
commit 8d03ac5f1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
10 changed files with 73 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.go
View File

@ -194,7 +194,6 @@ func main() {
if opts.Android.Enabled { if opts.Android.Enabled {
cfg.Android.Enabled = opts.Android.Enabled cfg.Android.Enabled = opts.Android.Enabled
req := notify.PushNotification{ req := notify.PushNotification{
Cfg: cfg,
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: message, Message: message,
Title: title, Title: title,
@ -219,7 +218,7 @@ func main() {
return return
} }
notify.PushToAndroid(req) notify.PushToAndroid(req, cfg)
return return
} }
@ -228,7 +227,6 @@ func main() {
if opts.Huawei.Enabled { if opts.Huawei.Enabled {
cfg.Huawei.Enabled = opts.Huawei.Enabled cfg.Huawei.Enabled = opts.Huawei.Enabled
req := notify.PushNotification{ req := notify.PushNotification{
Cfg: cfg,
Platform: core.PlatFormHuawei, Platform: core.PlatFormHuawei,
Message: message, Message: message,
Title: title, Title: title,
@ -253,7 +251,7 @@ func main() {
return return
} }
notify.PushToHuawei(req) notify.PushToHuawei(req, cfg)
return return
} }
@ -266,7 +264,6 @@ func main() {
cfg.Ios.Enabled = opts.Ios.Enabled cfg.Ios.Enabled = opts.Ios.Enabled
req := notify.PushNotification{ req := notify.PushNotification{
Cfg: cfg,
Platform: core.PlatFormIos, Platform: core.PlatFormIos,
Message: message, Message: message,
Title: title, Title: title,
@ -294,7 +291,7 @@ func main() {
if err := notify.InitAPNSClient(cfg); err != nil { if err := notify.InitAPNSClient(cfg); err != nil {
return return
} }
notify.PushToIOS(req) notify.PushToIOS(req, cfg)
return return
} }
@ -322,7 +319,7 @@ func main() {
case core.LocalQueue: case core.LocalQueue:
w = simple.NewWorker( w = simple.NewWorker(
simple.WithQueueNum(int(cfg.Core.QueueNum)), simple.WithQueueNum(int(cfg.Core.QueueNum)),
simple.WithRunFunc(notify.Run), simple.WithRunFunc(notify.Run(cfg)),
) )
case core.NSQ: case core.NSQ:
w = nsq.NewWorker( w = nsq.NewWorker(
@ -330,7 +327,7 @@ func main() {
nsq.WithTopic(cfg.Queue.NSQ.Topic), nsq.WithTopic(cfg.Queue.NSQ.Topic),
nsq.WithChannel(cfg.Queue.NSQ.Channel), nsq.WithChannel(cfg.Queue.NSQ.Channel),
nsq.WithMaxInFlight(int(cfg.Core.WorkerNum)), nsq.WithMaxInFlight(int(cfg.Core.WorkerNum)),
nsq.WithRunFunc(notify.Run), nsq.WithRunFunc(notify.Run(cfg)),
) )
default: default:
logx.LogError.Fatalf("we don't support queue engine: %s", cfg.Queue.Engine) logx.LogError.Fatalf("we don't support queue engine: %s", cfg.Queue.Engine)

16
notify/notification.go
View File

@ -65,8 +65,6 @@ type ResponsePush struct {
// PushNotification is single notification request // PushNotification is single notification request
type PushNotification struct { type PushNotification struct {
Cfg config.ConfYaml
// Common // Common
ID string `json:"notif_id,omitempty"` ID string `json:"notif_id,omitempty"`
Tokens []string `json:"tokens" binding:"required"` Tokens []string `json:"tokens" binding:"required"`
@ -235,7 +233,7 @@ func CheckPushConf(cfg config.ConfYaml) error {
} }
// SendNotification send notification // SendNotification send notification
func SendNotification(req queue.QueuedMessage) (resp *ResponsePush, err error) { func SendNotification(req queue.QueuedMessage, cfg config.ConfYaml) (resp *ResponsePush, err error) {
v, ok := req.(*PushNotification) v, ok := req.(*PushNotification)
if !ok { if !ok {
if err = json.Unmarshal(req.Bytes(), &v); err != nil { if err = json.Unmarshal(req.Bytes(), &v); err != nil {
@ -245,18 +243,20 @@ func SendNotification(req queue.QueuedMessage) (resp *ResponsePush, err error) {
switch v.Platform { switch v.Platform {
case core.PlatFormIos: case core.PlatFormIos:
resp, err = PushToIOS(*v) resp, err = PushToIOS(*v, cfg)
case core.PlatFormAndroid: case core.PlatFormAndroid:
resp, err = PushToAndroid(*v) resp, err = PushToAndroid(*v, cfg)
case core.PlatFormHuawei: case core.PlatFormHuawei:
resp, err = PushToHuawei(*v) resp, err = PushToHuawei(*v, cfg)
} }
return return
} }
// Run send notification // Run send notification
var Run = func(ctx context.Context, msg queue.QueuedMessage) error { var Run = func(cfg config.ConfYaml) func(ctx context.Context, msg queue.QueuedMessage) error {
_, err := SendNotification(msg) return func(ctx context.Context, msg queue.QueuedMessage) error {
_, err := SendNotification(msg, cfg)
return err return err
}
} }

20
notify/notification_apns.go
View File

@ -388,16 +388,16 @@ func getApnsClient(cfg config.ConfYaml, req PushNotification) (client *apns2.Cli
} }
// PushToIOS provide send notification to APNs server. // PushToIOS provide send notification to APNs server.
func PushToIOS(req PushNotification) (resp *ResponsePush, err error) { func PushToIOS(req PushNotification, cfg config.ConfYaml) (resp *ResponsePush, err error) {
logx.LogAccess.Debug("Start push notification for iOS") logx.LogAccess.Debug("Start push notification for iOS")
if req.Cfg.Core.Sync && !core.IsLocalQueue(core.Queue(req.Cfg.Queue.Engine)) { if cfg.Core.Sync && !core.IsLocalQueue(core.Queue(cfg.Queue.Engine)) {
req.Cfg.Core.Sync = false cfg.Core.Sync = false
} }
var ( var (
retryCount = 0 retryCount = 0
maxRetry = req.Cfg.Ios.MaxRetry maxRetry = cfg.Ios.MaxRetry
) )
if req.Retry > 0 && req.Retry < maxRetry { if req.Retry > 0 && req.Retry < maxRetry {
@ -410,7 +410,7 @@ Retry:
var newTokens []string var newTokens []string
notification := GetIOSNotification(req) notification := GetIOSNotification(req)
client := getApnsClient(req.Cfg, req) client := getApnsClient(cfg, req)
var wg sync.WaitGroup var wg sync.WaitGroup
for _, token := range req.Tokens { for _, token := range req.Tokens {
@ -430,16 +430,16 @@ Retry:
} }
// apns server error // apns server error
errLog := logPush(req.Cfg, core.FailedPush, token, req, err) errLog := logPush(cfg, core.FailedPush, token, req, err)
if req.Cfg.Core.Sync { if cfg.Core.Sync {
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
} else if req.Cfg.Core.FeedbackURL != "" { } else if cfg.Core.FeedbackURL != "" {
go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) { go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) {
err := DispatchFeedback(log, url, timeout) err := DispatchFeedback(log, url, timeout)
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
} }
}(logx.LogError, errLog, req.Cfg.Core.FeedbackURL, req.Cfg.Core.FeedbackTimeout) }(logx.LogError, errLog, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
} }
status.StatStorage.AddIosError(1) status.StatStorage.AddIosError(1)
@ -451,7 +451,7 @@ Retry:
} }
if res != nil && res.Sent() { if res != nil && res.Sent() {
logPush(req.Cfg, core.SucceededPush, token, req, nil) logPush(cfg, core.SucceededPush, token, req, nil)
status.StatStorage.AddIosSuccess(1) status.StatStorage.AddIosSuccess(1)
} }

3
notify/notification_apns_test.go
View File

@ -726,14 +726,13 @@ func TestPushToIOS(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
req := PushNotification{ req := PushNotification{
Cfg: cfg,
Tokens: []string{"11aa01229f15f0f0c52029d8cf8cd0aeaf2365fe4cebc4af26cd6d76b7919ef7", "11aa01229f15f0f0c52029d8cf8cd0aeaf2365fe4cebc4af26cd6d76b7919ef1"}, Tokens: []string{"11aa01229f15f0f0c52029d8cf8cd0aeaf2365fe4cebc4af26cd6d76b7919ef7", "11aa01229f15f0f0c52029d8cf8cd0aeaf2365fe4cebc4af26cd6d76b7919ef1"},
Platform: 1, Platform: 1,
Message: "Welcome", Message: "Welcome",
} }
// send fail // send fail
PushToIOS(req) PushToIOS(req, cfg)
} }
func TestApnsHostFromRequest(t *testing.T) { func TestApnsHostFromRequest(t *testing.T) {

48
notify/notification_fcm.go
View File

@ -106,17 +106,17 @@ func GetAndroidNotification(req PushNotification) *fcm.Message {
} }
// PushToAndroid provide send notification to Android server. // PushToAndroid provide send notification to Android server.
func PushToAndroid(req PushNotification) (resp *ResponsePush, err error) { func PushToAndroid(req PushNotification, cfg config.ConfYaml) (resp *ResponsePush, err error) {
logx.LogAccess.Debug("Start push notification for Android") logx.LogAccess.Debug("Start push notification for Android")
if req.Cfg.Core.Sync && !core.IsLocalQueue(core.Queue(req.Cfg.Queue.Engine)) { if cfg.Core.Sync && !core.IsLocalQueue(core.Queue(cfg.Queue.Engine)) {
req.Cfg.Core.Sync = false cfg.Core.Sync = false
} }
var ( var (
client *fcm.Client client *fcm.Client
retryCount = 0 retryCount = 0
maxRetry = req.Cfg.Android.MaxRetry maxRetry = cfg.Android.MaxRetry
) )
if req.Retry > 0 && req.Retry < maxRetry { if req.Retry > 0 && req.Retry < maxRetry {
@ -136,9 +136,9 @@ Retry:
notification := GetAndroidNotification(req) notification := GetAndroidNotification(req)
if req.APIKey != "" { if req.APIKey != "" {
client, err = InitFCMClient(req.Cfg, req.APIKey) client, err = InitFCMClient(cfg, req.APIKey)
} else { } else {
client, err = InitFCMClient(req.Cfg, req.Cfg.Android.APIKey) client, err = InitFCMClient(cfg, cfg.Android.APIKey)
} }
if err != nil { if err != nil {
@ -153,30 +153,30 @@ Retry:
logx.LogError.Error("FCM server send message error: " + err.Error()) logx.LogError.Error("FCM server send message error: " + err.Error())
if req.IsTopic() { if req.IsTopic() {
errLog := logPush(req.Cfg, core.FailedPush, req.To, req, err) errLog := logPush(cfg, core.FailedPush, req.To, req, err)
if req.Cfg.Core.Sync { if cfg.Core.Sync {
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
} else if req.Cfg.Core.FeedbackURL != "" { } else if cfg.Core.FeedbackURL != "" {
go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) { go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) {
err := DispatchFeedback(log, url, timeout) err := DispatchFeedback(log, url, timeout)
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
} }
}(logx.LogError, errLog, req.Cfg.Core.FeedbackURL, req.Cfg.Core.FeedbackTimeout) }(logx.LogError, errLog, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
} }
status.StatStorage.AddAndroidError(1) status.StatStorage.AddAndroidError(1)
} else { } else {
for _, token := range req.Tokens { for _, token := range req.Tokens {
errLog := logPush(req.Cfg, core.FailedPush, token, req, err) errLog := logPush(cfg, core.FailedPush, token, req, err)
if req.Cfg.Core.Sync { if cfg.Core.Sync {
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
} else if req.Cfg.Core.FeedbackURL != "" { } else if cfg.Core.FeedbackURL != "" {
go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) { go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) {
err := DispatchFeedback(log, url, timeout) err := DispatchFeedback(log, url, timeout)
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
} }
}(logx.LogError, errLog, req.Cfg.Core.FeedbackURL, req.Cfg.Core.FeedbackTimeout) }(logx.LogError, errLog, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
} }
} }
status.StatStorage.AddAndroidError(int64(len(req.Tokens))) status.StatStorage.AddAndroidError(int64(len(req.Tokens)))
@ -208,21 +208,21 @@ Retry:
newTokens = append(newTokens, to) newTokens = append(newTokens, to)
} }
errLog := logPush(req.Cfg, core.FailedPush, to, req, result.Error) errLog := logPush(cfg, core.FailedPush, to, req, result.Error)
if req.Cfg.Core.Sync { if cfg.Core.Sync {
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
} else if req.Cfg.Core.FeedbackURL != "" { } else if cfg.Core.FeedbackURL != "" {
go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) { go func(logger *logrus.Logger, log logx.LogPushEntry, url string, timeout int64) {
err := DispatchFeedback(log, url, timeout) err := DispatchFeedback(log, url, timeout)
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
} }
}(logx.LogError, errLog, req.Cfg.Core.FeedbackURL, req.Cfg.Core.FeedbackTimeout) }(logx.LogError, errLog, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
} }
continue continue
} }
logPush(req.Cfg, core.SucceededPush, to, req, nil) logPush(cfg, core.SucceededPush, to, req, nil)
} }
// result from Send messages to topics // result from Send messages to topics
@ -236,11 +236,11 @@ Retry:
logx.LogAccess.Debug("Send Topic Message: ", to) logx.LogAccess.Debug("Send Topic Message: ", to)
// Success // Success
if res.MessageID != 0 { if res.MessageID != 0 {
logPush(req.Cfg, core.SucceededPush, to, req, nil) logPush(cfg, core.SucceededPush, to, req, nil)
} else { } else {
// failure // failure
errLog := logPush(req.Cfg, core.FailedPush, to, req, res.Error) errLog := logPush(cfg, core.FailedPush, to, req, res.Error)
if req.Cfg.Core.Sync { if cfg.Core.Sync {
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
} }
} }
@ -250,8 +250,8 @@ Retry:
if len(res.FailedRegistrationIDs) > 0 { if len(res.FailedRegistrationIDs) > 0 {
newTokens = append(newTokens, res.FailedRegistrationIDs...) newTokens = append(newTokens, res.FailedRegistrationIDs...)
errLog := logPush(req.Cfg, core.FailedPush, notification.To, req, errors.New("device group: partial success or all fails")) errLog := logPush(cfg, core.FailedPush, notification.To, req, errors.New("device group: partial success or all fails"))
if req.Cfg.Core.Sync { if cfg.Core.Sync {
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
} }
} }

15
notify/notification_fcm_test.go
View File

@ -40,14 +40,13 @@ func TestPushToAndroidWrongToken(t *testing.T) {
cfg.Android.APIKey = os.Getenv("ANDROID_API_KEY") cfg.Android.APIKey = os.Getenv("ANDROID_API_KEY")
req := PushNotification{ req := PushNotification{
Cfg: cfg,
Tokens: []string{"aaaaaa", "bbbbb"}, Tokens: []string{"aaaaaa", "bbbbb"},
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: "Welcome", Message: "Welcome",
} }
// Android Success count: 0, Failure count: 2 // Android Success count: 0, Failure count: 2
PushToAndroid(req) PushToAndroid(req, cfg)
} }
func TestPushToAndroidRightTokenForJSONLog(t *testing.T) { func TestPushToAndroidRightTokenForJSONLog(t *testing.T) {
@ -61,13 +60,12 @@ func TestPushToAndroidRightTokenForJSONLog(t *testing.T) {
androidToken := os.Getenv("ANDROID_TEST_TOKEN") androidToken := os.Getenv("ANDROID_TEST_TOKEN")
req := PushNotification{ req := PushNotification{
Cfg: cfg,
Tokens: []string{androidToken}, Tokens: []string{androidToken},
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: "Welcome", Message: "Welcome",
} }
PushToAndroid(req) PushToAndroid(req, cfg)
} }
func TestPushToAndroidRightTokenForStringLog(t *testing.T) { func TestPushToAndroidRightTokenForStringLog(t *testing.T) {
@ -79,13 +77,12 @@ func TestPushToAndroidRightTokenForStringLog(t *testing.T) {
androidToken := os.Getenv("ANDROID_TEST_TOKEN") androidToken := os.Getenv("ANDROID_TEST_TOKEN")
req := PushNotification{ req := PushNotification{
Cfg: cfg,
Tokens: []string{androidToken}, Tokens: []string{androidToken},
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: "Welcome", Message: "Welcome",
} }
PushToAndroid(req) PushToAndroid(req, cfg)
} }
func TestOverwriteAndroidAPIKey(t *testing.T) { func TestOverwriteAndroidAPIKey(t *testing.T) {
@ -98,7 +95,6 @@ func TestOverwriteAndroidAPIKey(t *testing.T) {
androidToken := os.Getenv("ANDROID_TEST_TOKEN") androidToken := os.Getenv("ANDROID_TEST_TOKEN")
req := PushNotification{ req := PushNotification{
Cfg: cfg,
Tokens: []string{androidToken, "bbbbb"}, Tokens: []string{androidToken, "bbbbb"},
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: "Welcome", Message: "Welcome",
@ -107,7 +103,7 @@ func TestOverwriteAndroidAPIKey(t *testing.T) {
} }
// FCM server error: 401 error: 401 Unauthorized (Wrong API Key) // FCM server error: 401 error: 401 Unauthorized (Wrong API Key)
resp, err := PushToAndroid(req) resp, err := PushToAndroid(req, cfg)
assert.Error(t, err) assert.Error(t, err)
assert.Len(t, resp.Logs, 2) assert.Len(t, resp.Logs, 2)
@ -199,14 +195,13 @@ func TestCheckAndroidMessage(t *testing.T) {
timeToLive := uint(2419201) timeToLive := uint(2419201)
req := PushNotification{ req := PushNotification{
Cfg: cfg,
Tokens: []string{"aaaaaa", "bbbbb"}, Tokens: []string{"aaaaaa", "bbbbb"},
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: "Welcome", Message: "Welcome",
TimeToLive: &timeToLive, TimeToLive: &timeToLive,
} }
PushToAndroid(req) PushToAndroid(req, cfg)
} }
func TestAndroidNotificationStructure(t *testing.T) { func TestAndroidNotificationStructure(t *testing.T) {

12
notify/notification_hms.go
View File

@ -166,17 +166,17 @@ func GetHuaweiNotification(req PushNotification) (*model.MessageRequest, error)
} }
// PushToHuawei provide send notification to Android server. // PushToHuawei provide send notification to Android server.
func PushToHuawei(req PushNotification) (resp *ResponsePush, err error) { func PushToHuawei(req PushNotification, cfg config.ConfYaml) (resp *ResponsePush, err error) {
logx.LogAccess.Debug("Start push notification for Huawei") logx.LogAccess.Debug("Start push notification for Huawei")
if req.Cfg.Core.Sync && !core.IsLocalQueue(core.Queue(req.Cfg.Queue.Engine)) { if cfg.Core.Sync && !core.IsLocalQueue(core.Queue(cfg.Queue.Engine)) {
req.Cfg.Core.Sync = false cfg.Core.Sync = false
} }
var ( var (
client *client.HMSClient client *client.HMSClient
retryCount = 0 retryCount = 0
maxRetry = req.Cfg.Huawei.MaxRetry maxRetry = cfg.Huawei.MaxRetry
) )
if req.Retry > 0 && req.Retry < maxRetry { if req.Retry > 0 && req.Retry < maxRetry {
@ -190,7 +190,7 @@ func PushToHuawei(req PushNotification) (resp *ResponsePush, err error) {
return return
} }
client, err = InitHMSClient(req.Cfg, req.Cfg.Huawei.AppSecret, req.Cfg.Huawei.AppID) client, err = InitHMSClient(cfg, cfg.Huawei.AppSecret, cfg.Huawei.AppID)
if err != nil { if err != nil {
// HMS server error // HMS server error
@ -208,7 +208,7 @@ Retry:
res, err := client.SendMessage(context.Background(), notification) res, err := client.SendMessage(context.Background(), notification)
if err != nil { if err != nil {
// Send Message error // Send Message error
errLog := logPush(req.Cfg, core.FailedPush, req.To, req, err) errLog := logPush(cfg, core.FailedPush, req.To, req, err)
resp.Logs = append(resp.Logs, errLog) resp.Logs = append(resp.Logs, errLog)
logx.LogError.Error("HMS server send message error: " + err.Error()) logx.LogError.Error("HMS server send message error: " + err.Error())
return return

7
router/server.go
View File

@ -273,7 +273,6 @@ func handleNotification(ctx context.Context, cfg config.ConfYaml, req notify.Req
continue continue
} }
} }
notification.Cfg = cfg
newNotification = append(newNotification, notification) newNotification = append(newNotification, notification)
} }
@ -284,10 +283,10 @@ func handleNotification(ctx context.Context, cfg config.ConfYaml, req notify.Req
} }
if core.IsLocalQueue(core.Queue(cfg.Queue.Engine)) && cfg.Core.Sync { if core.IsLocalQueue(core.Queue(cfg.Queue.Engine)) && cfg.Core.Sync {
func(msg *notify.PushNotification) { func(msg *notify.PushNotification, cfg config.ConfYaml) {
q.QueueTask(func(ctx context.Context) error { q.QueueTask(func(ctx context.Context) error {
defer wg.Done() defer wg.Done()
resp, err := notify.SendNotification(msg) resp, err := notify.SendNotification(msg, cfg)
if err != nil { if err != nil {
return err return err
} }
@ -299,7 +298,7 @@ func handleNotification(ctx context.Context, cfg config.ConfYaml, req notify.Req
return nil return nil
}) })
}(notification) }(notification, cfg)
} else if err := q.Queue(notification); err != nil { } else if err := q.Queue(notification); err != nil {
resp := markFailedNotification(cfg, notification, "max capacity reached") resp := markFailedNotification(cfg, notification, "max capacity reached")
// add log // add log

12
router/server_test.go
View File

@ -36,10 +36,17 @@ func TestMain(m *testing.M) {
log.Fatal(err) log.Fatal(err)
} }
cfg.Android.Enabled = true
cfg.Android.APIKey = os.Getenv("ANDROID_API_KEY")
if _, err := notify.InitFCMClient(cfg, ""); err != nil {
log.Fatal(err)
}
w = simple.NewWorker( w = simple.NewWorker(
simple.WithRunFunc(func(ctx context.Context, msg queue.QueuedMessage) error { simple.WithRunFunc(func(ctx context.Context, msg queue.QueuedMessage) error {
notify.SendNotification(msg) _, err := notify.SendNotification(msg, cfg)
return nil return err
}), }),
) )
q, _ = queue.NewQueue( q, _ = queue.NewQueue(
@ -600,7 +607,6 @@ func TestSyncModeForTopicNotification(t *testing.T) {
// android // android
{ {
// success // success
Cfg: cfg,
Condition: "'dogs' in topics || 'cats' in topics", Condition: "'dogs' in topics || 'cats' in topics",
Platform: core.PlatFormAndroid, Platform: core.PlatFormAndroid,
Message: "This is a Firebase Cloud Messaging Topic Message!", Message: "This is a Firebase Cloud Messaging Topic Message!",

3
rpc/server.go
View File

@ -56,7 +56,6 @@ func (s *Server) Check(ctx context.Context, in *proto.HealthCheckRequest) (*prot
func (s *Server) Send(ctx context.Context, in *proto.NotificationRequest) (*proto.NotificationReply, error) { func (s *Server) Send(ctx context.Context, in *proto.NotificationRequest) (*proto.NotificationReply, error) {
badge := int(in.Badge) badge := int(in.Badge)
notification := notify.PushNotification{ notification := notify.PushNotification{
Cfg: s.cfg,
Platform: int(in.Platform), Platform: int(in.Platform),
Tokens: in.Tokens, Tokens: in.Tokens,
Message: in.Message, Message: in.Message,
@ -102,7 +101,7 @@ func (s *Server) Send(ctx context.Context, in *proto.NotificationRequest) (*prot
} }
} }
go notify.SendNotification(&notification) go notify.SendNotification(&notification, s.cfg)
return &proto.NotificationReply{ return &proto.NotificationReply{
Success: true, Success: true,