org.linagora.linshare.core.dao.impl.JackRabbitFileDataStoreImpl.java Source code

Introduction

Here is the source code for org.linagora.linshare.core.dao.impl.JackRabbitFileDataStoreImpl.java

Source

/*
 * LinShare is an open source filesharing software, part of the LinPKI software
 * suite, developed by Linagora.
 * 
 * Copyright (C) 2016 LINAGORA
 * 
 * This program is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Affero General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option) any
 * later version, provided you comply with the Additional Terms applicable for
 * LinShare software by Linagora pursuant to Section 7 of the GNU Affero General
 * Public License, subsections (b), (c), and (e), pursuant to which you must
 * notably (i) retain the display of the LinShare? trademark/logo at the top
 * of the interface window, the display of the You are using the Open Source
 * and free version of LinShare, powered by Linagora  20092015. Contribute to
 * Linshare R&D by subscribing to an Enterprise offer!? infobox and in the
 * e-mails sent with the Program, (ii) retain all hypertext links between
 * LinShare and linshare.org, between linagora.com and Linagora, and (iii)
 * refrain from infringing Linagora intellectual property rights over its
 * trademarks and commercial brands. Other Additional Terms apply, see
 * <http://www.linagora.com/licenses/> for more details.
 * 
 * This program 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 Affero General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Affero General Public License and
 * its applicable Additional Terms for LinShare along with this program. If not,
 * see <http://www.gnu.org/licenses/> for the GNU Affero General Public License
 * version 3 and <http://www.linagora.com/licenses/> for the Additional Terms
 * applicable to LinShare software.
 */

package org.linagora.linshare.core.dao.impl;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.UUID;

import org.apache.commons.lang.Validate;
import org.linagora.linshare.core.dao.FileDataStore;
import org.linagora.linshare.core.dao.FileSystemDao;
import org.linagora.linshare.core.domain.objects.FileMetaData;
import org.linagora.linshare.core.exception.TechnicalErrorCode;
import org.linagora.linshare.core.exception.TechnicalException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class JackRabbitFileDataStoreImpl implements FileDataStore {

    protected Logger logger = LoggerFactory.getLogger(this.getClass());

    private static final String jackRabbitDefaultPath = "ee0e115f-240e-428b-8f5c-427184cf67b1";

    private FileSystemDao dao;

    public JackRabbitFileDataStoreImpl(FileSystemDao dao) {
        super();
        this.dao = dao;
    }

    @Override
    public void remove(FileMetaData metadata) {
        Validate.notNull(metadata);
        String uuid = metadata.getUuid();
        try {
            dao.removeFileByUUID(uuid);
        } catch (org.springmodules.jcr.JcrSystemException e1) {
            logger.warn(e1.getMessage(), e1);
        } catch (org.springframework.dao.DataRetrievalFailureException e) {
            logger.warn(e.getMessage(), e);
        }
    }

    @Override
    public FileMetaData add(File file, FileMetaData metadata) {
        Validate.notNull(metadata);
        String mimeType = metadata.getMimeType();
        Long size = metadata.getSize();
        Validate.notEmpty(mimeType);
        Validate.notNull(size);
        try (FileInputStream fis = new FileInputStream(file)) {
            String uuid = dao.insertFile(jackRabbitDefaultPath, fis, size, UUID.randomUUID().toString(), mimeType);
            metadata.setUuid(uuid);
            return metadata;
        } catch (IOException e) {
            logger.error(e.getMessage(), e);
            throw new TechnicalException(TechnicalErrorCode.GENERIC, "Can not add a new file : " + e.getMessage());
        }
    }

    @Override
    public FileMetaData add(InputStream file, FileMetaData metadata) {
        String mimeType = metadata.getMimeType();
        Long size = metadata.getSize();
        Validate.notEmpty(mimeType);
        Validate.notNull(size);
        String uuid = dao.insertFile(jackRabbitDefaultPath, file, size, UUID.randomUUID().toString(), mimeType);
        metadata.setUuid(uuid);
        return metadata;
    }

    @Override
    public InputStream get(FileMetaData metadata) {
        Validate.notEmpty(metadata.getUuid());
        InputStream inputStream = dao.getFileContentByUUID(metadata.getUuid());
        if (inputStream == null) {
            throw new TechnicalException(TechnicalErrorCode.GENERIC,
                    "Can not find document in jackrabbit : " + metadata.getUuid());
        }
        return inputStream;
    }

    @Override
    public boolean exists(FileMetaData metadata) {
        try (InputStream is = dao.getFileContentByUUID(metadata.getUuid())) {
            if (is != null) {
                return true;
            }
        } catch (IOException e) {
            logger.error(e.getMessage(), e);
        }
        return false;
    }

}