Changeset 8738 for trunk/gsdl3/extensions/gsdl-as/src/org/greenstone/gsdlas/profiles/Predicate.java

Timestamp:

2004-12-06T11:44:03+13:00 (19 years ago)

Author:

schweer

Message:

user authentication works; user information and subscriptions/predicates are stored to thedatabase

File:

• trunk/gsdl3/extensions/gsdl-as/src/org/greenstone/gsdlas/profiles/Predicate.java (modified) (4 diffs)

trunk/gsdl3/extensions/gsdl-as/src/org/greenstone/gsdlas/profiles/Predicate.java

@@ -9 +9 @@
 package org.greenstone.gsdlas.profiles;
 
+import java.sql.*;
+import java.sql.SQLException;
+import java.sql.Statement;
 import java.util.*;
 import java.util.Set;
 import java.util.TreeSet;
+
+import org.greenstone.gsdlas.database.DatabaseException;
+import org.greenstone.gsdlas.database.DatabaseManager;
+import org.greenstone.gsdlas.util.ArrayHelper;
 
 /**
@@ -26 +33 @@
     protected int id;
     
-    public void addSubscription(Subscription sub) {
-        subscriptions.add(sub);
+
+    public static final String[] singleValueFields = new String[] {
+            "document_title",
+            "document_content",
+            "metadata_has_field",
+            "host_query",
+            "collection_query" };
+    public static final String[] multiValueFields = new String[] {
+            "type",
+            "host",
+            "collection" };
+    
+    public void addSubscription(int subscriptionID) {
+        subscriptions.add(new Integer(subscriptionID));
+        // TODO save to db
     }
 
@@ -33 +53 @@
      * @see org.greenstone.gsdlas.profiles.Predicate#getSubscriptions()
      */
-    public Set getSubscriptions() {
+    public Set getSubscriptionIDs() {
         return Collections.unmodifiableSet(subscriptions);
     }
@@ -74 +94 @@
     
     public abstract boolean isSatisfied(Map event);
+
+    /**
+     * @param key
+     * @return
+     */
+    public static boolean isSingleValued(String key) {
+        return ArrayHelper.contains(singleValueFields, key);
+    }
+
+    /**
+     * @param key
+     * @return
+     */
+    public static boolean isMultiValued(String key) {
+        return ArrayHelper.contains(singleValueFields, key);
+    }
+
+    /**
+     * @return
+     * @throws DatabaseException
+     * @throws SQLException
+     */
+    int saveToDatabase() throws SQLException, DatabaseException {
+        // TODO handle multi-valued predicate (in subclass?)
+        String query = "SELECT id FROM predicates " +
+                "WHERE type='single' AND field = '" + field + 
+                "' AND singleValue = '" + value + "';";
+        Statement statement = DatabaseManager.getInstance().getDatabaseConnection().createStatement();
+        System.out.println(query);
+        ResultSet result = statement.executeQuery(query);
+        if (result.next()) { // predicate already exists in database
+            System.out.println("predicate " + id + " already exists");
+            return result.getInt("id");
+        }
+        // predicate has been newly created
+        String insert = "INSERT INTO predicates (type, field, singleValue) " +
+        "VALUES ('single', '" + field + "', '" + value + "');";
+        System.out.println(insert);
+        statement.execute(insert);
+        result = statement.executeQuery(query);
+        if (result.next()) {
+            return result.getInt("id");
+        } else {
+            throw new DatabaseException("could not save predicate");
+        }
+    }
     
 }