This commit is contained in:
2024-05-09 16:59:25 +08:00
parent 65b9f3795d
commit 90f807defc
12 changed files with 1283 additions and 0 deletions

49
clauses/merge.go Normal file
View File

@ -0,0 +1,49 @@
package clauses
import (
"gorm.io/gorm/clause"
)
type Merge struct {
Table clause.Table
Using []clause.Interface
On []clause.Expression
}
func (merge Merge) Name() string {
return "MERGE"
}
func MergeDefaultExcludeName() string {
return "exclude"
}
// Build build from clause
func (merge Merge) Build(builder clause.Builder) {
clause.Insert{}.Build(builder)
builder.WriteString(" USING (")
for idx, iface := range merge.Using {
if idx > 0 {
builder.WriteByte(' ')
}
builder.WriteString(iface.Name())
builder.WriteByte(' ')
iface.Build(builder)
}
builder.WriteString(") ")
builder.WriteString(MergeDefaultExcludeName())
builder.WriteString(" ON (")
for idx, on := range merge.On {
if idx > 0 {
builder.WriteString(", ")
}
on.Build(builder)
}
builder.WriteString(")")
}
// MergeClause merge values clauses
func (merge Merge) MergeClause(clause *clause.Clause) {
clause.Name = merge.Name()
clause.Expression = merge
}

10
clauses/returning_into.go Normal file
View File

@ -0,0 +1,10 @@
package clauses
import (
"gorm.io/gorm/clause"
)
type ReturningInto struct {
Variables []clause.Column
Into []*clause.Values
}

39
clauses/when_matched.go Normal file
View File

@ -0,0 +1,39 @@
package clauses
import (
"gorm.io/gorm/clause"
)
type WhenMatched struct {
clause.Set
Where, Delete clause.Where
}
func (w WhenMatched) Name() string {
return "WHEN MATCHED"
}
func (w WhenMatched) Build(builder clause.Builder) {
if len(w.Set) > 0 {
builder.WriteString(" THEN")
builder.WriteString(" UPDATE ")
builder.WriteString(w.Name())
builder.WriteByte(' ')
w.Build(builder)
buildWhere := func(where clause.Where) {
builder.WriteString(where.Name())
builder.WriteByte(' ')
where.Build(builder)
}
if len(w.Where.Exprs) > 0 {
buildWhere(w.Where)
}
if len(w.Delete.Exprs) > 0 {
builder.WriteString(" DELETE ")
buildWhere(w.Delete)
}
}
}

View File

@ -0,0 +1,32 @@
package clauses
import (
"gorm.io/gorm/clause"
)
type WhenNotMatched struct {
clause.Values
Where clause.Where
}
func (w WhenNotMatched) Name() string {
return "WHEN NOT MATCHED"
}
func (w WhenNotMatched) Build(builder clause.Builder) {
if len(w.Columns) > 0 {
if len(w.Values.Values) != 1 {
panic("cannot insert more than one rows due to Oracle SQL language restriction")
}
builder.WriteString(" THEN")
builder.WriteString(" INSERT ")
w.Build(builder)
if len(w.Where.Exprs) > 0 {
builder.WriteString(w.Where.Name())
builder.WriteByte(' ')
w.Where.Build(builder)
}
}
}