org.alfresco.bm.devicesync.eventprocessor.UploadFileForSubscription.java Source code

Introduction

Here is the source code for org.alfresco.bm.devicesync.eventprocessor.UploadFileForSubscription.java

Source

/*
 * Copyright (C) 2005-2014 Alfresco Software Limited.
 *
 * This file is part of Alfresco
 *
 * Alfresco is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Alfresco is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
 */
package org.alfresco.bm.devicesync.eventprocessor;

import org.alfresco.bm.devicesync.dao.SubscriptionsService;
import org.alfresco.bm.devicesync.data.CMISEventData;
import org.alfresco.bm.devicesync.data.SubscriptionData;
import org.alfresco.bm.devicesync.data.UploadFileData;
import org.alfresco.bm.devicesync.util.UploadFileException;
import org.alfresco.bm.devicesync.util.UploadFileHelper;
import org.alfresco.bm.devicesync.util.UploadFileHelper.UploadListener;
import org.alfresco.bm.event.Event;
import org.alfresco.bm.event.EventResult;

import com.mongodb.BasicDBObjectBuilder;
import com.mongodb.DBObject;

/**
 * Upload a new file
 * 
 * <h1>Input</h1>
 * 
 * A {@link CMISEventData data object } containing an existing folder.
 * 
 * <h1>Actions</h1>
 * 
 * Upload a random file to the current folder
 * 
 * <h1>Output</h1>
 * 
 * {@link #EVENT_NAME_FILE_UPLOADED}: The {@link CMISEventData data object} with
 * the new file<br/>
 * 
 * @author sglover
 * @since 1.0
 */
public class UploadFileForSubscription extends AbstractCMISEventProcessor {
    public static final String EVENT_NAME_FILE_UPLOADED = "fileUploaded";

    private final SubscriptionsService subscriptionsService;
    private String eventNameFileUploaded;
    private final UploadFileHelper uploadFileHelper;

    /**
     * @param testFileService
     *            service to provide sample files for upload
     */
    public UploadFileForSubscription(SubscriptionsService subscriptionsService, UploadFileHelper uploadFileHelper) {
        super();
        this.subscriptionsService = subscriptionsService;
        this.uploadFileHelper = uploadFileHelper;
        this.eventNameFileUploaded = EVENT_NAME_FILE_UPLOADED;
    }

    /**
     * Override the {@link #EVENT_NAME_FILE_UPLOADED default} event name for
     * 'file uploaded'.
     */
    public void setEventNameFileUploaded(String eventNameFileUploaded) {
        this.eventNameFileUploaded = eventNameFileUploaded;
    }

    @Override
    protected EventResult processCMISEvent(Event event) throws Exception {
        super.suspendTimer(); // Timer control

        DBObject dbObject = (DBObject) event.getData();
        UploadFileData uploadFileData = UploadFileData.fromDBObject(dbObject);
        String subscriptionId = uploadFileData.getSubscriptionId();
        SubscriptionData subscriptionData = subscriptionsService.getSubscription(subscriptionId);

        try {
            DBObject data = uploadFileHelper.doUpload(uploadFileData, subscriptionData, new UploadListener() {
                @Override
                public void beforeUpload() {
                    UploadFileForSubscription.super.resumeTimer(); // Timer
                                                                   // control
                }

                @Override
                public void afterUpload() {
                    UploadFileForSubscription.super.stopTimer(); // Timer
                                                                 // control
                }

                @Override
                public void onException(Exception e) {
                    UploadFileForSubscription.super.stopTimer(); // Timer
                                                                 // control
                }
            }, super.getName());

            // Done
            Event doneEvent = new Event(eventNameFileUploaded, data);
            EventResult result = new EventResult(BasicDBObjectBuilder.start()
                    .append("msg", "Successfully uploaded document.").append("document", data).get(), doneEvent);

            // Done
            return result;
        } catch (UploadFileException e) {
            return new EventResult(
                    BasicDBObjectBuilder.start().append("msg", "Exception uploading document.")
                            .append("exception", e.getE().getMessage()).append("document", e.getData()).get(),
                    false);
        }
    }
}