gorm/clause/clause.go

89 lines
1.7 KiB
Go
Raw Normal View History

2020-01-29 14:22:44 +03:00
package clause
2020-02-07 18:45:35 +03:00
// Interface clause interface
type Interface interface {
Name() string
Build(Builder)
MergeClause(*Clause)
}
// ClauseBuilder clause builder, allows to custmize how to build clause
2020-05-29 17:34:35 +03:00
type ClauseBuilder func(Clause, Builder)
2020-02-07 18:45:35 +03:00
2020-03-09 12:07:00 +03:00
type Writer interface {
WriteByte(byte) error
WriteString(string) (int, error)
}
2020-02-07 18:45:35 +03:00
// Builder builder interface
type Builder interface {
2020-03-09 12:07:00 +03:00
Writer
2020-07-16 06:27:04 +03:00
WriteQuoted(field interface{})
2020-03-09 12:07:00 +03:00
AddVar(Writer, ...interface{})
2020-02-07 18:45:35 +03:00
}
2020-01-30 10:14:48 +03:00
// Clause
type Clause struct {
2020-06-06 17:52:08 +03:00
Name string // WHERE
BeforeExpression Expression
AfterNameExpression Expression
AfterExpression Expression
Expression Expression
Builder ClauseBuilder
2020-01-30 10:14:48 +03:00
}
// Build build clause
func (c Clause) Build(builder Builder) {
if c.Builder != nil {
2020-05-29 17:34:35 +03:00
c.Builder(c, builder)
2020-06-06 17:52:08 +03:00
} else if c.Expression != nil {
if c.BeforeExpression != nil {
c.BeforeExpression.Build(builder)
builder.WriteByte(' ')
}
2020-01-30 10:14:48 +03:00
if c.Name != "" {
2020-06-06 17:52:08 +03:00
builder.WriteString(c.Name)
builder.WriteByte(' ')
2020-01-30 10:14:48 +03:00
}
2020-06-06 17:52:08 +03:00
if c.AfterNameExpression != nil {
2020-06-14 06:46:17 +03:00
c.AfterNameExpression.Build(builder)
2020-06-06 17:52:08 +03:00
builder.WriteByte(' ')
2020-01-30 10:14:48 +03:00
}
2020-06-06 17:52:08 +03:00
c.Expression.Build(builder)
if c.AfterExpression != nil {
builder.WriteByte(' ')
c.AfterExpression.Build(builder)
2020-01-30 10:14:48 +03:00
}
}
2020-01-29 14:22:44 +03:00
}
2020-02-07 18:45:35 +03:00
const (
2020-06-06 17:52:08 +03:00
PrimaryKey string = "@@@py@@@" // primary key
CurrentTable string = "@@@ct@@@" // current table
Associations string = "@@@as@@@" // associations
2020-02-07 18:45:35 +03:00
)
2020-01-29 22:03:06 +03:00
2020-02-07 18:45:35 +03:00
var (
currentTable = Table{Name: CurrentTable}
PrimaryColumn = Column{Table: CurrentTable, Name: PrimaryKey}
)
2020-01-29 14:22:44 +03:00
2020-02-07 18:45:35 +03:00
// Column quote with name
type Column struct {
Table string
Name string
Alias string
Raw bool
2020-01-29 14:22:44 +03:00
}
2020-02-07 18:45:35 +03:00
// Table quote with name
type Table struct {
Name string
Alias string
Raw bool
2020-01-29 14:22:44 +03:00
}