From 02e02718d2097061778667befcf491a32a5a9998 Mon Sep 17 00:00:00 2001 From: Nimrod Gilboa Markevich <59927337+nimrod-up9@users.noreply.github.com> Date: Sun, 8 Aug 2021 14:36:24 +0300 Subject: [PATCH] Fixed fetch not using from/to options (#179) --- agent/pkg/controllers/entries_controller.go | 16 ++++++++-------- agent/pkg/models/models.go | 14 +++++++------- 2 files changed, 15 insertions(+), 15 deletions(-) diff --git a/agent/pkg/controllers/entries_controller.go b/agent/pkg/controllers/entries_controller.go index 7eda00063..a1ad22d42 100644 --- a/agent/pkg/controllers/entries_controller.go +++ b/agent/pkg/controllers/entries_controller.go @@ -57,7 +57,7 @@ func GetEntries(c *gin.Context) { } func GetHARs(c *gin.Context) { - entriesFilter := &models.HarFetchRequestBody{} + entriesFilter := &models.HarFetchRequestQuery{} order := database.OrderDesc if err := c.BindQuery(entriesFilter); err != nil { c.JSON(http.StatusBadRequest, err) @@ -146,12 +146,12 @@ func GetHARs(c *gin.Context) { func UploadEntries(c *gin.Context) { rlog.Infof("Upload entries - started\n") - uploadRequestBody := &models.UploadEntriesRequestBody{} - if err := c.BindQuery(uploadRequestBody); err != nil { + uploadParams := &models.UploadEntriesRequestQuery{} + if err := c.BindQuery(uploadParams); err != nil { c.JSON(http.StatusBadRequest, err) return } - if err := validation.Validate(uploadRequestBody); err != nil { + if err := validation.Validate(uploadParams); err != nil { c.JSON(http.StatusBadRequest, err) return } @@ -160,19 +160,19 @@ func UploadEntries(c *gin.Context) { return } - rlog.Infof("Upload entries - creating token. dest %s\n", uploadRequestBody.Dest) - token, err := up9.CreateAnonymousToken(uploadRequestBody.Dest) + rlog.Infof("Upload entries - creating token. dest %s\n", uploadParams.Dest) + token, err := up9.CreateAnonymousToken(uploadParams.Dest) if err != nil { c.String(http.StatusServiceUnavailable, "Cannot analyze, mizu is already analyzing") return } rlog.Infof("Upload entries - uploading. token: %s model: %s\n", token.Token, token.Model) - go up9.UploadEntriesImpl(token.Token, token.Model, uploadRequestBody.Dest, uploadRequestBody.SleepIntervalSec) + go up9.UploadEntriesImpl(token.Token, token.Model, uploadParams.Dest, uploadParams.SleepIntervalSec) c.String(http.StatusOK, "OK") } func GetFullEntries(c *gin.Context) { - entriesFilter := &models.HarFetchRequestBody{} + entriesFilter := &models.HarFetchRequestQuery{} if err := c.BindQuery(entriesFilter); err != nil { c.JSON(http.StatusBadRequest, err) } diff --git a/agent/pkg/models/models.go b/agent/pkg/models/models.go index 13013f819..0648d4f5c 100644 --- a/agent/pkg/models/models.go +++ b/agent/pkg/models/models.go @@ -119,19 +119,19 @@ func (fedex *FullEntryDetailsExtra) UnmarshalData(entry *MizuEntry) error { } type EntriesFilter struct { - Limit int `query:"limit" validate:"required,min=1,max=200"` - Operator string `query:"operator" validate:"required,oneof='lt' 'gt'"` - Timestamp int64 `query:"timestamp" validate:"required,min=1"` + Limit int `form:"limit" validate:"required,min=1,max=200"` + Operator string `form:"operator" validate:"required,oneof='lt' 'gt'"` + Timestamp int64 `form:"timestamp" validate:"required,min=1"` } -type UploadEntriesRequestBody struct { +type UploadEntriesRequestQuery struct { Dest string `form:"dest"` SleepIntervalSec int `form:"interval"` } -type HarFetchRequestBody struct { - From int64 `query:"from"` - To int64 `query:"to"` +type HarFetchRequestQuery struct { + From int64 `form:"from"` + To int64 `form:"to"` } type WebSocketEntryMessage struct {