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/TransferItem.java | 133 +++++++++++++++++++++
 1 file changed, 133 insertions(+)
 create mode 100644 src/main/java/com/c2kernel/entity/transfer/TransferItem.java

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

diff --git a/src/main/java/com/c2kernel/entity/transfer/TransferItem.java b/src/main/java/com/c2kernel/entity/transfer/TransferItem.java
new file mode 100644
index 0000000..0e3b764
--- /dev/null
+++ b/src/main/java/com/c2kernel/entity/transfer/TransferItem.java
@@ -0,0 +1,133 @@
+package com.c2kernel.entity.transfer;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.Enumeration;
+
+import com.c2kernel.common.ObjectNotFoundException;
+import com.c2kernel.entity.C2KLocalObject;
+import com.c2kernel.entity.TraceableEntity;
+import com.c2kernel.lifecycle.instance.Workflow;
+import com.c2kernel.lookup.DomainPath;
+import com.c2kernel.lookup.EntityPath;
+import com.c2kernel.lookup.Path;
+import com.c2kernel.persistency.ClusterStorage;
+import com.c2kernel.persistency.outcome.Outcome;
+import com.c2kernel.process.Gateway;
+import com.c2kernel.property.Property;
+import com.c2kernel.property.PropertyArrayList;
+import com.c2kernel.utils.CastorXMLUtility;
+import com.c2kernel.utils.FileStringUtility;
+import com.c2kernel.utils.Logger;
+
+public class TransferItem {
+    public ArrayList<String> domainPaths;
+    public int sysKey;
+    static int importAgentId;
+
+    public TransferItem() throws Exception {
+        try {
+            importAgentId = Gateway.getLDAPLookup().getRoleManager().getAgentPath("system").getSysKey();
+        } catch (ObjectNotFoundException e) {
+            Logger.error("TransferItem - System user not found!");
+            throw e;
+        }
+    }
+
+    public TransferItem(int sysKey) throws Exception {
+        this.sysKey = sysKey;
+        domainPaths = new ArrayList<String>();
+        Property name = (Property)Gateway.getStorage().get(sysKey, ClusterStorage.PROPERTY + "/Name", null);
+        Enumeration<Path> paths = Gateway.getLDAPLookup().search(new DomainPath(), name.getValue());
+        while (paths.hasMoreElements()) {
+            DomainPath thisPath = (DomainPath)paths.nextElement();
+            domainPaths.add(thisPath.toString());
+        }
+    }
+
+    public void exportItem(File dir, String path) throws Exception {
+        Logger.msg("Path " + path + " in " + sysKey);
+        String[] contents = Gateway.getStorage().getClusterContents(sysKey, path);
+        if (contents.length > 0) {
+            FileStringUtility.createNewDir(dir.getCanonicalPath());
+            for (String content : contents) {
+                exportItem(new File(dir, content), path + "/" + content);
+            }
+        } else { //no children, try to dump object
+            try {
+                C2KLocalObject obj = Gateway.getStorage().get(sysKey, path, null);
+                Logger.msg("Dumping object " + path + " in " + sysKey);
+                File dumpPath = new File(dir.getCanonicalPath() + ".xml");
+                FileStringUtility.string2File(dumpPath, CastorXMLUtility.marshall(obj));
+                return;
+            } catch (ObjectNotFoundException ex) {
+            } // not an object
+        }
+    }
+
+    public void importItem(File dir) throws Exception {
+        // check if already exists
+        try {
+            Property name = (Property)Gateway.getStorage().get(sysKey, ClusterStorage.PROPERTY + "/Name", null);
+            throw new Exception("Syskey " + sysKey + " already in use as " + name.getValue());
+        } catch (Exception ex) {
+        }
+
+        // retrieve objects
+        ArrayList<String> objectFiles = FileStringUtility.listDir(dir.getCanonicalPath(), false, true);
+        ArrayList<C2KLocalObject> objects = new ArrayList<C2KLocalObject>();
+        for (String element : objectFiles) {
+            String xmlFile = FileStringUtility.file2String(element);
+            C2KLocalObject newObj;
+            String choppedPath = element.substring(dir.getCanonicalPath().length()+1, element.length()-4);
+            Logger.msg(choppedPath);
+            if (choppedPath.startsWith(ClusterStorage.OUTCOME))
+                newObj = new Outcome(choppedPath, xmlFile);
+            else
+                newObj = (C2KLocalObject)CastorXMLUtility.unmarshall(xmlFile);
+
+            objects.add(newObj);
+        }
+
+        // create item
+        EntityPath entityPath = new EntityPath(sysKey);
+        TraceableEntity newItem = (TraceableEntity)Gateway.getCorbaServer().createEntity(entityPath);
+        Gateway.getLDAPLookup().add(entityPath);
+
+        PropertyArrayList props = new PropertyArrayList();
+        Workflow wf = null;
+        // put objects
+        for (C2KLocalObject obj : objects) {
+            if (obj instanceof Property)
+                props.list.add((Property)obj);
+            else if (obj instanceof Workflow)
+                wf = (Workflow)obj;
+        }
+
+        if (wf == null)
+            throw new Exception("No workflow found in import for "+sysKey);
+
+        // init item
+        newItem.initialise(importAgentId, CastorXMLUtility.marshall(props), CastorXMLUtility.marshall(wf.search("workflow/domain")));
+
+        // store objects
+        importByType(ClusterStorage.COLLECTION, objects);
+        importByType(ClusterStorage.HISTORY, objects);
+        importByType(ClusterStorage.OUTCOME, objects);
+        importByType(ClusterStorage.VIEWPOINT, objects);
+        Gateway.getStorage().commit(this);
+        // add domPaths
+        for (String element : domainPaths) {
+            DomainPath newPath = new DomainPath(element, entityPath);
+            Gateway.getLDAPLookup().add(newPath);
+        }
+    }
+
+    private void importByType(String type, ArrayList<C2KLocalObject> objects) throws Exception {
+        for (C2KLocalObject element : objects) {
+            if (element.getClusterType().equals(type))
+                Gateway.getStorage().put(sysKey, element, this);
+        }
+
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3