From b086f57f56bf0eb9dab9cf321a0f69aaaae84347 Mon Sep 17 00:00:00 2001
From: Andrew Branson <andrew.branson@cern.ch>
Date: Wed, 30 May 2012 08:37:45 +0200
Subject: Initial Maven Conversion

---
 .../com/c2kernel/entity/transfer/TransferSet.java  | 103 +++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 src/main/java/com/c2kernel/entity/transfer/TransferSet.java

(limited to 'src/main/java/com/c2kernel/entity/transfer/TransferSet.java')

diff --git a/src/main/java/com/c2kernel/entity/transfer/TransferSet.java b/src/main/java/com/c2kernel/entity/transfer/TransferSet.java
new file mode 100644
index 0000000..71a593a
--- /dev/null
+++ b/src/main/java/com/c2kernel/entity/transfer/TransferSet.java
@@ -0,0 +1,103 @@
+package com.c2kernel.entity.transfer;
+
+import java.io.File;
+import java.util.ArrayList;
+
+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<TransferItem> items;
+
+    public TransferSet() {
+    }
+
+    public TransferSet(int[] sysKeys) {
+        items = new ArrayList<TransferItem>();
+        for (int sysKey : sysKeys) {
+            try {
+                items.add(new TransferItem(sysKey));
+            } catch (Exception ex) {
+                Logger.error("Could not add item "+sysKey);
+                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 (TransferItem element : items) {
+            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 (TransferItem element : items) {
+            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 (TransferItem element : items) {
+            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);
+        }
+    }
+}
-- 
cgit v1.2.3