107 lines
1.8 KiB
Go
107 lines
1.8 KiB
Go
package model
|
|
|
|
type QueryCondition struct {
|
|
Where ParseWhere
|
|
From string
|
|
Fields string
|
|
Group string
|
|
Order SqlBuilder
|
|
Join SqlBuilder
|
|
Having SqlBuilder
|
|
Limit int
|
|
Offset int
|
|
In [][]any
|
|
Relation map[string]*QueryCondition
|
|
}
|
|
|
|
func Conditions(fns ...Condition) QueryCondition {
|
|
r := QueryCondition{}
|
|
for _, fn := range fns {
|
|
fn(&r)
|
|
}
|
|
if r.Fields == "" {
|
|
r.Fields = "*"
|
|
}
|
|
return r
|
|
}
|
|
func WithConditions(fns ...Condition) *QueryCondition {
|
|
r := QueryCondition{}
|
|
for _, fn := range fns {
|
|
fn(&r)
|
|
}
|
|
if r.Fields == "" {
|
|
r.Fields = "*"
|
|
}
|
|
return &r
|
|
}
|
|
|
|
type Condition func(c *QueryCondition)
|
|
|
|
func Where(where ParseWhere) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Where = where
|
|
}
|
|
}
|
|
func Fields(fields string) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Fields = fields
|
|
}
|
|
}
|
|
|
|
func From(from string) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.From = from
|
|
}
|
|
}
|
|
|
|
func Group(group string) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Group = group
|
|
}
|
|
}
|
|
|
|
func Order[T ~[][]string](order T) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Order = SqlBuilder(order)
|
|
}
|
|
}
|
|
|
|
func Join[T ~[][]string](join T) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Join = SqlBuilder(join)
|
|
}
|
|
}
|
|
|
|
func Having[T ~[][]string](having T) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Having = SqlBuilder(having)
|
|
}
|
|
}
|
|
|
|
func Limit(limit int) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Limit = limit
|
|
}
|
|
}
|
|
|
|
func Offset(offset int) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.Offset = offset
|
|
}
|
|
}
|
|
|
|
func In(in ...[]any) Condition {
|
|
return func(c *QueryCondition) {
|
|
c.In = append(c.In, in...)
|
|
}
|
|
}
|
|
|
|
func With(tableTag string, q *QueryCondition) Condition {
|
|
return func(c *QueryCondition) {
|
|
if c.Relation == nil {
|
|
c.Relation = map[string]*QueryCondition{}
|
|
}
|
|
c.Relation[tableTag] = q
|
|
}
|
|
}
|