132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package qm
|
|
|
|
import "github.com/nullbio/sqlboiler/boil"
|
|
|
|
// QueryMod to modify the query object
|
|
type QueryMod func(q *boil.Query)
|
|
|
|
// Apply the query mods to the Query object
|
|
func Apply(q *boil.Query, mods ...QueryMod) {
|
|
for _, mod := range mods {
|
|
mod(q)
|
|
}
|
|
}
|
|
|
|
// SQL allows you to execute a plain SQL statement
|
|
func SQL(sql string, args ...interface{}) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetSQL(q, sql, args...)
|
|
}
|
|
}
|
|
|
|
// Or surrounds where clauses to join them with OR as opposed to AND
|
|
func Or(whereMods ...QueryMod) QueryMod {
|
|
return func(q *boil.Query) {
|
|
if len(whereMods) < 2 {
|
|
panic("Or requires at least two arguments")
|
|
}
|
|
|
|
for _, w := range whereMods {
|
|
w(q)
|
|
boil.SetLastWhereAsOr(q)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Limit the number of returned rows
|
|
func Limit(limit int) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetLimit(q, limit)
|
|
}
|
|
}
|
|
|
|
// Offset into the results
|
|
func Offset(offset int) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetOffset(q, offset)
|
|
}
|
|
}
|
|
|
|
// InnerJoin on another table
|
|
func InnerJoin(clause string, args ...interface{}) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.AppendInnerJoin(q, boil.JoinInner, clause, args...)
|
|
}
|
|
}
|
|
|
|
// Select specific columns opposed to all columns
|
|
func Select(columns ...string) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.AppendSelect(q, columns...)
|
|
}
|
|
}
|
|
|
|
// Where allows you to specify a where clause for your statement
|
|
func Where(clause string, args ...interface{}) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.AppendWhere(q, clause, args...)
|
|
}
|
|
}
|
|
|
|
// GroupBy allows you to specify a group by clause for your statement
|
|
func GroupBy(clause string) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.ApplyGroupBy(q, clause)
|
|
}
|
|
}
|
|
|
|
// OrderBy allows you to specify a order by clause for your statement
|
|
func OrderBy(clause string) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.ApplyOrderBy(q, clause)
|
|
}
|
|
}
|
|
|
|
// Having allows you to specify a having clause for your statement
|
|
func Having(clause string) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.ApplyHaving(q, clause)
|
|
}
|
|
}
|
|
|
|
// From allows to specify the table for your statement
|
|
func From(from string) QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.AppendFrom(q, from)
|
|
}
|
|
}
|
|
|
|
// Count turns the query into a counting calculation
|
|
func Count() QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetCount(q)
|
|
}
|
|
}
|
|
|
|
// Avg turns the query into an average calculation
|
|
func Avg() QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetAvg(q)
|
|
}
|
|
}
|
|
|
|
// Min turns the query into a minimum calculation
|
|
func Min() QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetMin(q)
|
|
}
|
|
}
|
|
|
|
// Max turns the query into a maximum calculation
|
|
func Max() QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetMax(q)
|
|
}
|
|
}
|
|
|
|
// Sum turns the query into a sum calculation
|
|
func Sum() QueryMod {
|
|
return func(q *boil.Query) {
|
|
boil.SetSum(q)
|
|
}
|
|
}
|