From fea9f5836c4a13e2024c2f678061268ab7421b8d Mon Sep 17 00:00:00 2001
From: Andrew Branson <andrew.branson@cern.ch>
Date: Mon, 30 Jul 2012 22:09:32 +0200
Subject: Predefined steps audit, including new steps for collection management
 Fixes #19

---
 .../predefined/RemoveSlotFromCollection.java       | 121 +++++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100644 src/main/java/com/c2kernel/lifecycle/instance/predefined/RemoveSlotFromCollection.java

(limited to 'src/main/java/com/c2kernel/lifecycle/instance/predefined/RemoveSlotFromCollection.java')

diff --git a/src/main/java/com/c2kernel/lifecycle/instance/predefined/RemoveSlotFromCollection.java b/src/main/java/com/c2kernel/lifecycle/instance/predefined/RemoveSlotFromCollection.java
new file mode 100644
index 0000000..897183d
--- /dev/null
+++ b/src/main/java/com/c2kernel/lifecycle/instance/predefined/RemoveSlotFromCollection.java
@@ -0,0 +1,121 @@
+/**************************************************************************
+ * AssignItemToSlot
+ *
+ * $Workfile$
+ * $Revision: 1.8 $
+ * $Date: 2004/10/21 08:02:19 $
+ *
+ * Copyright (C) 2001 CERN - European Organization for Nuclear Research
+ * All rights reserved.
+ **************************************************************************/
+
+package com.c2kernel.lifecycle.instance.predefined;
+
+
+import com.c2kernel.collection.Collection;
+import com.c2kernel.collection.CollectionMember;
+import com.c2kernel.collection.MembershipException;
+import com.c2kernel.common.AccessRightsException;
+import com.c2kernel.common.InvalidDataException;
+import com.c2kernel.common.InvalidTransitionException;
+import com.c2kernel.common.ObjectNotFoundException;
+import com.c2kernel.lookup.AgentPath;
+import com.c2kernel.lookup.EntityPath;
+import com.c2kernel.persistency.ClusterStorage;
+import com.c2kernel.persistency.ClusterStorageException;
+import com.c2kernel.process.Gateway;
+import com.c2kernel.utils.Logger;
+
+/**************************************************************************
+ *
+ * @author $Author: abranson $ $Date: 2004/10/21 08:02:19 $
+ * @version $Revision: 1.8 $
+ **************************************************************************/
+public class RemoveSlotFromCollection extends PredefinedStep
+{
+    /**************************************************************************
+    * Constructor for Castor
+    **************************************************************************/
+    public RemoveSlotFromCollection()
+    {
+        super();
+    }
+
+
+    /**
+     * Params:
+     * 0 - collection name
+     * 1 - slot number OR if null:
+     * 2 - target entity key
+     */
+    @Override
+	public void request(AgentPath agent, int transitionID, String requestData)
+        throws AccessRightsException,
+               InvalidTransitionException,
+               InvalidDataException
+    {
+        EntityPath entityPath = getItemEntityPath();
+        String collName;
+        int slotNo;
+        int entityKey = -1;
+        Collection<? extends CollectionMember> coll;
+
+        Logger.msg(1, "RemoveSlotFromCollection::request() - Starting.");
+
+        checkAccessRights(agent);
+
+        // extract parameters
+        try {
+            entityPath = getItemEntityPath();
+            String[] params = getDataList(requestData);
+            collName = params[0];
+            slotNo = params[1].length()>0?Integer.parseInt(params[1]):-1;
+            if (params.length>2) entityKey = params[2].length()>0?Integer.parseInt(params[2]):-1;
+        } catch (Exception e) {
+            throw new InvalidDataException("Invalid parameters", "");
+        }
+        
+        if (slotNo == -1 && entityKey == -1)
+        	throw new InvalidDataException("Must give either slot number or entity key", "");
+
+        // load collection
+        try {
+			coll = (Collection<? extends CollectionMember>)Gateway.getStorage().get(entityPath.getSysKey(), ClusterStorage.COLLECTION+"/"+collName, null);
+		} catch (ObjectNotFoundException ex) {
+			throw new InvalidDataException("Collection '"+collName+"' not found in this Item", "");
+		} catch (ClusterStorageException ex) {
+			Logger.error(ex);
+			throw new InvalidDataException("Error loading collection '\"+collName+\"': "+ex.getMessage(), "");
+		}
+
+        if (slotNo == -1) { // find slot from entity key
+        	for (CollectionMember member : coll.getMembers().list) {
+        		if (member.getEntityKey() == entityKey) {
+        			slotNo = member.getID();
+        			break;
+        		}
+            }
+        }
+        if (slotNo == -1) {
+            throw new InvalidDataException("No match", "");
+        }
+        
+        // Remove the slot
+        try {
+			coll.removeMember(slotNo);
+		} catch (MembershipException e) {
+			throw new InvalidDataException(e.getMessage(), "");
+		}
+
+        // Store the collection
+		try {
+            Gateway.getStorage().put(entityPath.getSysKey(), coll, null);
+        } catch (ClusterStorageException e) {
+            Logger.error(e);
+            throw new InvalidDataException("Error storing collection", "");
+        }
+		sendEventStoreOutcome(transitionID, requestData, agent);
+
+        Logger.msg(1, "RemoveSlotFromCollection::request() - DONE.");
+    }
+}
-- 
cgit v1.2.3