net.shopxx.plugin.ossStorage.OssStoragePlugin.java Source code

Introduction

Here is the source code for net.shopxx.plugin.ossStorage.OssStoragePlugin.java

Source

/*
 * Copyright 2005-2015 shopxx.net. All rights reserved.
 * Support: http://3936242.01p.com/
 * License: http://3936242.01p.com/license
 */
package net.shopxx.plugin.ossStorage;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.InputStream;

import net.shopxx.entity.PluginConfig;
import net.shopxx.plugin.StoragePlugin;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.springframework.stereotype.Component;

import com.aliyun.oss.OSSClient;
import com.aliyun.oss.model.ObjectMetadata;

@Component("ossStoragePlugin")
public class OssStoragePlugin extends StoragePlugin {

    @Override
    public String getName() {
        return "";
    }

    @Override
    public String getVersion() {
        return "1.0";
    }

    @Override
    public String getAuthor() {
        return "xiaohe";
    }

    @Override
    public String getSiteUrl() {
        return "http://3936242.01p.com/";
    }

    @Override
    public String getInstallUrl() {
        return "oss_storage/install.jhtml";
    }

    @Override
    public String getUninstallUrl() {
        return "oss_storage/uninstall.jhtml";
    }

    @Override
    public String getSettingUrl() {
        return "oss_storage/setting.jhtml";
    }

    @Override
    public void upload(String path, File file, String contentType) {
        PluginConfig pluginConfig = getPluginConfig();
        if (pluginConfig != null) {
            String endpoint = pluginConfig.getAttribute("endpoint");
            String accessId = pluginConfig.getAttribute("accessId");
            String accessKey = pluginConfig.getAttribute("accessKey");
            String bucketName = pluginConfig.getAttribute("bucketName");
            InputStream inputStream = null;
            try {
                inputStream = new BufferedInputStream(new FileInputStream(file));
                OSSClient ossClient = new OSSClient(endpoint, accessId, accessKey);
                ObjectMetadata objectMetadata = new ObjectMetadata();
                objectMetadata.setContentType(contentType);
                objectMetadata.setContentLength(file.length());
                ossClient.putObject(bucketName, StringUtils.removeStart(path, "/"), inputStream, objectMetadata);
            } catch (FileNotFoundException e) {
                throw new RuntimeException(e.getMessage(), e);
            } finally {
                IOUtils.closeQuietly(inputStream);
            }
        }
    }

    @Override
    public String getUrl(String path) {
        PluginConfig pluginConfig = getPluginConfig();
        if (pluginConfig != null) {
            String urlPrefix = pluginConfig.getAttribute("urlPrefix");
            return urlPrefix + path;
        }
        return null;
    }

}