com.jeesuite.cos.provider.AbstractProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-cos-adapter Show documentation
云存储适配器，支持七牛、腾讯云，阿里云
The newest version!
package com.jeesuite.cos.provider;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.Validate;

import com.jeesuite.common.util.HttpUtils;
import com.jeesuite.cos.BucketConfig;
import com.jeesuite.cos.CosProvider;
import com.jeesuite.cos.CosProviderConfig;

/**
 * @description 

 * @author vakin
 * @date 2017年1月7日
 */
public abstract class AbstractProvider implements CosProvider{

	protected static final String URL_PREFIX_PATTERN = "(http).*\\.(com|cn)\\/";
	protected static final String HTTP_PREFIX = "http://";
	protected static final String HTTPS_PREFIX = "https://";
	protected static final String DIR_SPLITER = "/";
	private Map bucketMappings = new HashMap<>();
	
	
	protected CosProviderConfig conf;
	
	
	public AbstractProvider(CosProviderConfig conf) {
		Validate.notBlank(conf.getAccessKey(), "[accessKey] not defined");
		Validate.notBlank(conf.getSecretKey(), "[secretKey] not defined");
		this.conf = conf;
		
		List bucketConfigs = conf.getBucketConfigs();
		for (BucketConfig bucketConfig : bucketConfigs) {
			addBucketConfig(bucketConfig);
		}
	}


	protected String getFullPath(String bucketName,String file) {
		if(file.startsWith(HTTP_PREFIX) || file.startsWith(HTTPS_PREFIX)){
			return file;
		}
		return getBucketUrlPrefix(bucketName) + file;
	}
	
	protected String resolveFileKey(String bucketName,String fileUrl) {
		if(!fileUrl.startsWith(HTTP_PREFIX) && !fileUrl.startsWith(HTTPS_PREFIX)){
			return fileUrl;
		}
		String urlprefix = getBucketUrlPrefix(bucketName);
		return fileUrl.replace(urlprefix, StringUtils.EMPTY);
	}

	@Override
	public String downloadAndSaveAs(String bucketName,String file, String localSaveDir) {
		return HttpUtils.downloadFile(getDownloadUrl(bucketName,file,300), localSaveDir);
	}
	
	@Override
	public String getDownloadUrl(String bucketName,String fileKey, int expireInSeconds) {
		BucketConfig config = currentBucketConfig(bucketName);
		String url;
		if(config.isAuth()){
			fileKey = resolveFileKey(bucketName, fileKey);
			String presignedUrl = generatePresignedUrl(bucketName,fileKey,expireInSeconds);
			String urlprefix = getBucketUrlPrefix(bucketName);
			url = presignedUrl.replaceFirst(URL_PREFIX_PATTERN, urlprefix);
		}else{
			url = getFullPath(bucketName,fileKey);
		}
		return url;
	}


	/**
	 * @param bucketName
	 * @return
	 */
	protected String getBucketUrlPrefix(String bucketName) {
		BucketConfig config = currentBucketConfig(bucketName);
		if(StringUtils.isNotBlank(config.getUrlPrefix())) {
			return config.getUrlPrefix();
		}
		synchronized (bucketMappings) {
			if(StringUtils.isNotBlank(config.getUrlPrefix())) {
				return config.getUrlPrefix();
			}
			String urlPrefix = buildBucketUrlPrefix(bucketName);
			config.setUrlPrefix(urlPrefix);
			return urlPrefix;
		}
	}
	
	protected String buildBucketName(String bucketName) {
		if(StringUtils.isBlank(bucketName)){
			throw new IllegalArgumentException("[bucketName] not defined");
		}
		return bucketName;
	}
	
	public BucketConfig currentBucketConfig(String bucketName) {
		BucketConfig config = bucketMappings.get(buildBucketName(bucketName));
		if(config == null) {
			synchronized (bucketMappings) {
				config = getBucketConfig(bucketName);
				if(config != null) {
					addBucketConfig(config);
				}
			}
		}
		if(config == null) {
			throw new IllegalArgumentException("[bucketName] not exists");
		}
		return config;
	}
	
	public void addBucketConfig(BucketConfig bucketConfig) {
		if(bucketConfig == null || bucketConfig.getName() == null)return;
		if(StringUtils.isNotBlank(bucketConfig.getUrlPrefix()) && bucketConfig.getUrlPrefix().endsWith("/")) {
			bucketConfig.setUrlPrefix(bucketConfig.getUrlPrefix().substring(0,bucketConfig.getUrlPrefix().length() - 1));
		}
		bucketConfig.setName(buildBucketName(bucketConfig.getName()));
		bucketMappings.put(bucketConfig.getName(), bucketConfig);
	}


	protected abstract String buildBucketUrlPrefix(String bucketName);
	protected abstract String generatePresignedUrl(String bucketName,String fileKey, int expireInSeconds);

	
}