table_drop.go
1.2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package orm
type DropTableOptions struct {
IfExists bool
Cascade bool
}
func DropTable(db DB, model interface{}, opt *DropTableOptions) error {
return NewQuery(db, model).DropTable(opt)
}
type dropTableQuery struct {
q *Query
opt *DropTableOptions
}
var _ QueryAppender = (*dropTableQuery)(nil)
var _ queryCommand = (*dropTableQuery)(nil)
func newDropTableQuery(q *Query, opt *DropTableOptions) *dropTableQuery { //nolint:deadcode
return &dropTableQuery{
q: q,
opt: opt,
}
}
func (q *dropTableQuery) Clone() queryCommand {
return &dropTableQuery{
q: q.q.Clone(),
opt: q.opt,
}
}
func (q *dropTableQuery) Query() *Query {
return q.q
}
func (q *dropTableQuery) AppendTemplate(b []byte) ([]byte, error) {
return q.AppendQuery(dummyFormatter{}, b)
}
func (q *dropTableQuery) AppendQuery(fmter QueryFormatter, b []byte) (_ []byte, err error) {
if q.q.stickyErr != nil {
return nil, q.q.stickyErr
}
if q.q.model == nil {
return nil, errModelNil
}
b = append(b, "DROP TABLE "...)
if q.opt != nil && q.opt.IfExists {
b = append(b, "IF EXISTS "...)
}
b, err = q.q.appendFirstTable(fmter, b)
if err != nil {
return nil, err
}
if q.opt != nil && q.opt.Cascade {
b = append(b, " CASCADE"...)
}
return b, q.q.stickyErr
}