Changeset 8867 for trunk/greenstone3-extensions/gsdl-as/src/org/greenstone/gsdlas/ProfileStore.java

Timestamp:

2004-12-21T12:11:00+13:00 (19 years ago)

Author:

schweer

Message:

fixed BTS 12: restore subscriptions/predicates from database

File:

• trunk/greenstone3-extensions/gsdl-as/src/org/greenstone/gsdlas/ProfileStore.java (modified) (4 diffs)

trunk/greenstone3-extensions/gsdl-as/src/org/greenstone/gsdlas/ProfileStore.java

@@ -54 +54 @@
      */
     private void addSubscription(Subscription subscription) {
-        // TODO change to lists
         List documentIdPredicate = subscription.getPredicateList("documentID");
         List collectionIdPredicate = subscription.getPredicateList("collectionID");
@@ -237 +236 @@
         return matchedSubscriptions;
     }
-//
-//    public String toString() {
-//        Set allSubscriptions = getAllSubscriptions();
-//        
-//        StringBuffer buffer = new StringBuffer();
-//        
-//        for (Iterator iter = allSubscriptions.iterator(); iter.hasNext();) {
-//            Subscription sub = (Subscription) iter.next();
-//            buffer.append(sub);
-//            buffer.append("\n");
-//        }
-//        
-//        return buffer.toString();
-//    }
-//
-//    /**
-//     * @return
-//     */
-//    public Set getAllSubscriptions() {
-//        Set allSubscriptions = new TreeSet();
-//        allSubscriptions.addAll(noEqualsSubscriptions);
-//        
-//        for(Iterator iter = docIdEqualsSubscriptions.values().iterator(); iter.hasNext();) {
-//            Set values = (Set) iter.next();
-//            allSubscriptions.addAll(values);
-//        }
-//        
-//        for(Iterator iter = docCollectionIdEqualsSubscriptions.values().iterator(); iter.hasNext();) {
-//            Set values = (Set) iter.next();
-//            allSubscriptions.addAll(values);
-//        }
-//        
-//        for(Iterator iter = collectionIdEqualsSubscriptions.values().iterator(); iter.hasNext();) {
-//            Set values = (Set) iter.next();
-//            allSubscriptions.addAll(values);
-//        }
-//        return Collections.unmodifiableSet(allSubscriptions);
-//    }
 
 
@@ -289 +250 @@
     /**
      * @param subscriptionID
-     */
-    public void deleteSubscription(String subscriptionID) {
-        // delete row from subscriptions table 
-        // for each predicate:
-        // delete row from subs_to_predicates
-        // if there aren't any other subscriptions using this predicate, delete predicate
-        // TODO Auto-generated method stub
+     * @throws DatabaseException
+     * @throws SQLException
+     */
+    public void deleteSubscription(String subscriptionID) throws DatabaseException, SQLException {
+        Integer subID = new Integer(subscriptionID);
+        System.out.println("deleting subscription " + subscriptionID);
+        if (!subscriptions.containsKey(subID)) {
+            return;
+        }
+        Subscription sub = (Subscription) subscriptions.get(subID);
+        for (Iterator iter = dontCareSubscriptions.values().iterator(); iter.hasNext();) {
+            Set entry = (Set) iter.next();
+            if (entry.contains(subID))
+                entry.remove(subID);
+        }
+        if (noEqualsSubscriptions.contains(subID))
+            noEqualsSubscriptions.remove(subID);
+        for (Iterator iter = sub.getPredicates().iterator(); iter.hasNext();) {
+            Predicate pred = (Predicate) iter.next();
+            pred.removeSubscription(subID);
+        }
+
+        subscriptions.remove(subID);
+        
+        
+        Connection conn = DatabaseManager.getInstance().getDatabaseConnection();
+        
+        Statement statement = conn.createStatement();
+        
+        String sqlString = "DELETE FROM subscriptions WHERE id = " + subscriptionID;
+        statement.executeUpdate(sqlString);
+        
+        sqlString = "SELECT predicate FROM subs_to_predicates " +
+                "WHERE subscription = " + subscriptionID;
+        ResultSet predicates = statement.executeQuery(sqlString);
+        while (predicates.next()) {
+            // if there aren't any other subscriptions using this predicate, delete predicate
+            int predicateID = predicates.getInt("predicate");
+            
+        }
+        
         
     }
@@ -344 +339 @@
         return result;
     }
+
+    /**
+     * 
+     */
+    public void restoreFromDatabase() {
+        try {
+            Connection conn = DatabaseManager.getInstance().getDatabaseConnection();
+            Statement statement = conn.createStatement();
+            String sqlString = "SELECT * FROM subscriptions";
+            ResultSet subs = statement.executeQuery(sqlString);
+            while (subs.next()) {
+                // construct map for all predicates
+                int id = subs.getInt("id");
+                Map map = new TreeMap();
+                map.put("username", subs.getString("user"));
+                map.put("email", subs.getString("email"));
+                map.put("subscription_name", subs.getString("name"));
+                Vector ways = new Vector();
+                if (subs.getInt("rss") != 0) {
+                    ways.add("rss");
+                }
+                if (subs.getInt("page") != 0) {
+                    ways.add("page");
+                }
+                map.put("way", ways);
+                Statement stmnt = conn.createStatement();
+                sqlString = "SELECT p.type, p.field, p.value from subs_to_predicates stp join predicates p on(stp.predicate = p.id) where stp.subscription = " + id + ";";
+                ResultSet predicates = stmnt.executeQuery(sqlString);
+                while (predicates.next()) {
+                    String type = predicates.getString("type");
+                    String field = predicates.getString("field");
+                    if (type.equals(SubstringMatchPredicate.class.getName())) {
+                        if (field.equals("colletionID")) {
+                            field = "collection_query";
+                        } else if (field.equals("hostID")) {
+                            field = "host_query";
+                        }
+                    }
+                    String value = predicates.getString("value");
+                    if (type.equals(IdEqualsPredicate.class.getName())) {
+                        if (!map.containsKey(field)) {
+                            map.put(field, new Vector());
+                        }
+                        ((List) map.get(field)).add(value);
+                    } else {
+                        map.put(field, value);
+                    }
+                }
+                try {
+                    createSubscription(map);
+                } catch (Exception e1) {
+                    // TODO Auto-generated catch block
+                    e1.printStackTrace();
+                }
+            }
+        } catch (SQLException e) {
+            // TODO Auto-generated catch block
+            e.printStackTrace();
+        } catch (DatabaseException e) {
+            // TODO Auto-generated catch block
+            e.printStackTrace();
+        }
+    }
     
 }