2 files changed, 46 insertions, 8 deletions
diff --git a/juick-server/src/main/java/com/juick/service/PushQueriesService.java b/juick-server/src/main/java/com/juick/service/PushQueriesService.java
index 2e2ac3901..7d4bc295f 100644
--- a/juick-server/src/main/java/com/juick/service/PushQueriesService.java
+++ b/juick-server/src/main/java/com/juick/service/PushQueriesService.java
@@ -7,17 +7,27 @@ import java.util.List;
  * Created by aalexeev on 11/13/16.
  */
 public interface PushQueriesService {
-    List<String> getAndroidRegID(int uid);
+    List<String> getGCMRegID(int uid);
 
-    List<String> getAndroidTokens(Collection<Integer> uids);
+    List<String> getGCMTokens(Collection<Integer> uids);
 
-    List<String> getWinPhoneURL(int uid);
+    boolean addGCMToken(Integer uid, String token);
 
-    List<String> getWindowsTokens(Collection<Integer> uids);
+    boolean deleteGCMToken(String token);
+
+    List<String> getMPNSURL(int uid);
+
+    List<String> getMPNSTokens(Collection<Integer> uids);
+
+    boolean addMPNSToken(Integer uid, String token);
+
+    boolean deleteMPNSToken(String token);
 
     List<String> getAPNSToken(int uid);
 
     List<String> getAPNSTokens(Collection<Integer> uids);
 
+    boolean addAPNSToken(Integer uid, String token);
+
     boolean deleteAPNSToken(String token);
 }
diff --git a/juick-server/src/main/java/com/juick/service/PushQueriesServiceImpl.java b/juick-server/src/main/java/com/juick/service/PushQueriesServiceImpl.java
index b11f0a795..aca48a197 100644
--- a/juick-server/src/main/java/com/juick/service/PushQueriesServiceImpl.java
+++ b/juick-server/src/main/java/com/juick/service/PushQueriesServiceImpl.java
@@ -24,7 +24,7 @@ public class PushQueriesServiceImpl extends BaseJdbcService implements PushQueri
     }
 
     @Override
-    public List<String> getAndroidRegID(final int uid) {
+    public List<String> getGCMRegID(final int uid) {
         return getJdbcTemplate().queryForList(
                 "SELECT regid FROM android WHERE user_id=?",
                 String.class,
@@ -32,7 +32,7 @@ public class PushQueriesServiceImpl extends BaseJdbcService implements PushQueri
     }
 
     @Override
-    public List<String> getAndroidTokens(final Collection<Integer> uids) {
+    public List<String> getGCMTokens(final Collection<Integer> uids) {
         if (CollectionUtils.isEmpty(uids))
             return Collections.emptyList();
 
@@ -43,7 +43,18 @@ public class PushQueriesServiceImpl extends BaseJdbcService implements PushQueri
     }
 
     @Override
-    public List<String> getWinPhoneURL(final int uid) {
+    public boolean addGCMToken(Integer uid, String token) {
+        return getJdbcTemplate().update("INSERT IGNORE INTO android(user_id,regid) VALUES (?, ?)",
+                uid, token) > 0;
+    }
+
+    @Override
+    public boolean deleteGCMToken(String token) {
+        return getJdbcTemplate().update("DELETE FROM android WHERE regid=?", token) > 0;
+    }
+
+    @Override
+    public List<String> getMPNSURL(final int uid) {
         return getJdbcTemplate().queryForList(
                 "SELECT url FROM winphone WHERE user_id=?",
                 String.class,
@@ -51,7 +62,7 @@ public class PushQueriesServiceImpl extends BaseJdbcService implements PushQueri
     }
 
     @Override
-    public List<String> getWindowsTokens(final Collection<Integer> uids) {
+    public List<String> getMPNSTokens(final Collection<Integer> uids) {
         if (CollectionUtils.isEmpty(uids))
             return Collections.emptyList();
 
@@ -62,6 +73,17 @@ public class PushQueriesServiceImpl extends BaseJdbcService implements PushQueri
     }
 
     @Override
+    public boolean addMPNSToken(Integer uid, String token) {
+        return getJdbcTemplate().update("INSERT IGNORE INTO winphone(user_id,url) VALUES (?, ?)",
+                uid, token) > 0;
+    }
+
+    @Override
+    public boolean deleteMPNSToken(String token) {
+        return getJdbcTemplate().update("DELETE FROM winphone WHERE url=?", token) > 0;
+    }
+
+    @Override
     public List<String> getAPNSToken(final int uid) {
         return getJdbcTemplate().queryForList(
                 "SELECT token from ios WHERE user_id=?",
@@ -84,4 +106,10 @@ public class PushQueriesServiceImpl extends BaseJdbcService implements PushQueri
                 new MapSqlParameterSource("ids", uids),
                 String.class);
     }
+
+    @Override
+    public boolean addAPNSToken(Integer uid, String token) {
+        return getJdbcTemplate().update("INSERT IGNORE INTO ios(user_id,token) VALUES (?, ?)",
+                uid, token) > 0;
+    }
 }