1 files changed, 26 insertions, 0 deletions
diff --git a/src/main/java/com/juick/server/PushQueries.java b/src/main/java/com/juick/server/PushQueries.java
index 84f42722..bf7505ca 100644
--- a/src/main/java/com/juick/server/PushQueries.java
+++ b/src/main/java/com/juick/server/PushQueries.java
@@ -9,6 +9,7 @@ import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.ArrayList;
+import java.util.List;
 
 /**
  *
@@ -65,4 +66,29 @@ public class PushQueries {
         }
         return urls;
     }
+
+    public static String getAPNSToken(Connection sql, int uid) {
+        return SQLHelpers.getString(sql, "SELECT token from ios WHERE user_id=?", uid);
+    }
+
+    public static List<String> getAPNSSubscribers(Connection sql, int uid) {
+        List<String> urls = new ArrayList<String>();
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT token FROM ios INNER JOIN subscr_users ON (subscr_users.user_id=? AND ios.user_id=subscr_users.suser_id)");
+            stmt.setInt(1, uid);
+            rs = stmt.executeQuery();
+            rs.beforeFirst();
+            while (rs.next()) {
+                urls.add(rs.getString(1));
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+        return urls;
+    }
 }