refactor: send notification.

gorush/worker.go

@@ -13,15 +13,20 @@ func InitWorkers(workerNum int64, queueNum int64) {
 	}
 }
 
+// SendNotification is send message to iOS or Android
+func SendNotification(msg PushNotification) {
+	switch msg.Platform {
+	case PlatFormIos:
+		PushToIOS(msg)
+	case PlatFormAndroid:
+		PushToAndroid(msg)
+	}
+}
+
 func startWorker() {
 	for {
 		notification := <-QueueNotification
-		switch notification.Platform {
+		SendNotification(notification)
-		case PlatFormIos:
-			PushToIOS(notification)
-		case PlatFormAndroid:
-			PushToAndroid(notification)
-		}
 	}
 }
 

rpc/server.go

@@ -29,16 +29,10 @@ func (s *server) Send(ctx context.Context, in *pb.NotificationRequest) (*pb.Noti
 		APIKey:   in.Key,
 	}
 
-	success := false
+	gorush.SendNotification(notification)
-	switch notification.Platform {
-	case gorush.PlatFormIos:
-		success = gorush.PushToIOS(notification)
-	case gorush.PlatFormAndroid:
-		success = gorush.PushToAndroid(notification)
-	}
 
 	return &pb.NotificationReply{
-		Success: success,
+		Success: false,
 		Counts:  int32(len(notification.Tokens)),
 	}, nil
 }