From e5d4897bfa7347100c39ea9504f6dcaeae247dbc Mon Sep 17 00:00:00 2001 From: SoDoEasy Date: Wed, 15 Sep 2021 12:03:40 +0800 Subject: [PATCH] =?UTF-8?q?SSO=E7=B1=BB=E5=9E=8B=20=E5=A2=9E=E5=8A=A0MiniO?= =?UTF-8?q?=E5=AF=B9=E8=B1=A1=E5=AD=98=E5=82=A8=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tduck-storage/pom.xml | 7 +- .../storage/cloud/MIniOStorageService.java | 69 ++++++ .../storage/cloud/OssStorageFactory.java | 2 + .../storage/entity/enums/MimeTypeEnum.java | 216 ++++++++++++++++++ .../storage/entity/enums/OssTypeEnum.java | 6 +- 5 files changed, 298 insertions(+), 2 deletions(-) create mode 100644 tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/MIniOStorageService.java create mode 100644 tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/MimeTypeEnum.java diff --git a/tduck-storage/pom.xml b/tduck-storage/pom.xml index b472bc4..d68dfe3 100644 --- a/tduck-storage/pom.xml +++ b/tduck-storage/pom.xml @@ -38,5 +38,10 @@ com.upyun java-sdk + + io.minio + minio + 7.0.2 + - \ No newline at end of file + diff --git a/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/MIniOStorageService.java b/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/MIniOStorageService.java new file mode 100644 index 0000000..4f6da96 --- /dev/null +++ b/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/MIniOStorageService.java @@ -0,0 +1,69 @@ +package com.tduck.cloud.storage.cloud; + +import com.tduck.cloud.storage.entity.enums.MimeTypeEnum; +import com.tduck.cloud.storage.exception.StorageException; +import io.minio.MinioClient; +import io.minio.PutObjectOptions; +import io.minio.errors.InvalidEndpointException; +import io.minio.errors.InvalidPortException; + +import java.io.ByteArrayInputStream; +import java.io.InputStream; + +/** + * @author pfwang + */ +public class MIniOStorageService extends OssStorageService { + + private MinioClient client; + + public MIniOStorageService(OssStorageConfig config) { + this.config = config; + //初始化 + init(); + } + + private void init() { + try { + client = new MinioClient(config.getEndpoint(), config.getAccessKeyId(), config.getAccessKeySecret(), false); + } catch (InvalidEndpointException e) { + e.printStackTrace(); + } catch (InvalidPortException e) { + e.printStackTrace(); + } + } + + @Override + public String upload(InputStream inputStream, String path) { + try { + PutObjectOptions poo = new PutObjectOptions(inputStream.available(), -1); + poo.setContentType(MimeTypeEnum.getContentType(path)); + client.putObject(config.getBucketName(), path, inputStream, poo); + } catch (Exception e) { + throw new StorageException("上传文件失败，请检查配置信息", e); + } + return config.getDomain() + "/" + path; + } + + @Override + public String upload(byte[] data, String path) { + try { + PutObjectOptions poo = new PutObjectOptions(data.length, -1); + poo.setContentType(MimeTypeEnum.getContentType(path)); + client.putObject(config.getBucketName(), path, new ByteArrayInputStream(data), poo); + } catch (Exception e) { + throw new StorageException("上传文件失败，请检查配置信息", e); + } + return config.getDomain() + "/" + path; + } + + @Override + public void delete(String path) { + try { + client.removeObject(config.getBucketName(), path); + } catch (Exception e) { + throw new StorageException("删除文件失败", e); + } + } + +} diff --git a/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/OssStorageFactory.java b/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/OssStorageFactory.java index 6c54f87..4c2e7ef 100644 --- a/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/OssStorageFactory.java +++ b/tduck-storage/src/main/java/com/tduck/cloud/storage/cloud/OssStorageFactory.java @@ -26,6 +26,8 @@ public final class OssStorageFactory { return new UpyunStorageService(config); } else if (config.getOssType() == OssTypeEnum.LOCAL) { return new localStorageService(config); + }else if (config.getOssType() == OssTypeEnum.MINIO) { + return new MIniOStorageService(config); } return null; } diff --git a/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/MimeTypeEnum.java b/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/MimeTypeEnum.java new file mode 100644 index 0000000..d6efc27 --- /dev/null +++ b/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/MimeTypeEnum.java @@ -0,0 +1,216 @@ +package com.tduck.cloud.storage.entity.enums; + +import lombok.Getter; +import org.apache.commons.lang3.StringUtils; + +/** + * 文件后缀名与ContentType + * @author pfwang + */ +@Getter +public enum MimeTypeEnum { + + AAC("acc", "AAC音频", "audio/aac"), + + ABW("abw", "AbiWord文件", "application/x-abiword"), + + ARC("arc", "存档文件", "application/x-freearc"), + + AVI("avi", "音频视频交错格式", "video/x-msvideo"), + + AZW("azw", "亚马逊Kindle电子书格式", "application/vnd.amazon.ebook"), + + BIN("bin", "任何类型的二进制数据", "application/octet-stream"), + + BMP("bmp", "Windows OS / 2位图图形", "image/bmp"), + + BZ("bz", "BZip存档", "application/x-bzip"), + + BZ2("bz2", "BZip2存档", "application/x-bzip2"), + + CSH("csh", "C-Shell脚本", "application/x-csh"), + + CSS("css", "级联样式表（CSS）", "text/css"), + + CSV("csv", "逗号分隔值（CSV）", "text/csv"), + + DOC("doc", "微软Word文件", "application/msword"), + + DOCX("docx", "Microsoft Word（OpenXML）", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"), + + EOT("eot", "MS Embedded OpenType字体", "application/vnd.ms-fontobject"), + + EPUB("epub", "电子出版物（EPUB）", "application/epub+zip"), + + GZ("gz", "GZip压缩档案", "application/gzip"), + + GIF("gif", "图形交换格式（GIF）", "image/gif"), + + HTM("htm", "超文本标记语言（HTML）", "text/html"), + + HTML("html", "超文本标记语言（HTML）", "text/html"), + + ICO("ico", "图标格式", "image/vnd.microsoft.icon"), + + ICS("ics", "iCalendar格式", "text/calendar"), + + JAR("jar", "Java存档", "application/java-archive"), + + JPEG("jpeg", "JPEG图像", "image/jpeg"), + + JPG("jpg", "JPEG图像", "image/jpeg"), + + JS("js", "JavaScript", "text/javascript"), + + JSON("json", "JSON格式", "application/json"), + + JSONLD("jsonld", "JSON-LD格式", "application/ld+json"), + + MID("mid", "乐器数字接口（MIDI）", "audio/midi"), + + MIDI("midi", "乐器数字接口（MIDI）", "audio/midi"), + + MJS("mjs", "JavaScript模块", "text/javascript"), + + MP3("mp3", "MP3音频", "audio/mpeg"), + + MPEG("mpeg", "MPEG视频", "video/mpeg"), + + MPKG("mpkg", "苹果安装程序包", "application/vnd.apple.installer+xml"), + + ODP("odp", "OpenDocument演示文稿文档", "application/vnd.oasis.opendocument.presentation"), + + ODS("ods", "OpenDocument电子表格文档", "application/vnd.oasis.opendocument.spreadsheet"), + + ODT("odt", "OpenDocument文字文件", "application/vnd.oasis.opendocument.text"), + + OGA("oga", "OGG音讯", "audio/ogg"), + + OGV("ogv", "OGG视频", "video/ogg"), + + OGX("ogx", "OGG", "application/ogg"), + + OPUS("opus", "OPUS音频", "audio/opus"), + + OTF("otf", "otf字体", "font/otf"), + + PNG("png", "便携式网络图形", "image/png"), + + PDF("pdf", "Adobe 可移植文档格式（PDF）", "application/pdf"), + + PHP("php", "php", "application/x-httpd-php"), + + PPT("ppt", "Microsoft PowerPoint", "application/vnd.ms-powerpoint"), + + PPTX("pptx", "Microsoft PowerPoint（OpenXML）", "application/vnd.openxmlformats-officedocument.presentationml.presentation"), + + RAR("rar", "RAR档案", "application/vnd.rar"), + + RTF("rtf", "富文本格式", "application/rtf"), + + SH("sh", "Bourne Shell脚本", "application/x-sh"), + + SVG("svg", "可缩放矢量图形（SVG）", "image/svg+xml"), + + SWF("swf", "小型Web格式（SWF）或Adobe Flash文档", "application/x-shockwave-flash"), + + TAR("tar", "磁带存档（TAR）", "application/x-tar"), + + TIF("tif", "标记图像文件格式（TIFF）", "image/tiff"), + + TIFF("tiff", "标记图像文件格式（TIFF）", "image/tiff"), + + TS("ts", "MPEG传输流", "video/mp2t"), + + TTF("ttf", "ttf字体", "font/ttf"), + + TXT("txt", "文本（通常为ASCII或ISO 8859- n", "text/plain"), + + VSD("vsd", "微软Visio", "application/vnd.visio"), + + WAV("wav", "波形音频格式", "audio/wav"), + + WEBA("weba", "WEBM音频", "audio/webm"), + + WEBM("webm", "WEBM视频", "video/webm"), + + WEBP("webp", "WEBP图像", "image/webp"), + + WOFF("woff", "Web开放字体格式（WOFF）", "font/woff"), + + WOFF2("woff2", "Web开放字体格式（WOFF）", "font/woff2"), + + XHTML("xhtml", "XHTML", "application/xhtml+xml"), + + XLS("xls", "微软Excel", "application/vnd.ms-excel"), + + XLSX("xlsx", "微软Excel（OpenXML）", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"), + + XML("xml", "XML", "application/xml"), + + XUL("xul", "XUL", "application/vnd.mozilla.xul+xml"), + + ZIP("zip", "ZIP", "application/zip"), + + MIME_3GP("3gp", "3GPP audio/video container", "video/3gpp"), + + MIME_3GP_WITHOUT_VIDEO("3gp", "3GPP audio/video container doesn't contain video", "audio/3gpp2"), + + MIME_3G2("3g2", "3GPP2 audio/video container", "video/3gpp2"), + + MIME_3G2_WITHOUT_VIDEO("3g2", "3GPP2 audio/video container doesn't contain video", "audio/3gpp2"), + + MIME_7Z("7z", "7-zip存档", "application/x-7z-compressed"); + + //扩展名 + private final String extension; + //说明 + private final String explain; + //contentType/mime类型 + private final String mimeType; + + /** + * @param extension 上传的文件扩展名 + * @param explain 类型说明 + * @param mimeType Mime对应的类型 + */ + MimeTypeEnum(String extension, String explain, String mimeType) { + this.extension = extension; + this.explain = explain; + this.mimeType = mimeType; + } + + /** + * 通过扩展名获取枚举类型 + * + * @param extension 扩展名 + * @return 枚举类 + */ + public static MimeTypeEnum getByExtension(String extension) { + if (StringUtils.isEmpty(extension)) { + return null; + } + for (MimeTypeEnum typesEnum : MimeTypeEnum.values()) { + if (extension.equals(typesEnum.getExtension())) { + return typesEnum; + } + } + return null; + } + + /** + * Content-Type常用对照 + * 根据后缀获取Mime + * + * @param path 文件名 + * @return mime类型 + */ + public static String getContentType(String path) { + String fileType = path.substring(path.lastIndexOf(".") + 1); + MimeTypeEnum mimeTypeEnum = MimeTypeEnum.getByExtension(fileType); + if (mimeTypeEnum != null) { + return mimeTypeEnum.getMimeType(); + } + return "application/octet-stream"; + } +} diff --git a/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/OssTypeEnum.java b/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/OssTypeEnum.java index c0eae9b..8fe18a4 100644 --- a/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/OssTypeEnum.java +++ b/tduck-storage/src/main/java/com/tduck/cloud/storage/entity/enums/OssTypeEnum.java @@ -29,8 +29,12 @@ public enum OssTypeEnum { /** * 本地存储 */ - LOCAL(3); + LOCAL(3), + /** + * minio + */ + MINIO(4); @EnumValue @JsonValue