weather-api/storage/weather-query.go
2021-04-05 11:39:55 +02:00

68 lines
1.4 KiB
Go

package storage
import (
"fmt"
"net/url"
"strconv"
"time"
"github.com/google/uuid"
)
type WeatherQuery struct {
Start time.Time
End time.Time
SensorId uuid.UUID
Values map[SensorValueType]bool
}
//NewWeatherQuery creates a new empty WeatherQuery
func NewWeatherQuery() *WeatherQuery {
query := new(WeatherQuery)
query.Values = make(map[SensorValueType]bool)
return query
}
func (query *WeatherQuery) Init() {
query.Start = time.Now().Add(-1 * time.Hour * 24 * 14)
query.End = time.Now()
query.SensorId = uuid.Nil
for _, sensorValueType := range GetSensorValueTypes() {
query.Values[sensorValueType] = true
}
}
func ParseFromUrlQuery(query url.Values) (*WeatherQuery, error) {
result := NewWeatherQuery()
result.Init()
start := query.Get("start")
end := query.Get("end")
if len(start) != 0 {
if tval, err := time.Parse(time.RFC3339, start); err == nil {
result.Start = tval
} else if err != nil {
fmt.Println(err)
return nil, err
}
}
if len(end) != 0 {
if tval, err := time.Parse(time.RFC3339, end); err == nil {
result.End = tval
} else if err != nil {
fmt.Println(err)
return nil, err
}
}
for _, sensorValueType := range GetSensorValueTypes() {
queryParam := query.Get(string(sensorValueType))
if bval, err := strconv.ParseBool(queryParam); err == nil {
result.Values[sensorValueType] = bval
}
}
return result, nil
}