Attachments management

[gpl/argeo-slc.git] / runtime / org.argeo.slc.support.simple / src / main / java / org / argeo / slc / core / execution / tasks / UploadAttachments.java

diff --git a/runtime/org.argeo.slc.support.simple/src/main/java/org/argeo/slc/core/execution/tasks/UploadAttachments.java b/runtime/org.argeo.slc.support.simple/src/main/java/org/argeo/slc/core/execution/tasks/UploadAttachments.java
index 499e119484a2b77e84c5b6315f3be5a42645e1c6..e4cefa85a65c27340dbcc987d36c8c920f989ca9 100644 (file)
--- a/runtime/org.argeo.slc.support.simple/src/main/java/org/argeo/slc/core/execution/tasks/UploadAttachments.java
+++ b/runtime/org.argeo.slc.support.simple/src/main/java/org/argeo/slc/core/execution/tasks/UploadAttachments.java
@@ -5,6 +5,7 @@ import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+import org.argeo.slc.SlcException;
 import org.argeo.slc.core.attachment.Attachment;
 import org.argeo.slc.core.attachment.AttachmentUploader;
 import org.argeo.slc.core.attachment.AttachmentsEnabled;
@@ -12,18 +13,30 @@ import org.springframework.core.io.Resource;
 
 public class UploadAttachments implements Runnable {
        private AttachmentUploader attachmentUploader;
+       private Attachment attachment = null;
+       private Resource resource = null;
        private Map<Attachment, Resource> attachments = new HashMap<Attachment, Resource>();
        private List<AttachmentsEnabled> attachTo = new ArrayList<AttachmentsEnabled>();
 
        public void run() {
-               for (Attachment attachment : attachments.keySet()) {
-                       Resource resource = attachments.get(attachment);
-                       attachmentUploader.upload(attachment, resource);
-                       for (AttachmentsEnabled attachmentsEnabled : attachTo) {
-                               attachmentsEnabled.addAttachment(attachment);
-                       }
+               if (attachment != null) {
+                       if (resource == null)
+                               throw new SlcException("A resource must be specified.");
+                       uploadAndAdd(attachment, resource);
                }
 
+               for (Attachment attachmentT : attachments.keySet()) {
+                       Resource resourceT = attachments.get(attachmentT);
+                       uploadAndAdd(attachmentT, resourceT);
+               }
+
+       }
+
+       protected void uploadAndAdd(Attachment attachment, Resource resource) {
+               attachmentUploader.upload(attachment, resource);
+               for (AttachmentsEnabled attachmentsEnabled : attachTo) {
+                       attachmentsEnabled.addAttachment(attachment);
+               }
        }
 
        public void setAttachmentUploader(AttachmentUploader attachmentUploader) {
@@ -38,5 +51,12 @@ public class UploadAttachments implements Runnable {
                this.attachTo = attachTo;
        }
 
-       
+       public void setAttachment(Attachment attachment) {
+               this.attachment = attachment;
+       }
+
+       public void setResource(Resource resource) {
+               this.resource = resource;
+       }
+
 }