simon987/ws_bucket
1
0
Fork 0
mirror of https://github.com/simon987/ws_bucket.git synced 2025-04-10 14:06:46 +00:00
Code Issues Projects Releases Wiki Activity

Accept token in query string

Browse Source
This commit is contained in:
simon987 2019-03-24 12:41:58 -04:00
parent d9e8f6a53b
commit b3e4f8dd3c
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/slot.go
View File

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