tidb ir 源码
tidb ir 代码
文件路径：/dumpling/export/ir.go
// Copyright 2020 PingCAP, Inc. Licensed under Apache-2.0.
package export
import (
"bytes"
"database/sql"
"strings"
"github.com/pingcap/errors"
tcontext "github.com/pingcap/tidb/dumpling/context"
)
// TableDataIR is table data intermediate representation.
// A table may be split into multiple TableDataIRs.
type TableDataIR interface {
Start(*tcontext.Context, *sql.Conn) error
Rows() SQLRowIter
Close() error
RawRows() *sql.Rows
}
// TableMeta contains the meta information of a table.
type TableMeta interface {
DatabaseName() string
TableName() string
ColumnCount() uint
ColumnTypes() []string
ColumnNames() []string
SelectedField() string
SelectedLen() int
SpecialComments() StringIter
ShowCreateTable() string
ShowCreateView() string
AvgRowLength() uint64
HasImplicitRowID() bool
}
// SQLRowIter is the iterator on a collection of sql.Row.
type SQLRowIter interface {
Decode(RowReceiver) error
Next()
Error() error
HasNext() bool
// release SQLRowIter
Close() error
}
// RowReceiverStringer is a combined interface of RowReceiver and Stringer
type RowReceiverStringer interface {
RowReceiver
Stringer
}
// Stringer is an interface which represents sql types that support writing to buffer in sql/csv type
type Stringer interface {
WriteToBuffer(*bytes.Buffer, bool)
WriteToBufferInCsv(*bytes.Buffer, bool, *csvOption)
}
// RowReceiver is an interface which represents sql types that support bind address for *sql.Rows
type RowReceiver interface {
BindAddress([]interface{})
}
func decodeFromRows(rows *sql.Rows, args []interface{}, row RowReceiver) error {
row.BindAddress(args)
if err := rows.Scan(args...); err != nil {
rows.Close()
return errors.Trace(err)
}
return nil
}
// StringIter is the iterator on a collection of strings.
type StringIter interface {
Next() string
HasNext() bool
}
// MetaIR is the interface that wraps database/table/view's metadata
type MetaIR interface {
SpecialComments() StringIter
TargetName() string
MetaSQL() string
}
func setTableMetaFromRows(rows *sql.Rows) (TableMeta, error) {
tps, err := rows.ColumnTypes()
if err != nil {
return nil, errors.Trace(err)
}
nms, err := rows.Columns()
if err != nil {
return nil, errors.Trace(err)
}
for i := range nms {
nms[i] = wrapBackTicks(nms[i])
}
return &tableMeta{
colTypes: tps,
selectedField: strings.Join(nms, ","),
selectedLen: len(nms),
specCmts: []string{"/*!40101 SET NAMES binary*/;"},
}, nil
}
相关信息
相关文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦