jp.co.nemuzuka.service.impl.UploadFileServiceImpl.java Source code

Introduction

Here is the source code for jp.co.nemuzuka.service.impl.UploadFileServiceImpl.java

Source

/*
 * Copyright 2012 Kazumune Katagiri. (http://d.hatena.ne.jp/nemuzuka)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */
package jp.co.nemuzuka.service.impl;

import java.nio.ByteBuffer;
import java.util.ConcurrentModificationException;
import java.util.List;

import jp.co.nemuzuka.dao.UploadFileDao;
import jp.co.nemuzuka.model.UploadFileModel;
import jp.co.nemuzuka.service.UploadFileService;

import org.apache.commons.lang.StringUtils;
import org.slim3.controller.upload.FileItem;
import org.slim3.datastore.Datastore;
import org.slim3.util.BeanUtil;
import org.slim3.util.CopyOptions;

import com.google.appengine.api.blobstore.BlobInfo;
import com.google.appengine.api.blobstore.BlobInfoFactory;
import com.google.appengine.api.blobstore.BlobKey;
import com.google.appengine.api.blobstore.BlobstoreService;
import com.google.appengine.api.blobstore.BlobstoreServiceFactory;
import com.google.appengine.api.datastore.Key;
import com.google.appengine.api.datastore.Text;
import com.google.appengine.api.files.AppEngineFile;
import com.google.appengine.api.files.FileService;
import com.google.appengine.api.files.FileServiceFactory;
import com.google.appengine.api.files.FileWriteChannel;

/**
 * UploadFileService?.
 * @author k-katagiri
 */
public class UploadFileServiceImpl implements UploadFileService {

    UploadFileDao uploadFileDao = UploadFileDao.getInstance();

    private static UploadFileServiceImpl impl = new UploadFileServiceImpl();

    /**
     * ?.
     * @return 
     */
    public static UploadFileServiceImpl getInstance() {
        return impl;
    }

    /**
     * .
     */
    private UploadFileServiceImpl() {
    }

    /* (non-Javadoc)
     * @see jp.co.nemuzuka.service.UploadFileService#put(org.slim3.controller.upload.FileItem, java.lang.String, java.lang.String, java.lang.String)
     */
    @Override
    public String put(FileItem fileItem, String comment, String ticketKeyToString, String projectKeyString) {

        BlobKey blobKey = null;
        try {
            FileService fileService = FileServiceFactory.getFileService();
            AppEngineFile blobFile = fileService.createNewBlobFile(fileItem.getContentType());
            FileWriteChannel writeChannel = fileService.openWriteChannel(blobFile, true);
            writeChannel.write(ByteBuffer.wrap(fileItem.getData()));
            writeChannel.closeFinally();
            blobKey = fileService.getBlobKey(blobFile);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

        //?
        UploadFileModel model = new UploadFileModel();
        BlobInfoFactory factory = new BlobInfoFactory();
        model.setBlobKey(blobKey.getKeyString());
        BlobInfo blobInfo = factory.loadBlobInfo(blobKey);
        BeanUtil.copy(blobInfo, model, new CopyOptions().exclude("blobKey"));
        model.setParentKey(Datastore.stringToKey(ticketKeyToString));
        model.setProjectKey(Datastore.stringToKey(projectKeyString));
        model.setFilename(fileItem.getShortFileName());
        model.setComment(new Text(StringUtils.defaultString(comment)));

        //
        Datastore.put(model);
        return Datastore.keyToString(model.getKey());
    }

    /* (non-Javadoc)
     * @see jp.co.nemuzuka.service.UploadFileService#delete(java.lang.String, java.lang.String, java.lang.String, java.lang.Long)
     */
    @Override
    public void delete(String uploadFileKeyString, String ticketKeyToString, String projectKeyString,
            Long version) {

        //?
        UploadFileModel model = uploadFileDao.get(uploadFileKeyString, ticketKeyToString, projectKeyString,
                version);
        if (model == null) {
            //??????Exception
            throw new ConcurrentModificationException();
        }
        delete(model);
    }

    /* (non-Javadoc)
     * @see jp.co.nemuzuka.service.UploadFileService#getList(java.lang.String, java.lang.String)
     */
    @Override
    public List<UploadFileModel> getList(String ticketKeyToString, String projectKeyString) {
        return uploadFileDao.getList(ticketKeyToString, projectKeyString);
    }

    /* (? Javadoc)
     * @see jp.co.nemuzuka.service.UploadFileService#get(java.lang.String, java.lang.String, java.lang.String)
     */
    @Override
    public UploadFileModel get(String keyString, String ticketKeyToString, String projectKeyString) {
        Key key = Datastore.stringToKey(keyString);
        Key parentKey = Datastore.stringToKey(ticketKeyToString);
        Key projectKey = Datastore.stringToKey(projectKeyString);
        UploadFileModel model = uploadFileDao.get(key);
        if (model == null || model.getParentKey().equals(parentKey) == false
                || model.getProjectKey().equals(projectKey) == false) {
            return null;
        }
        return model;
    }

    /* (non-Javadoc)
     * @see jp.co.nemuzuka.service.UploadFileService#delete4ticketKeyString(java.lang.String, java.lang.String)
     */
    @Override
    public void delete4ticketKeyString(String ticketKeyToString, String projectKeyString) {
        //??
        List<UploadFileModel> list = getList(ticketKeyToString, projectKeyString);
        for (UploadFileModel target : list) {
            //????
            delete(target);
        }
    }

    /**
     * .
     * Model?blob???
     * @param model Model
     */
    void delete(UploadFileModel model) {
        //blobKey???Entity?
        BlobKey blobKey = new BlobKey(model.getBlobKey());
        BlobstoreService bs = BlobstoreServiceFactory.getBlobstoreService();
        bs.delete(blobKey);

        uploadFileDao.delete(model.getKey());
    }
}