-
Notifications
You must be signed in to change notification settings - Fork 45
/
query_delete.go
112 lines (101 loc) · 3.29 KB
/
query_delete.go
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
* Copyright 2022 CECTC, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package exec
import (
"context"
"fmt"
"strings"
"github.com/cectc/dbpack/pkg/constant"
"github.com/cectc/dbpack/pkg/driver"
"github.com/cectc/dbpack/pkg/dt/schema"
"github.com/cectc/dbpack/pkg/log"
"github.com/cectc/dbpack/pkg/meta"
"github.com/cectc/dbpack/pkg/misc"
"github.com/cectc/dbpack/pkg/resource"
"github.com/cectc/dbpack/pkg/tracing"
"github.com/cectc/dbpack/third_party/parser/ast"
"github.com/cectc/dbpack/third_party/parser/format"
)
type queryDeleteExecutor struct {
appid string
conn *driver.BackendConnection
stmt *ast.DeleteStmt
}
func NewQueryDeleteExecutor(
appid string,
conn *driver.BackendConnection,
stmt *ast.DeleteStmt) Executor {
return &queryDeleteExecutor{
appid: appid,
conn: conn,
stmt: stmt,
}
}
func (executor *queryDeleteExecutor) BeforeImage(ctx context.Context) (*schema.TableRecords, error) {
spanCtx, span := tracing.GetTraceSpan(ctx, tracing.ExecutorFetchBeforeImage)
defer span.End()
tableMeta, err := executor.GetTableMeta(spanCtx)
if err != nil {
tracing.RecordErrorSpan(span, err)
return nil, err
}
sql := executor.buildBeforeImageSql(tableMeta)
result, _, err := executor.conn.ExecuteWithWarningCount(spanCtx, sql, true)
if err != nil {
tracing.RecordErrorSpan(span, err)
return nil, err
}
return schema.BuildTableRecords(tableMeta, result), nil
}
func (executor *queryDeleteExecutor) AfterImage(ctx context.Context) (*schema.TableRecords, error) {
return nil, nil
}
func (executor *queryDeleteExecutor) GetTableMeta(ctx context.Context) (schema.TableMeta, error) {
dbName := executor.conn.DataSourceName()
db := resource.GetDBManager(executor.appid).GetDB(dbName)
return meta.GetTableMetaCache().GetTableMeta(ctx, db, executor.GetTableName())
}
func (executor *queryDeleteExecutor) GetTableName() string {
var sb strings.Builder
if err := executor.stmt.TableRefs.TableRefs.Restore(format.NewRestoreCtx(constant.DBPackRestoreFormat, &sb)); err != nil {
log.Panic(err)
}
return sb.String()
}
func (executor *queryDeleteExecutor) buildBeforeImageSql(tableMeta schema.TableMeta) string {
var b strings.Builder
b.WriteString("SELECT ")
columnCount := len(tableMeta.Columns)
for i, column := range tableMeta.Columns {
b.WriteString(misc.CheckAndReplace(column))
if i < columnCount-1 {
b.WriteByte(',')
} else {
b.WriteByte(' ')
}
}
b.WriteString(fmt.Sprintf("FROM %s WHERE ", executor.GetTableName()))
b.WriteString(executor.GetWhereCondition())
b.WriteString(" FOR UPDATE")
return b.String()
}
func (executor *queryDeleteExecutor) GetWhereCondition() string {
var sb strings.Builder
if err := executor.stmt.Where.Restore(format.NewRestoreCtx(constant.DBPackRestoreFormat, &sb)); err != nil {
log.Panic(err)
}
return sb.String()
}