Skip to content

Commit

Permalink
refactor: from and to validation
Browse files Browse the repository at this point in the history
  • Loading branch information
AleksandrMatsko committed Jan 23, 2025
1 parent 16bac93 commit a6eff63
Show file tree
Hide file tree
Showing 3 changed files with 54 additions and 35 deletions.
31 changes: 11 additions & 20 deletions api/handler/event.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,6 @@ import (
"fmt"
"net/http"
"regexp"
"strconv"
"time"

"github.com/go-graphite/carbonapi/date"

"github.com/go-chi/chi"
"github.com/go-chi/render"
Expand Down Expand Up @@ -52,23 +48,18 @@ func getEventsList(writer http.ResponseWriter, request *http.Request) {
page := middleware.GetPage(request)
fromStr := middleware.GetFromStr(request)
toStr := middleware.GetToStr(request)
var err error

if fromStr != "-inf" {
from := date.DateParamToEpoch(fromStr, "UTC", 0, time.UTC)
if from == 0 {
render.Render(writer, request, api.ErrorInvalidRequest(fmt.Errorf("can not parse from: %s", fromStr))) //nolint
return
}
fromStr = strconv.FormatInt(from, 10)
fromStr, err = validateFromStr(fromStr)
if err != nil {
render.Render(writer, request, api.ErrorInvalidRequest(err)) //nolint
return
}

if toStr != "+inf" {
to := date.DateParamToEpoch(toStr, "UTC", 0, time.UTC)
if to == 0 {
render.Render(writer, request, api.ErrorInvalidRequest(fmt.Errorf("can not parse to: %v", to))) //nolint
return
}
toStr = strconv.FormatInt(to, 10)
toStr, err = validateToStr(toStr)
if err != nil {
render.Render(writer, request, api.ErrorInvalidRequest(err)) //nolint
return
}

metricStr := middleware.GetMetric(request)
Expand All @@ -80,9 +71,9 @@ func getEventsList(writer http.ResponseWriter, request *http.Request) {

states := middleware.GetStates(request)

eventsList, err := controller.GetTriggerEvents(database, triggerID, page, size, fromStr, toStr, metricRegexp, states)
eventsList, errRsp := controller.GetTriggerEvents(database, triggerID, page, size, fromStr, toStr, metricRegexp, states)
if err != nil {
render.Render(writer, request, err) //nolint
render.Render(writer, request, errRsp) //nolint
return
}
if err := render.Render(writer, request, eventsList); err != nil {
Expand Down
25 changes: 10 additions & 15 deletions api/handler/triggers.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@ import (
"strings"
"time"

"github.com/go-graphite/carbonapi/date"
prometheus "github.com/prometheus/client_golang/api/prometheus/v1"

"github.com/go-chi/chi"
Expand Down Expand Up @@ -434,22 +433,18 @@ func getTriggerNoisiness(writer http.ResponseWriter, request *http.Request) {
toStr := middleware.GetToStr(request)
sort := middleware.GetSortOrder(request)

if fromStr != "-inf" {
from := date.DateParamToEpoch(fromStr, "UTC", 0, time.UTC)
if from == 0 {
render.Render(writer, request, api.ErrorInvalidRequest(fmt.Errorf("can not parse from: %s", fromStr))) //nolint
return
}
fromStr = strconv.FormatInt(from, 10)
var err error

fromStr, err = validateFromStr(fromStr)
if err != nil {
render.Render(writer, request, api.ErrorInvalidRequest(err)) //nolint
return
}

if toStr != "+inf" {
to := date.DateParamToEpoch(toStr, "UTC", 0, time.UTC)
if to == 0 {
render.Render(writer, request, api.ErrorInvalidRequest(fmt.Errorf("can not parse to: %v", to))) //nolint
return
}
toStr = strconv.FormatInt(to, 10)
toStr, err = validateToStr(toStr)
if err != nil {
render.Render(writer, request, api.ErrorInvalidRequest(err)) //nolint
return
}

triggersNoisinessList, errorResponse := controller.GetTriggerNoisiness(database, page, size, fromStr, toStr, sort)
Expand Down
33 changes: 33 additions & 0 deletions api/handler/validate.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
package handler

import (
"fmt"
"strconv"
"time"

"github.com/go-graphite/carbonapi/date"
)

func validateFromStr(fromStr string) (string, error) {
if fromStr != "-inf" {
from := date.DateParamToEpoch(fromStr, "UTC", 0, time.UTC)
if from == 0 {
return "", fmt.Errorf("can not parse from: %s", fromStr)
}
fromStr = strconv.FormatInt(from, 10)
}

return fromStr, nil
}

func validateToStr(toStr string) (string, error) {
if toStr != "+inf" {
to := date.DateParamToEpoch(toStr, "UTC", 0, time.UTC)
if to == 0 {
return "", fmt.Errorf("can not parse to: %v", to)
}
toStr = strconv.FormatInt(to, 10)
}

return toStr, nil
}

0 comments on commit a6eff63

Please sign in to comment.