db

1 files changed, 15 insertions, 4 deletions

diff --git a/lib/okpush/db.js b/lib/okpush/db.js
index 35a4c86e..4e6c2ee1 100644
--- a/lib/okpush/db.js
+++ b/lib/okpush/db.js
@@ -19,6 +19,7 @@ function init (config) {
     token: {
       type: 'String',
       required: true,
+      unique: true,
     },
     channel: {
       type: 'String',
@@ -51,9 +52,9 @@ function init (config) {
     }
   })
 
-  PushToken = db.model('PushToken', pushTokenSchema)
-  Notification = db.model('Notification', notificationSchema)
-  Channel = db.model('Channel', channelSchema)
+  module.exports.PushToken = PushToken = db.model('PushToken', pushTokenSchema)
+  module.exports.Notification = Notification = db.model('Notification', notificationSchema)
+  module.exports.Channel = Channel = db.model('Channel', channelSchema)
 }
 
 function errorHandler (error) {
@@ -74,6 +75,15 @@ function getAllIOSTokens (channel, cb) {
     return cb(null, items)
   })
 }
+function getAllAndroidTokens (channel, cb) {
+  PushToken.find({ channel: channel, platform: 'android' }, function (err, items) {
+    if (err) return cb(err, null)
+    var items = _.map(items, function (item) {
+      return item.token
+    })
+    return cb(null, items)
+  })
+}
 function getAllTokens (channel, cb) {
   PushToken.find({ channel: channel }, function (err, items) {
     if (err) return cb(err, null)
@@ -139,12 +149,13 @@ function getNotifications (cb) {
 
 /* wrap functions for some reason */
 
-module.exports = {
+var exports = module.exports = {
   init: init,
   addToken: addToken,
   removeToken: removeToken,
   getAllTokens: getAllTokens,
   getAllIOSTokens: getAllIOSTokens,
+  getAllAndroidTokens: getAllAndroidTokens,
   getDeviceCount: getDeviceCount,
   addNotification: addNotification,
   getNotifications: getNotifications,