forked from mirror/gorm
49 lines
834 B
Go
49 lines
834 B
Go
|
package clause
|
||
|
|
|
||
|
|
type For struct {
|
||
|
|
Lockings []Locking
|
||
|
|
}
|
||
|
|
|
||
|
|
type Locking struct {
|
||
|
|
Strength string
|
||
|
|
Table Table
|
||
|
|
Options string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Name where clause name
|
||
|
|
func (f For) Name() string {
|
||
|
|
return "FOR"
|
||
|
|
}
|
||
|
|
|
||
|
|
// Build build where clause
|
||
|
|
func (f For) Build(builder Builder) {
|
||
|
|
for idx, locking := range f.Lockings {
|
||
|
|
if idx > 0 {
|
||
|
|
builder.WriteByte(' ')
|
||
|
|
}
|
||
|
|
|
||
|
|
builder.Write("FOR ")
|
||
|
|
builder.Write(locking.Strength)
|
||
|
|
if locking.Table.Name != "" {
|
||
|
|
builder.Write(" OF ")
|
||
|
|
builder.WriteQuoted(locking.Table)
|
||
|
|
}
|
||
|
|
|
||
|
|
if locking.Options != "" {
|
||
|
|
builder.WriteByte(' ')
|
||
|
|
builder.Write(locking.Options)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// MergeClause merge order by clauses
|
||
|
|
func (f For) MergeClause(clause *Clause) {
|
||
|
|
clause.Name = ""
|
||
|
|
|
||
|
|
if v, ok := clause.Expression.(For); ok {
|
||
|
|
f.Lockings = append(v.Lockings, f.Lockings...)
|
||
|
|
}
|
||
|
|
|
||
|
|
clause.Expression = f
|
||
|
|
}
|