Minio预签名URL自定义响应文件名之Minio源码改造
需求说明
用户上传文件到Minio时,一般存储在Minio中的对象名称都是后端以UUID或者其他随机或非随机方案生成的唯一标识做为文件名,这个对象名称一般都不会是用户上传时的原文件名称。
在用户下载时,想让文件流不通过后端服务器,而是用户直接申请并使用某个要下载对象的Minio预签名的url,直接从Minio所部署的服务器下载该文件。
但是浏览器通过预签名的url下载文件时,由于无法自定义Minio下载文件的请求响应头中的文件名称,所以在浏览器下载时,保存的文件名是以对象名称进行保存的,那么这个文件名是对用户感知等都是不友好的。
所以需要根据预签名url下载文件(我这里是用GET)请求中的filename参数,把响应头的Content-Disposition内容上指定文件名称。
注:本方案是修改Minio源代码实现该功能,因为Minio好像没有实现这个功能
Minio源码改造
一、环境准备
这里大家自己百度查询相关教程
- Go语言环境,并设置好Go的依赖下载代理
go env -w GO111MODULE=on go env -w GOPROXY=https://goproxy.io,direct
- GoLand编辑器(当然用别的也行)
我使用的go版本:go version go1.21.0 linux/amd64
二、下载Minio源代码
使用git下载
git clone https://github.com/minio/minio.git
三、修改源代码
1.修改cmd目录下的api-router.go这个代码文件
搜索改文件内容:GetObjectHandler
进入api.GetObjectHandler这个handler查看代码,在这个函数前面加上一个自定义的函数GetUrlArgs,这个函数的功能是获取GET请求url中的某个参数值
/** 获取URL的get参数 */ func GetUrlArgs(r *http.Request, name string) string { var arg string values := r.URL.Query() arg = values.Get(name) return arg }
2.将filename参数值设置到响应头
在这个函数里面,添加这段代码,作用是将请求url中的filename参数值设置到响应头的Content-Disposition中
// 将filename参数值添加到响应头做为响应名 var filename string = GetUrlArgs(r, "filename") if filename != "" { // 该写法可以解决中文文件名乱码问题 w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename*=UTF-8''%s", url.QueryEscape(filename))) }
4.修改验证签名时是否需要带入filename参数验证
如果你在获取预签名url时,就携带了filename参数,就不用做这一步了,那么这个预签名url的任何参数值都是不能被更改的,包括filename。
例如我的java minio客户端获取预签名url代码:
Map<String, String> query = new HashMap<>(2, 1); query.put("filename", responseFileName); GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder() .bucket(bucketName) .object(objectName) // 对象名称 .expiry(10, TimeUnit.SECONDS) // 该url签名10秒过期 .method(Method.GET) // 该url允许的请求方式 .extraQueryParams(query) // 把filename加在查询字符串上 .build(); // 创建预签名url String preSignedObjectUrl = minioClient.getPresignedObjectUrl(args);
如果你在获取预签名url时根本就没有携带上filename进行url预签名,而是想让前端或者请求者在url上加上filename=xxx这个参数,文件名可以随便由请求者设置,那么就需要修改Minio源码中的验证签名操作,让filename这个参数不参与到验证签名中。
例如这是java minio客户端不带filename参数创建预签名url的代码:
GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder() .bucket(bucketName) .object(objectName) // 对象名称 .expiry(10, TimeUnit.SECONDS) // 该url签名10秒过期 .method(Method.GET) // 该url允许的请求方式 .build(); // 创建预签名url String preSignedObjectUrl = minioClient.getPresignedObjectUrl(args);
修改Minio源码不让filename参与到验证签名的操作步骤:
代码文件:cmd/signature-v4.go
搜索函数:doesPresignedSignatureMatch
在该函数的这个循环中,修改这个循环,判断条件中加入参数k不是filename的条件即可,加入的是图中红色方框这里的代码
//Add missing query parameters if any provided in the request URL for k, v := range req.Form { if !defaultSigParams.Contains(k) && k != "filename" { query[k] = v } }
四、大功告成,编译go代码生成可执行文件
按需执行以下命令,下载好相关依赖之后,就会编译生成一个可执行文件minio,当然这个可执行文件名可以在命令中修改,打包之后,这个可执行文件就是可以和官方下载的可执行文件一样运行了,依旧按照官方的文档使用。
本机运行打包命令
go build main.go
ARM打包命令
# CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -o 可执行文件名 main.go CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -o minio main.go
AMD打包命令
# CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o 可执行文件名 main.go CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o minio main.go
五、使用预签名URL下载文件测试
我获取预签名url是设置的filename是a.txt,存储在Minio中是一大长串xxxxxx.txt,浏览器保存的文件名也是a.txt。如果是中文名的话会自动进行编码后响应,浏览器能自动转码为中文文件名进行保存。