simon987/ws_bucket
1
0
Fork 0
mirror of https://github.com/simon987/ws_bucket.git synced 2025-12-14 07:09:07 +00:00
Code Issues Projects Releases Wiki Activity

Allow passing the upload token as query string for reading

Browse Source
This commit is contained in:
simon987
2019-03-10 11:14:44 -04:00
parent b05c9f2b1a
commit c41ed01184
2 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/slot.go
View File

@@ -142,6 +142,9 @@ func (api *WebApi) Upload(ctx *fasthttp.RequestCtx) {
func (api *WebApi) ReadUploadSlot(ctx *fasthttp.RequestCtx) {
tokenStr := string(ctx.Request.Header.Peek("X-Upload-Token"))
if tokenStr == "" {
tokenStr = string(ctx.Request.URI().QueryArgs().Peek("token"))
}
slot := UploadSlot{}
err := api.db.Where("token=?", tokenStr).First(&slot).Error