Changeset 5945 for trunk/gsdl3/src/java/org/greenstone/gsdl3/gs3build/metadata/METSFileSet.java

Timestamp:

2003-11-24T14:26:53+13:00 (20 years ago)

Author:

cs025

Message:

Extensive additions to metadata

File:

• trunk/gsdl3/src/java/org/greenstone/gsdl3/gs3build/metadata/METSFileSet.java (modified) (5 diffs)

trunk/gsdl3/src/java/org/greenstone/gsdl3/gs3build/metadata/METSFileSet.java

@@ -8 +8 @@
 import java.util.Map;
 import java.util.HashMap;
+import java.util.List;
+import java.util.ArrayList;
 import java.util.Iterator;
 
 import java.io.File;
 import java.io.PrintWriter;
+
+import java.sql.SQLException;
+import java.sql.ResultSet;
+
 import java.net.URL;
 
@@ -20 +26 @@
 
 public class METSFileSet
-{ Map fileGroups;
+{ Map    fileGroups;
+  String reference;
 
   public METSFileSet()
   { this.fileGroups = new HashMap();
     this.fileGroups.put("default", new METSFileGroup("default"));
-  }
-
+    this.reference = null;
+  }
+  
+  /**
+   *  Get the group that corresponds to the given name...
+   *
+   *  @param <code>String</code> the name of the group.
+   *  @return <code>METSFileGroup</code> the group object - this will be
+   *          <code>null</code> if the group is not found
+   */
   public METSFileGroup getGroup(String name)
   { if (!this.fileGroups.containsKey(name))
@@ -34 +49 @@
   }
 
+  /**
+   *  Get the Nth file from the default group...
+   *
+   *  @param <code>int</code> the index into the default group to use...
+   *  @return <code>METSFile</code> the file.
+   */
   public METSFile getFile(int index)
   { METSFileGroup group = this.getGroup("default");
@@ -70 +91 @@
   }
 
+  /**
+   *  Add a group at the top level of the METS File set.
+   *
+   *  @param <code>METSFileGroup</code> the file group to add.
+   */
   public void addGroup(METSFileGroup group)
   { this.fileGroups.put(group.getName(), group);
+  }
+
+  /**
+   *  Get all the groups that contain a given URL/file.
+   *
+   *  @param <code>URL</code> the location of the file.
+   *  @return <code>List</code> the list of group references that contain the file.
+   */
+  public List findGroups(URL file)
+  { List resultList = new ArrayList();
+
+    Iterator groups = this.fileGroups.values().iterator();
+
+    while (groups.hasNext())
+    { METSFileGroup group = (METSFileGroup) groups.next();
+      
+      group.findGroups(file, resultList);
+    }
+    return resultList;
   }
 
@@ -86 +131 @@
   }
 
-  public void writeSQL(DocumentInterface document, GS3SQLConnection connection)
+  public boolean writeSQL(DocumentInterface document, GS3SQLConnection connection)
   { Iterator groups = this.fileGroups.values().iterator();
 
-    GS3SQLInsert insert = new GS3SQLInsert("filesection");
-    insert.addValue("DocID", document.getID().toString());
-    insert.addValue("FileSecID", "test"); // TODO: remove magic string
-
-    connection.execute(insert.toString());
+    // write the set if the reference does not already exist...
+    if (this.reference == null)
+    { // Insert the file section into the database
+      GS3SQLInsert insert = new GS3SQLInsert("filesection");
+      insert.addValue("DocID", document.getID().toString());
+      insert.addValue("FileSecID", "test"); // TODO: remove magic string
+
+      if (!connection.execute(insert.toString()))
+      { return false;
+      }
+
+      // find the file section number
+      GS3SQLSelect select = new GS3SQLSelect("filesection");
+      select.addField("*");
+      GS3SQLWhereItem whereItem = new GS3SQLWhereItem("FileSecID", "=", "test");
+      GS3SQLWhereItem whereDoc  = new GS3SQLWhereItem("DocID", "=", document.getID().toString());
+      GS3SQLWhere where = new GS3SQLWhere(whereItem);
+      where.add(whereDoc);
+      select.setWhere(where);
+
+      try {
+    connection.execute(select.toString());
+    
+    ResultSet set = connection.getResultSet();
+    set.first();
+    int sectionRef = set.getInt("FileSectionRef");
+
+    this.reference = Integer.toString(sectionRef);
+      }
+      catch (SQLException ex)
+      { System.out.println(ex);
+        return false;
+      }
+    }
     
+    // write out the children
     while (groups.hasNext())
     { METSFileGroup group = (METSFileGroup) groups.next();
-
-      group.writeSQL(document, connection);
-    }
-  }
+      
+      if (!group.writeSQL(document, this.reference, true, connection))
+      { return false;
+      }
+    }
+
+    return true;
+  }
+
+  public static METSFileSet readSQL(DocumentInterface document, GS3SQLConnection connection)
+  { 
+    METSFileSet set = new METSFileSet();    
+
+    // Get file sections from the filesection table (currently redundant)
+    GS3SQLSelect select = new GS3SQLSelect("filesection");
+    select.addField("*");
+    GS3SQLWhereItem whereItem = new GS3SQLWhereItem("DocID", "=", document.getID().toString());
+    select.setWhere(new GS3SQLWhere(whereItem));
+    connection.execute(select.toString());
+    
+    // Get the identifier for this file set, etc.
+    ResultSet sections = connection.getResultSet();
+    int fileSetRef;
+    try {
+      sections.first();
+      fileSetRef     = sections.getInt("FileSectionRef");
+      set.reference = Integer.toString(fileSetRef);
+    }
+    catch (SQLException ex)
+    { System.out.println(ex);
+      return null;
+    }
+    
+    // Get child file groups
+    select = new GS3SQLSelect("filegroups");
+    select.addField("*");
+    whereItem = new GS3SQLWhereItem("DocID", "=", document.getID().toString());    
+    GS3SQLWhere where = new GS3SQLWhere(whereItem);
+    whereItem = new GS3SQLWhereItem("ParentRef", "=", Integer.toString(fileSetRef), 
+                    GS3SQLField.INTEGER_TYPE);
+    where.add(whereItem);
+    whereItem = new GS3SQLWhereItem("ParentType", "=", METSFileGroup.SECTION_PARENT);
+    where.add(whereItem);
+    select.setWhere(where);
+    connection.execute(select.toString());
+
+    // start going through the matching file groups
+    try {
+      ResultSet resultSet = connection.getResultSet();
+      resultSet.first();
+      do {
+    METSFileGroup filegroup = METSFileGroup.readSQL(document, connection, resultSet);
+    if (filegroup != null) {
+      set.addGroup(filegroup);
+    }
+      } while (resultSet.next());
+    }
+    catch (SQLException sqlEx) {
+      System.out.println(sqlEx);
+      System.exit(1);
+    }
+
+    return set;
+  }
+
 }