forked from gitea/gitea
126 lines
4.5 KiB
Go
126 lines
4.5 KiB
Go
// Copyright 2013 The ql Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSES/QL-LICENSE file.
|
|
|
|
// Copyright 2015 PingCAP, 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,
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package table
|
|
|
|
import (
|
|
"github.com/juju/errors"
|
|
"github.com/pingcap/tidb/column"
|
|
"github.com/pingcap/tidb/context"
|
|
"github.com/pingcap/tidb/evaluator"
|
|
"github.com/pingcap/tidb/kv"
|
|
"github.com/pingcap/tidb/meta/autoid"
|
|
"github.com/pingcap/tidb/model"
|
|
"github.com/pingcap/tidb/mysql"
|
|
"github.com/pingcap/tidb/util/types"
|
|
)
|
|
|
|
// RecordIterFunc is used for low-level record iteration.
|
|
type RecordIterFunc func(h int64, rec []types.Datum, cols []*column.Col) (more bool, err error)
|
|
|
|
// Table is used to retrieve and modify rows in table.
|
|
type Table interface {
|
|
// IterRecords iterates records in the table and calls fn.
|
|
IterRecords(ctx context.Context, startKey kv.Key, cols []*column.Col, fn RecordIterFunc) error
|
|
|
|
// RowWithCols returns a row that contains the given cols.
|
|
RowWithCols(ctx context.Context, h int64, cols []*column.Col) ([]types.Datum, error)
|
|
|
|
// Row returns a row for all columns.
|
|
Row(ctx context.Context, h int64) ([]types.Datum, error)
|
|
|
|
// Cols returns the columns of the table which is used in select.
|
|
Cols() []*column.Col
|
|
|
|
// Indices returns the indices of the table.
|
|
Indices() []*column.IndexedCol
|
|
|
|
// RecordPrefix returns the record key prefix.
|
|
RecordPrefix() kv.Key
|
|
|
|
// IndexPrefix returns the index key prefix.
|
|
IndexPrefix() kv.Key
|
|
|
|
// FirstKey returns the first key.
|
|
FirstKey() kv.Key
|
|
|
|
// RecordKey returns the key in KV storage for the column.
|
|
RecordKey(h int64, col *column.Col) kv.Key
|
|
|
|
// Truncate truncates the table.
|
|
Truncate(ctx context.Context) (err error)
|
|
|
|
// AddRecord inserts a row into the table.
|
|
AddRecord(ctx context.Context, r []types.Datum) (recordID int64, err error)
|
|
|
|
// UpdateRecord updates a row in the table.
|
|
UpdateRecord(ctx context.Context, h int64, currData []types.Datum, newData []types.Datum, touched map[int]bool) error
|
|
|
|
// RemoveRecord removes a row in the table.
|
|
RemoveRecord(ctx context.Context, h int64, r []types.Datum) error
|
|
|
|
// AllocAutoID allocates an auto_increment ID for a new row.
|
|
AllocAutoID() (int64, error)
|
|
|
|
// RebaseAutoID rebases the auto_increment ID base.
|
|
// If allocIDs is true, it will allocate some IDs and save to the cache.
|
|
// If allocIDs is false, it will not allocate IDs.
|
|
RebaseAutoID(newBase int64, allocIDs bool) error
|
|
|
|
// Meta returns TableInfo.
|
|
Meta() *model.TableInfo
|
|
|
|
// LockRow locks a row.
|
|
LockRow(ctx context.Context, h int64, forRead bool) error
|
|
|
|
// Seek returns the handle greater or equal to h.
|
|
Seek(ctx context.Context, h int64) (handle int64, found bool, err error)
|
|
}
|
|
|
|
// TableFromMeta builds a table.Table from *model.TableInfo.
|
|
// Currently, it is assigned to tables.TableFromMeta in tidb package's init function.
|
|
var TableFromMeta func(alloc autoid.Allocator, tblInfo *model.TableInfo) (Table, error)
|
|
|
|
// GetColDefaultValue gets default value of the column.
|
|
func GetColDefaultValue(ctx context.Context, col *model.ColumnInfo) (types.Datum, bool, error) {
|
|
// Check no default value flag.
|
|
if mysql.HasNoDefaultValueFlag(col.Flag) && col.Tp != mysql.TypeEnum {
|
|
return types.Datum{}, false, errors.Errorf("Field '%s' doesn't have a default value", col.Name)
|
|
}
|
|
|
|
// Check and get timestamp/datetime default value.
|
|
if col.Tp == mysql.TypeTimestamp || col.Tp == mysql.TypeDatetime {
|
|
if col.DefaultValue == nil {
|
|
return types.Datum{}, true, nil
|
|
}
|
|
|
|
value, err := evaluator.GetTimeValue(ctx, col.DefaultValue, col.Tp, col.Decimal)
|
|
if err != nil {
|
|
return types.Datum{}, true, errors.Errorf("Field '%s' get default value fail - %s", col.Name, errors.Trace(err))
|
|
}
|
|
return types.NewDatum(value), true, nil
|
|
} else if col.Tp == mysql.TypeEnum {
|
|
// For enum type, if no default value and not null is set,
|
|
// the default value is the first element of the enum list
|
|
if col.DefaultValue == nil && mysql.HasNotNullFlag(col.Flag) {
|
|
return types.NewDatum(col.FieldType.Elems[0]), true, nil
|
|
}
|
|
}
|
|
|
|
return types.NewDatum(col.DefaultValue), true, nil
|
|
}
|