com.ewcms.content.resource.service.operator.FileOperator.java Source code

Introduction

Here is the source code for com.ewcms.content.resource.service.operator.FileOperator.java

Source

/**
 * Copyright (c)2010-2011 Enterprise Website Content Management System(EWCMS), All rights reserved.
 * EWCMS PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
 * http://www.ewcms.com
 */

package com.ewcms.content.resource.service.operator;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.ewcms.publication.PublishException;
import com.ewcms.publication.uri.UriRuleable;

/**
 * ???
 * <br>
 * ????
 *
 * @author wangwei
 */
public class FileOperator implements ResourceOperatorable {

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

    private String rootDirPath;

    public FileOperator(String rootDirPath) {
        this.rootDirPath = rootDirPath;
    }

    @Override
    public String write(InputStream source, UriRuleable uriRule) throws IOException {
        return write(source, uriRule, null);
    }

    @Override
    public String write(InputStream source, UriRuleable uriRule, String suffix) throws IOException {

        try {
            String uri = uriRule.getUri();
            if (suffix != null && !suffix.equals("")) {
                uri = uri + "." + suffix;
            }
            OutputStream target = FileUtils.openOutputStream(getLocalFile(uri));
            byte[] buff = new byte[1024 * 10];
            while (source.read(buff) > 0) {
                target.write(buff);
            }
            target.flush();
            target.close();
            source.close();

            return uri;
        } catch (PublishException e) {
            logger.error("Resource uri is error :{}", e);
            throw new IOException(e);
        }
    }

    @Override
    public void read(OutputStream output, String uri) throws IOException {

        byte[] buff = new byte[1024 * 10];
        InputStream input = new FileInputStream(getLocalFile(uri));
        while (input.read(buff) > 0) {
            output.write(buff);
        }
        output.flush();

        input.close();
        output.close();
    }

    @Override
    public void delete(String uri) throws IOException {
        FileUtils.forceDeleteOnExit(getLocalFile(uri));
    }

    /**
     * ?
     * 
     * @param uri ?
     * @return
     */
    private File getLocalFile(String uri) {
        String root = rootDirPath;
        root = StringUtils.removeStart(root, "/");
        root = StringUtils.removeEnd(root, "/");
        String path = StringUtils.isBlank(root) ? uri : ("/" + root + uri);
        logger.debug("Resource path is {}", path);
        return new File(path);
    }
}