feat: add is and not

sev-2 · Oct 16, 2024 · 280089d · 280089d
1 parent ad3d752
commit 280089d
Show file tree

Hide file tree

Showing 4 changed files with 279 additions and 6 deletions.
diff --git a/pkg/db/db.go b/pkg/db/db.go
@@ -17,6 +17,7 @@ type Query struct {
 	Relations    []string
 	WhereAndList *[]string
 	WhereOrList  *[]string
+	IsList       *[]string
 	OrderList    *[]string
 	LimitValue   int
 	OffsetValue  int
@@ -147,6 +148,11 @@ func buildQueryURI(q Query) string {
 		output += fmt.Sprintf("&or=(%s)", list)
 	}
 
+	if q.IsList != nil && len(*q.IsList) > 0 {
+		list := strings.Join(*q.IsList, ",")
+		output += fmt.Sprintf("&%s", list)
+	}
+
 	if q.OrderList != nil && len(*q.OrderList) > 0 {
 		orders := strings.Join(*q.OrderList, ",")
 		output += fmt.Sprintf("&order=%s", orders)

diff --git a/pkg/db/db_test.go b/pkg/db/db_test.go
@@ -16,12 +16,13 @@ var mockRaidenContext = raiden.Ctx{
 type ArticleMockModel struct {
 	ModelBase
 
-	Id        int64     `json:"id,omitempty" column:"name:id;type:bigint;primaryKey;autoIncrement;nullable:false"`
-	UserId    int64     `json:"user_id,omitempty" column:"name:user_id;type:bigint;nullable:false"`
-	Title     string    `json:"title,omitempty" column:"name:title;type:text;nullable:false"`
-	Body      string    `json:"body,omitempty" column:"name:body;type:text;nullable:true"`
-	Rating    int64     `json:"rating,omitempty" column:"name:rating;type:bigint;nullable:false"`
-	CreatedAt time.Time `json:"created_at,omitempty" column:"name:created_at;type:timestampz;nullable:false;default:now()"`
+	Id         int64     `json:"id,omitempty" column:"name:id;type:bigint;primaryKey;autoIncrement;nullable:false"`
+	UserId     int64     `json:"user_id,omitempty" column:"name:user_id;type:bigint;nullable:false"`
+	Title      string    `json:"title,omitempty" column:"name:title;type:text;nullable:false"`
+	Body       string    `json:"body,omitempty" column:"name:body;type:text;nullable:true"`
+	Rating     int64     `json:"rating,omitempty" column:"name:rating;type:bigint;nullable:false"`
+	IsFeatured bool      `json:"is_featured,omitempty" column:"name:is_featured;type:bool;nullable:false"`
+	CreatedAt  time.Time `json:"created_at,omitempty" column:"name:created_at;type:timestampz;nullable:false;default:now()"`
 
 	Metadata string `json:"-" schema:"public" tableName:"articles" rlsEnable:"true" rlsForced:"false"`
 

diff --git a/pkg/db/where.go b/pkg/db/where.go
@@ -21,6 +21,20 @@ func (q *Query) Eq(column string, value any) *Query {
 	return q
 }
 
+func (q *Query) NotEq(column string, value any) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.eq.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrEq(column string, value any) *Query {
 
 	if q.WhereOrList == nil {
@@ -49,6 +63,20 @@ func (q *Query) Neq(column string, value any) *Query {
 	return q
 }
 
+func (q *Query) NotNeq(column string, value any) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.neq.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrNeq(column string, value any) *Query {
 
 	if q.WhereOrList == nil {
@@ -77,6 +105,20 @@ func (q *Query) Lt(column string, value any) *Query {
 	return q
 }
 
+func (q *Query) NotLt(column string, value any) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.lt.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrLt(column string, value any) *Query {
 
 	if q.WhereOrList == nil {
@@ -105,6 +147,20 @@ func (q *Query) Lte(column string, value any) *Query {
 	return q
 }
 
+func (q *Query) NotLte(column string, value any) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.lte.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrLte(column string, value any) *Query {
 
 	if q.WhereOrList == nil {
@@ -133,6 +189,20 @@ func (q *Query) Gt(column string, value int) *Query {
 	return q
 }
 
+func (q *Query) NotGt(column string, value int) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.gt.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrGt(column string, value int) *Query {
 
 	if q.WhereOrList == nil {
@@ -161,6 +231,20 @@ func (q *Query) Gte(column string, value any) *Query {
 	return q
 }
 
+func (q *Query) NotGte(column string, value any) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.gte.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrGte(column string, value any) *Query {
 
 	if q.WhereOrList == nil {
@@ -191,6 +275,22 @@ func (q *Query) In(column string, value any) *Query {
 	return q
 }
 
+func (q *Query) NotIn(column string, value any) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	strValues := strings.Join(SliceToStringSlice(value), ",")
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.in.(%s)", column, strValues),
+	)
+
+	return q
+}
+
 func (q *Query) OrIn(column string, value any) *Query {
 
 	if q.WhereOrList == nil {
@@ -223,6 +323,22 @@ func (q *Query) Like(column string, value string) *Query {
 	return q
 }
 
+func (q *Query) NotLike(column string, value string) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	value = strings.ReplaceAll(value, "%", "*")
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.like.%s", column, getStringWithSpace(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrLike(column string, value string) *Query {
 
 	if q.WhereOrList == nil {
@@ -255,6 +371,22 @@ func (q *Query) Ilike(column string, value string) *Query {
 	return q
 }
 
+func (q *Query) NotIlike(column string, value string) *Query {
+
+	if q.WhereAndList == nil {
+		q.WhereAndList = &[]string{}
+	}
+
+	value = strings.ReplaceAll(value, "%", "*")
+
+	*q.WhereAndList = append(
+		*q.WhereAndList,
+		fmt.Sprintf("%s=not.ilike.%s", column, getStringWithSpace(value)),
+	)
+
+	return q
+}
+
 func (q *Query) OrIlike(column string, value string) *Query {
 
 	if q.WhereOrList == nil {
@@ -271,6 +403,38 @@ func (q *Query) OrIlike(column string, value string) *Query {
 	return q
 }
 
+func (q *Query) Is(column string, value any) *Query {
+
+	isValueWhitelist(value)
+
+	if q.IsList == nil {
+		q.IsList = &[]string{}
+	}
+
+	*q.IsList = append(
+		*q.IsList,
+		fmt.Sprintf("%s=is.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
+func (q *Query) NotIs(column string, value string) *Query {
+
+	isValueWhitelist(value)
+
+	if q.IsList == nil {
+		q.IsList = &[]string{}
+	}
+
+	*q.IsList = append(
+		*q.IsList,
+		fmt.Sprintf("%s=not.is.%s", column, getStringValue(value)),
+	)
+
+	return q
+}
+
 func getStringValue(value any) string {
 	return fmt.Sprintf("%v", value)
 }
@@ -304,3 +468,15 @@ func SliceToStringSlice(slice interface{}) []string {
 	}
 	return stringSlice
 }
+
+func isValueWhitelist(value any) bool {
+	switch getStringValue(value) {
+	case "true":
+	case "false":
+	case "null":
+	case "unknown":
+		return true
+	}
+
+	panic("isValueWhitelist: only \"true\", \"false\", \"null\", or \"unknown\" are allowed")
+}