com.jeesuite.cos.CUploadObject 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;

import java.io.File;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonInclude.Include;

import net.sf.jmimemagic.Magic;
import net.sf.jmimemagic.MagicMatch;

/**
 * 
 * @description 

 * @author vakin
 * @date 2017年1月5日
 */
@JsonInclude(Include.NON_NULL)
public class CUploadObject {

	private String bucketName;
	private String fileKey;
	private String extension;
	private String mimeType;
	@JsonIgnore
	private String folderPath;
	@JsonIgnore
	private byte[] bytes;
	@JsonIgnore
	private File file;
	private long fileSize;
	@JsonIgnore
	private InputStream inputStream;
	private Map metadata;

	public CUploadObject(String filePath) {
		this(new File(filePath));
	}

	public CUploadObject(File file) {
		this.file = file;
		this.fileSize = file.length();
		
		if(file.getName().contains(FilePathHelper.DOT)){
			this.extension = file.getName().substring(file.getName().lastIndexOf(FilePathHelper.DOT) + 1);
			this.mimeType = MimeTypeFileExtensionConvert.getFileMimeType(extension);
		}
		
		try {
			if(this.mimeType == null) {
				MagicMatch magicMatch = Magic.getMagicMatch(file, true, true);
				this.mimeType = StringUtils.trimToNull(magicMatch.getMimeType());
				if(StringUtils.isNotBlank(magicMatch.getExtension())) {
					extension = magicMatch.getExtension();
				}
			}
		} catch (Exception e) {
			System.err.println(e.getMessage());
		}
	}

	public CUploadObject(InputStream inputStream,long fileSize, String mimeType,String originFileName) {
		this.inputStream = inputStream;
		this.mimeType = mimeType;
		this.fileSize = fileSize;
		if(StringUtils.isNotBlank(originFileName) && originFileName.contains(FilePathHelper.DOT)){
			extension = originFileName.substring(originFileName.lastIndexOf(FilePathHelper.DOT));
		}else if(mimeType != null) {
			this.extension = MimeTypeFileExtensionConvert.getFileExtension(mimeType);
		}
	}

	public CUploadObject(byte[] bytes, String mimeType) {
		this.bytes = bytes;
		this.mimeType = mimeType;
		this.fileSize = bytes.length;
		if(mimeType != null) {
			this.extension = MimeTypeFileExtensionConvert.getFileExtension(mimeType);
		}
		try {
			if(this.extension == null) {
				MagicMatch magicMatch = Magic.getMagicMatch(bytes);
				extension = StringUtils.trimToNull(magicMatch.getExtension());
				if(StringUtils.isNotBlank(magicMatch.getMimeType())) {
					this.mimeType = magicMatch.getMimeType();
				}
			}
		} catch (Exception e) {
			System.err.println(e.getMessage());
		}
	}
	

	/**
	 * @return the bucketName
	 */
	public String getBucketName() {
		return bucketName;
	}
	/**
	 * @return the fileSize
	 */
	public long getFileSize() {
		return fileSize;
	}

	public byte[] getBytes() {
		return bytes;
	}

	public File getFile() {
		return file;
	}

	public InputStream getInputStream() {
		return inputStream;
	}

	public Map getMetadata() {
		return metadata;
	}

	public void setString(String mimeType) {
		this.mimeType = mimeType;
	}

	public void setMetadata(Map metadata) {
		this.metadata = metadata;
	}

	public CUploadObject addMetaData(String key, Object value) {
		if(metadata == null)metadata = new HashMap<>();
		metadata.put(key, value);
		return this;
	}
	
	public String getMimeType(){
		return mimeType;
	}

	public CUploadObject folderPath(String folderPath) {
		this.folderPath = folderPath;
		return this;
	}
	
	public CUploadObject fileKey(String fileKey) {
		this.fileKey = fileKey;
		return this;
	}
	
	public CUploadObject bucketName(String bucketName) {
		this.bucketName = bucketName;
		return this;
	}

	public String getFileKey() {
		if(StringUtils.isBlank(fileKey)) {
			StringBuilder builder = new StringBuilder();
			if(StringUtils.isNotBlank(this.folderPath)){
				builder.append(FilePathHelper.formatDirectoryPath(folderPath));
			}
			builder.append(FilePathHelper.genTimePathRandomFilePath(extension));
			fileKey = builder.toString();
		}
		
		if(!fileKey.startsWith(FilePathHelper.DIR_SPLITER)) {
			fileKey = FilePathHelper.DIR_SPLITER + fileKey;
		}
		
		return fileKey;
	}

}