1 files changed, 107 insertions, 0 deletions
diff --git a/source/com/c2kernel/entity/transfer/TransferSet.java b/source/com/c2kernel/entity/transfer/TransferSet.java
new file mode 100755
index 0000000..ee98f8d
--- /dev/null
+++ b/source/com/c2kernel/entity/transfer/TransferSet.java
@@ -0,0 +1,107 @@
+package com.c2kernel.entity.transfer;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.Iterator;
+
+import com.c2kernel.lookup.EntityPath;
+import com.c2kernel.lookup.NextKeyManager;
+import com.c2kernel.process.Gateway;
+import com.c2kernel.utils.CastorXMLUtility;
+import com.c2kernel.utils.FileStringUtility;
+import com.c2kernel.utils.Logger;
+
+/**************************************************************************
+ *
+ * $Revision: 1.5 $
+ * $Date: 2005/04/26 06:48:13 $
+ *
+ * Copyright (C) 2003 CERN - European Organization for Nuclear Research
+ * All rights reserved.
+ **************************************************************************/
+
+public class TransferSet {
+
+    public ArrayList items;
+    
+    public TransferSet() {
+    }
+    
+    public TransferSet(int[] sysKeys) {
+        items = new ArrayList();
+        for (int i = 0; i < sysKeys.length; i++) {
+            try {
+                items.add(new TransferItem(sysKeys[i]));
+            } catch (Exception ex) {
+                Logger.error("Could not add item "+sysKeys[i]);
+                Logger.error(ex);
+            }
+        }
+    }
+    
+    public void exportPackage(File dir) throws Exception {
+        if (items==null || items.size() == 0)
+            throw new Exception("Nothing to dump");
+        FileStringUtility.createNewDir(dir.getAbsolutePath());
+        for (Iterator iter = items.iterator(); iter.hasNext();) {
+            TransferItem element = (TransferItem)iter.next();
+            try {
+                element.exportItem(new File(dir, String.valueOf(element.sysKey)), "/");
+            } catch (Exception ex) {
+                Logger.error("Error dumping item "+element.sysKey);     
+                Logger.error(ex);  
+            }
+        }
+        
+        try {
+            String self = CastorXMLUtility.marshall(this);
+            FileStringUtility.string2File(new File(dir, "transferSet.xml"), self);
+        } catch (Exception ex) {
+            Logger.error("Error writing header file");
+            Logger.error(ex);
+        }
+    }
+    
+    public void importPackage(File rootDir) {
+        for (Iterator iter = items.iterator(); iter.hasNext();) {
+            TransferItem element = (TransferItem)iter.next();
+            Logger.msg(5, "Importing "+element.sysKey);
+            try {
+                element.importItem(new File(rootDir, String.valueOf(element.sysKey)));
+            } catch (Exception ex) {
+                Logger.error("Import of item "+element.sysKey+" failed. Rolling back");
+                Logger.error(ex);
+                Gateway.getStorage().abort(element);
+            }
+        }
+        checkLastKey();
+    }
+    
+    private void checkLastKey()
+    {   
+        // find highest key in out import set
+        int packageLastKey = 0;
+        for (Iterator iter = items.iterator(); iter.hasNext();) {
+            TransferItem element = (TransferItem)iter.next();
+            if (element.sysKey > packageLastKey)
+                packageLastKey = element.sysKey;
+        }
+        
+        try
+        {   // find the current last key
+            NextKeyManager nextKeyMan = Gateway.getLDAPLookup().getNextKeyManager();
+            EntityPath lastKey = nextKeyMan.getLastEntityPath();
+            Logger.msg(1, "Last key imported was "+packageLastKey+". LDAP lastkey was "+lastKey.getSysKey());
+
+                
+            if (packageLastKey > lastKey.getSysKey()) { // set new last
+                Logger.msg(1, "Updating lastKey to "+packageLastKey);
+                nextKeyMan.writeLastEntityKey(packageLastKey);
+            }
+        }
+        catch (Exception ex)
+        {
+            Logger.error("Exception::LoadKeys::processFile() " + ex);
+        }           
+    }
+}