From b086f57f56bf0eb9dab9cf321a0f69aaaae84347 Mon Sep 17 00:00:00 2001 From: Andrew Branson Date: Wed, 30 May 2012 08:37:45 +0200 Subject: Initial Maven Conversion --- .../com/c2kernel/entity/transfer/TransferItem.java | 133 +++++++++++++++++++++ .../com/c2kernel/entity/transfer/TransferSet.java | 103 ++++++++++++++++ 2 files changed, 236 insertions(+) create mode 100644 src/main/java/com/c2kernel/entity/transfer/TransferItem.java create mode 100644 src/main/java/com/c2kernel/entity/transfer/TransferSet.java (limited to 'src/main/java/com/c2kernel/entity/transfer') 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 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(); + Property name = (Property)Gateway.getStorage().get(sysKey, ClusterStorage.PROPERTY + "/Name", null); + Enumeration 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 objectFiles = FileStringUtility.listDir(dir.getCanonicalPath(), false, true); + ArrayList objects = new ArrayList(); + 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 objects) throws Exception { + for (C2KLocalObject element : objects) { + if (element.getClusterType().equals(type)) + Gateway.getStorage().put(sysKey, element, this); + } + + } +} \ No newline at end of file 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 items; + + public TransferSet() { + } + + public TransferSet(int[] sysKeys) { + items = new ArrayList(); + 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