refactor: send notification.

2017-07-24 21:17:11 +08:00
parent 41a8a609be
commit a64fae7de7
2 changed files with 13 additions and 14 deletions
--- a/gorush/worker.go
+++ b/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 {
-		case PlatFormIos:
-			PushToIOS(notification)
-		case PlatFormAndroid:
-			PushToAndroid(notification)
-		}
+		SendNotification(notification)
 	}
 }