add the ability to set Field

stdyum · May 9, 2024 · ca00bfb · ca00bfb
1 parent 270e11e
commit ca00bfb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/databases/pagination/created_at_model.go b/databases/pagination/created_at_model.go
@@ -10,10 +10,19 @@ type CreatedAtPageQuery struct {
 	QPage    int    `json:"page"`
 	QPerPage int    `json:"perPage"`
 	QOrder   string `json:"order"`
+	QField   string `json:"field"`
 }
 
 func (c *CreatedAtPageQuery) Field() string {
-	return "created_at"
+	if c.QField == "" {
+		return "created_at"
+	}
+
+	return c.QField
+}
+
+func (c *CreatedAtPageQuery) SetField(field string) {
+	c.QField = field
 }
 
 func (c *CreatedAtPageQuery) Order() string {

diff --git a/databases/pagination/interfaces.go b/databases/pagination/interfaces.go
@@ -2,6 +2,8 @@ package pagination
 
 type Query interface {
 	Field() string
+	SetField(field string)
+
 	Order() string
 	Cursor() any
 	PerPage() int