From 1f0f74693e59bfc5648889fa33ad7db7bd4cb9c5 Mon Sep 17 00:00:00 2001
From: Kichiyaki <dawidwysokinski000@gmail.com>
Date: Wed, 6 Jan 2021 14:12:50 +0100
Subject: [PATCH] [WIP] update ServerStatsFilter

---
 models/helpers.go      | 20 ++++++++++++++++++++
 models/server_stats.go | 32 ++++++++++++++++++++++++++------
 2 files changed, 46 insertions(+), 6 deletions(-)

diff --git a/models/helpers.go b/models/helpers.go
index 92ca444..1452836 100644
--- a/models/helpers.go
+++ b/models/helpers.go
@@ -17,6 +17,26 @@ func wrapStringInDoubleQuotes(str string) string {
 	return `"` + str + `"`
 }
 
+func buildConditionEquals(column string) string {
+	return column + "= ?"
+}
+
+func buildConditionLT(column string) string {
+	return column + "< ?"
+}
+
+func buildConditionLTE(column string) string {
+	return column + "<= ?"
+}
+
+func buildConditionGT(column string) string {
+	return column + "> ?"
+}
+
+func buildConditionGTE(column string) string {
+	return column + ">= ?"
+}
+
 func buildConditionMatch(column string) string {
 	return column + "LIKE ?"
 }
diff --git a/models/server_stats.go b/models/server_stats.go
index a34630a..1cac695 100644
--- a/models/server_stats.go
+++ b/models/server_stats.go
@@ -3,6 +3,8 @@ package models
 import (
 	"context"
 	"time"
+
+	"github.com/go-pg/pg/v10/orm"
 )
 
 type ServerStats struct {
@@ -30,15 +32,33 @@ func (s *ServerStats) BeforeInsert(ctx context.Context) (context.Context, error)
 }
 
 type ServerStatsFilter struct {
-	tableName struct{} `urlstruct:"stats"`
-
 	CreateDate    time.Time `json:"createDate" gqlgen:"createDate" xml:"createDate"`
 	CreateDateGT  time.Time `json:"createDateGT" gqlgen:"createDateGT" xml:"createDateGT"`
 	CreateDateGTE time.Time `json:"createDateGTE" gqlgen:"createDateGTE" xml:"createDateGTE"`
 	CreateDateLT  time.Time `json:"createDateLT" gqlgen:"createDateLT" xml:"createDateLT"`
 	CreateDateLTE time.Time `json:"createDateLTE" gqlgen:"createDateLTE" xml:"createDateLTE"`
-
-	Offset int    `urlstruct:",nowhere" json:"offset" gqlgen:"offset"`
-	Limit  int    `urlstruct:",nowhere" json:"limit" gqlgen:"limit"`
-	Sort   string `urlstruct:",nowhere" json:"sort" gqlgen:"sort"`
+}
+
+func (f *ServerStatsFilter) WhereWithAlias(q *orm.Query, alias string) (*orm.Query, error) {
+	if !f.CreateDate.IsZero() {
+		q = q.Where(buildConditionEquals(addAliasToColumnName("create_date", alias)), f.CreateDate)
+	}
+	if !f.CreateDateGT.IsZero() {
+		q = q.Where(buildConditionGT(addAliasToColumnName("create_date", alias)), f.CreateDateGT)
+	}
+	if !f.CreateDateGTE.IsZero() {
+		q = q.Where(buildConditionGTE(addAliasToColumnName("create_date", alias)), f.CreateDateGTE)
+	}
+	if !f.CreateDateLT.IsZero() {
+		q = q.Where(buildConditionLT(addAliasToColumnName("create_date", alias)), f.CreateDateLT)
+	}
+	if !f.CreateDateLTE.IsZero() {
+		q = q.Where(buildConditionLTE(addAliasToColumnName("create_date", alias)), f.CreateDateLTE)
+	}
+
+	return q, nil
+}
+
+func (f *ServerStatsFilter) Where(q *orm.Query) (*orm.Query, error) {
+	return f.WhereWithAlias(q, "stats")
 }