Uploader.java

/*
 * Copyright (c) www.bugull.com
 * 
 * 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 net.tooan.ynpay.third.mongodb.fs;

import net.tooan.ynpay.third.jfinal.log.Logger;
import net.tooan.ynpay.third.mongodb.mapper.StreamUtil;
import net.tooan.ynpay.third.mongodb.mapper.StringUtil;

import java.io.*;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;

/**
 * Convenient class for uploading a file to GridFS. It uses the BuguFS class internally.
 *
 * @author Frank Wen(xbwen@hotmail.com)
 */
public class Uploader {

    private final static Logger logger = Logger.getLogger(Uploader.class);

    protected InputStream input;
    protected String originalName;
    protected boolean rename;

    protected String bucketName = BuguFS.DEFAULT_BUCKET;
    protected long chunkSize = BuguFS.DEFAULT_CHUNKSIZE;
    protected String filename;
    protected Map<String, Object> params;

    public Uploader(File file, String originalName) {
        try {
            this.input = new FileInputStream(file);
        } catch (FileNotFoundException ex) {
            logger.error("Can not create the FileInputStream", ex);
        }
        this.originalName = originalName;
    }

    public Uploader(File file, String originalName, boolean rename) {
        this(file, originalName);
        this.rename = rename;
    }

    public Uploader(InputStream input, String originalName) {
        this.input = input;
        this.originalName = originalName;
    }

    public Uploader(InputStream input, String originalName, boolean rename) {
        this(input, originalName);
        this.rename = rename;
    }

    public Uploader(byte[] data, String originalName) {
        this.input = new ByteArrayInputStream(data);
        this.originalName = originalName;
    }

    public Uploader(byte[] data, String originalName, boolean rename) {
        this(data, originalName);
        this.rename = rename;
    }

    public void setBucketName(String bucketName) {
        this.bucketName = bucketName;
    }

    public void setChunkSize(long chunkSize) {
        this.chunkSize = chunkSize;
    }

    public void setAttribute(String key, Object value) {
        if (params == null) {
            params = new HashMap<String, Object>();
        }
        params.put(key, value);
    }

    public void save() {
        processFilename();
        saveInputStream();
    }

    public String getFilename() {
        return filename;
    }

    protected void processFilename() {
        if (rename) {
            long time = System.nanoTime();
            String timeStr = String.valueOf(time);
            //the nano time's last 2 bits is 00, replace it with random number
            String subStr = timeStr.substring(0, timeStr.length() - 2);
            StringBuilder sb = new StringBuilder();
            sb.append(subStr);
            int r = new Random().nextInt(100);
            if (r < 10) {
                sb.append("0").append(r);
            } else {
                sb.append(r);
            }
            String ext = StringUtil.getExtention(originalName);
            if (!StringUtil.isEmpty(ext)) {
                sb.append(".").append(ext);
            }
            filename = sb.toString();
        } else {
            filename = originalName;
        }
    }

    protected void saveInputStream() {
        BuguFS fs = new BuguFS(bucketName, chunkSize);
        fs.save(input, filename, params);
        StreamUtil.safeClose(input);
    }

}