/*
|
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the dreamlu.net developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: Chill 庄骞 (smallchill@163.com)
|
*/
|
package org.springblade.modules.resource.endpoint;
|
|
import com.alibaba.fastjson.JSON;
|
import io.minio.*;
|
import io.minio.errors.*;
|
import io.swagger.annotations.Api;
|
import lombok.AllArgsConstructor;
|
import lombok.SneakyThrows;
|
import org.apache.http.HttpEntity;
|
import org.apache.http.client.methods.CloseableHttpResponse;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.ContentType;
|
import org.apache.http.entity.StringEntity;
|
import org.apache.http.impl.client.CloseableHttpClient;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.http.util.EntityUtils;
|
import org.springblade.common.utils.arg;
|
import org.springblade.core.launch.constant.AppConstant;
|
import org.springblade.core.oss.model.BladeFile;
|
import org.springblade.core.oss.model.OssFile;
|
import org.springblade.core.secure.BladeUser;
|
import org.springblade.core.secure.annotation.PreAuth;
|
import org.springblade.core.secure.utils.AuthUtil;
|
import org.springblade.core.tenant.annotation.NonDS;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.constant.RoleConstant;
|
import org.springblade.core.tool.utils.FileUtil;
|
import org.springblade.core.tool.utils.Func;
|
import org.springblade.modules.resource.builder.oss.OssBuilder;
|
import org.springblade.modules.resource.entity.Attach;
|
import org.springblade.modules.resource.service.IAttachService;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.security.InvalidKeyException;
|
import java.security.NoSuchAlgorithmException;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.UUID;
|
|
/**
|
* 对象存储端点
|
*
|
* @author Chill
|
*/
|
@NonDS
|
@RestController
|
@AllArgsConstructor
|
@Api(value = "对象存储端点", tags = "对象存储端点")
|
@RequestMapping(AppConstant.APPLICATION_RESOURCE_NAME + "/oss/endpoint")
|
public class OssEndpoint {
|
|
/**
|
* 对象存储构建类
|
*/
|
private final OssBuilder ossBuilder;
|
|
/**
|
* 附件表服务
|
*/
|
private final IAttachService attachService;
|
|
/**
|
* 创建存储桶
|
*
|
* @param bucketName 存储桶名称
|
* @return Bucket
|
*/
|
@SneakyThrows
|
@PostMapping("/make-bucket")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
public R makeBucket(@RequestParam String bucketName) {
|
ossBuilder.template().makeBucket(bucketName);
|
return R.success("创建成功");
|
}
|
|
/**
|
* 创建存储桶
|
*
|
* @param bucketName 存储桶名称
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/remove-bucket")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
public R removeBucket(@RequestParam String bucketName) {
|
ossBuilder.template().removeBucket(bucketName);
|
return R.success("删除成功");
|
}
|
|
/**
|
* 拷贝文件
|
*
|
* @param fileName 存储桶对象名称
|
* @param destBucketName 目标存储桶名称
|
* @param destFileName 目标存储桶对象名称
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/copy-file")
|
public R copyFile(@RequestParam String fileName, @RequestParam String destBucketName, String destFileName) {
|
ossBuilder.template().copyFile(fileName, destBucketName, destFileName);
|
return R.success("操作成功");
|
}
|
|
/**
|
* 获取文件信息
|
*
|
* @param fileName 存储桶对象名称
|
* @return InputStream
|
*/
|
@SneakyThrows
|
@GetMapping("/stat-file")
|
public R<OssFile> statFile(@RequestParam String fileName) {
|
return R.data(ossBuilder.template().statFile(fileName));
|
}
|
|
/**
|
* 获取文件相对路径
|
*
|
* @param fileName 存储桶对象名称
|
* @return String
|
*/
|
@SneakyThrows
|
@GetMapping("/file-path")
|
public R<String> filePath(@RequestParam String fileName) {
|
return R.data(ossBuilder.template().filePath(fileName));
|
}
|
|
|
/**
|
* 获取文件外链
|
*
|
* @param fileName 存储桶对象名称
|
* @return String
|
*/
|
@SneakyThrows
|
@GetMapping("/file-link")
|
public R<String> fileLink(@RequestParam String fileName) {
|
return R.data(ossBuilder.template().fileLink(fileName));
|
}
|
|
/**
|
* 上传文件
|
*
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file")
|
public R<BladeFile> putFile(@RequestParam MultipartFile file) {
|
BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
|
return R.data(bladeFile);
|
}
|
|
|
/**
|
*
|
* app 文件上传
|
* @param file 图片对象
|
*/
|
@PostMapping("put-file-app")
|
public R putFileApp(@RequestParam MultipartFile file) throws IOException, ServerException, InsufficientDataException, InternalException, InvalidResponseException, InvalidKeyException, NoSuchAlgorithmException, XmlParserException, ErrorResponseException {
|
//填写你文件上传的地址以及相应信息
|
String url = "http://223.82.109.183:2081";
|
String access = "zhbaadmin";
|
String secret = "zhbapassword";
|
String bucket = "zhba";
|
MinioClient minioClient =
|
MinioClient.builder()
|
.endpoint(url)
|
.credentials(access, secret)
|
.build();
|
// 检查存储桶是否已经存在
|
boolean isExist = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucket).build());
|
if (!isExist) {
|
// 创建一个名为zip的存储桶，用于zip文件。
|
minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucket).build());
|
minioClient.setBucketPolicy(SetBucketPolicyArgs.builder().bucket(bucket).build());
|
}
|
String fileName = file.getOriginalFilename();
|
String newName = "upload/picture/" + UUID.randomUUID().toString().replaceAll("-", "")
|
+ fileName.substring(fileName.lastIndexOf("."));
|
InputStream in = file.getInputStream();
|
//创建头部信息
|
Map<String, String> headers = new HashMap<>(1 << 2);
|
//添加自定义内容类型
|
headers.put("Content-Type", "application/octet-stream");
|
//上传
|
minioClient.putObject(
|
PutObjectArgs.builder().bucket(bucket).object(newName).stream(
|
in, in.available(), -1)
|
.headers(headers)
|
.build());
|
in.close();
|
String urls = "http://223.82.109.183:2081/zhba/" + newName;
|
return R.data(urls);
|
}
|
|
|
/**
|
*
|
* 文件上传,自定义上传
|
* @param file 图片对象
|
*/
|
@PostMapping("put-files")
|
public R putFiles(@RequestParam MultipartFile file) throws IOException, ServerException, InsufficientDataException, InternalException, InvalidResponseException, InvalidKeyException, NoSuchAlgorithmException, XmlParserException, ErrorResponseException {
|
//填写你文件上传的地址以及相应信息
|
String url = "http://223.82.109.183:2081";
|
String access = "zhbaadmin";
|
String secret = "zhbapassword";
|
String bucket = "zhba";
|
MinioClient minioClient =
|
MinioClient.builder()
|
.endpoint(url)
|
.credentials(access, secret)
|
.build();
|
// 检查存储桶是否已经存在
|
boolean isExist = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucket).build());
|
if (!isExist) {
|
// 创建一个名为zip的存储桶，用于zip文件。
|
minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucket).build());
|
minioClient.setBucketPolicy(SetBucketPolicyArgs.builder().bucket(bucket).build());
|
}
|
String fileName = file.getOriginalFilename();
|
String newName = "upload/picture/" + UUID.randomUUID().toString().replaceAll("-", "")
|
+ fileName.substring(fileName.lastIndexOf("."));
|
InputStream in = file.getInputStream();
|
//创建头部信息
|
Map<String, String> headers = new HashMap<>(1 << 2);
|
//添加自定义内容类型
|
headers.put("Content-Type", "application/octet-stream");
|
//上传
|
minioClient.putObject(
|
PutObjectArgs.builder().bucket(bucket).object(newName).stream(
|
in, in.available(), -1)
|
.headers(headers)
|
.build());
|
in.close();
|
String urls = "http://223.82.109.183:2081/zhba/" + newName;
|
//数据封装
|
Map<String, Object> map = new HashMap<>(2);
|
map.put("name",newName);
|
map.put("url",urls);
|
//返回
|
return R.data(map);
|
}
|
|
/**
|
* 上传文件
|
*
|
* @param fileName 存储桶对象名称
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-by-name")
|
public R<BladeFile> putFile(@RequestParam String fileName, @RequestParam MultipartFile file) {
|
BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
|
return R.data(bladeFile);
|
}
|
|
/**
|
* 上传文件并保存至附件表
|
*
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-attach")
|
public R<BladeFile> putFileAttach(@RequestParam MultipartFile file,String deptid,String type,Long noticeId,String cardid) {
|
String fileName = file.getOriginalFilename();
|
BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
|
Long attachId = buildAttach(fileName, file.getSize(), bladeFile,deptid,type,noticeId,cardid);
|
bladeFile.setAttachId(attachId);
|
return R.data(bladeFile);
|
}
|
|
/**
|
* 上传文件并保存至附件表
|
*
|
* @param fileName 存储桶对象名称
|
* @param file 文件
|
* @return ObjectStat
|
*/
|
@SneakyThrows
|
@PostMapping("/put-file-attach-by-name")
|
public R<BladeFile> putFileAttach(@RequestParam String fileName, @RequestParam MultipartFile file,String deptid,String type,Long noticeId,String cardid) {
|
BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
|
Long attachId = buildAttach(fileName, file.getSize(), bladeFile,deptid,type,noticeId,cardid);
|
bladeFile.setAttachId(attachId);
|
return R.data(bladeFile);
|
}
|
|
/**
|
* 构建附件表
|
*
|
* @param fileName 文件名
|
* @param fileSize 文件大小
|
* @param bladeFile 对象存储文件
|
* @return attachId
|
*/
|
private Long buildAttach(String fileName, Long fileSize, BladeFile bladeFile,String deptid,String type,Long noticeId,String cardid) throws Exception {
|
//BladeUser user = AuthUtil.getUser();
|
String fileExtension = FileUtil.getFileExtension(fileName);
|
Attach attach = new Attach();
|
attach.setDomain(bladeFile.getDomain());
|
attach.setLink(bladeFile.getLink());
|
attach.setName(bladeFile.getName());
|
attach.setOriginalName(bladeFile.getOriginalName());
|
attach.setAttachSize(fileSize);
|
attach.setExtension(fileExtension);
|
attach.setDeptid(deptid);
|
attach.setCardid(cardid);
|
attach.setType(type);
|
if (null!=noticeId){
|
attach.setNoticeId(noticeId);
|
}
|
attachService.save(attach);
|
return attach.getId();
|
}
|
|
/**
|
* 删除文件
|
*
|
* @param fileName 存储桶对象名称
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/remove-file")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
public R removeFile(@RequestParam String fileName) {
|
ossBuilder.template().removeFile(fileName);
|
return R.success("操作成功");
|
}
|
|
/**
|
* 批量删除文件
|
*
|
* @param fileNames 存储桶对象名称集合
|
* @return R
|
*/
|
@SneakyThrows
|
@PostMapping("/remove-files")
|
@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
|
public R removeFiles(@RequestParam String fileNames) {
|
ossBuilder.template().removeFiles(Func.toStrList(fileNames));
|
return R.success("操作成功");
|
}
|
|
}
|
