mirror of
https://github.com/go-gitea/gitea
synced 2024-12-23 00:07:52 +01:00
cdef92b3ff
Co-authored-by: techknowlogick <techknowlogick@gitea.io>
196 lines
4.6 KiB
Go
196 lines
4.6 KiB
Go
// Copyright 2019 The Xorm Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package schemas
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// Table represents a database table
|
|
type Table struct {
|
|
Name string
|
|
Type reflect.Type
|
|
columnsSeq []string
|
|
columnsMap map[string][]*Column
|
|
columns []*Column
|
|
Indexes map[string]*Index
|
|
PrimaryKeys []string
|
|
AutoIncrement string
|
|
Created map[string]bool
|
|
Updated string
|
|
Deleted string
|
|
Version string
|
|
StoreEngine string
|
|
Charset string
|
|
Comment string
|
|
}
|
|
|
|
// NewEmptyTable creates an empty table
|
|
func NewEmptyTable() *Table {
|
|
return NewTable("", nil)
|
|
}
|
|
|
|
// NewTable creates a new Table object
|
|
func NewTable(name string, t reflect.Type) *Table {
|
|
return &Table{Name: name, Type: t,
|
|
columnsSeq: make([]string, 0),
|
|
columns: make([]*Column, 0),
|
|
columnsMap: make(map[string][]*Column),
|
|
Indexes: make(map[string]*Index),
|
|
Created: make(map[string]bool),
|
|
PrimaryKeys: make([]string, 0),
|
|
}
|
|
}
|
|
|
|
// Columns returns table's columns
|
|
func (table *Table) Columns() []*Column {
|
|
return table.columns
|
|
}
|
|
|
|
// ColumnsSeq returns table's column names according sequence
|
|
func (table *Table) ColumnsSeq() []string {
|
|
return table.columnsSeq
|
|
}
|
|
|
|
func (table *Table) columnsByName(name string) []*Column {
|
|
for k, cols := range table.columnsMap {
|
|
if strings.EqualFold(k, name) {
|
|
return cols
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetColumn returns column according column name, if column not found, return nil
|
|
func (table *Table) GetColumn(name string) *Column {
|
|
cols := table.columnsByName(name)
|
|
if cols != nil {
|
|
return cols[0]
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetColumnIdx returns column according name and idx
|
|
func (table *Table) GetColumnIdx(name string, idx int) *Column {
|
|
cols := table.columnsByName(name)
|
|
if cols != nil && idx < len(cols) {
|
|
return cols[idx]
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// PKColumns reprents all primary key columns
|
|
func (table *Table) PKColumns() []*Column {
|
|
columns := make([]*Column, len(table.PrimaryKeys))
|
|
for i, name := range table.PrimaryKeys {
|
|
columns[i] = table.GetColumn(name)
|
|
}
|
|
return columns
|
|
}
|
|
|
|
func (table *Table) ColumnType(name string) reflect.Type {
|
|
t, _ := table.Type.FieldByName(name)
|
|
return t.Type
|
|
}
|
|
|
|
func (table *Table) AutoIncrColumn() *Column {
|
|
return table.GetColumn(table.AutoIncrement)
|
|
}
|
|
|
|
func (table *Table) VersionColumn() *Column {
|
|
return table.GetColumn(table.Version)
|
|
}
|
|
|
|
func (table *Table) UpdatedColumn() *Column {
|
|
return table.GetColumn(table.Updated)
|
|
}
|
|
|
|
func (table *Table) DeletedColumn() *Column {
|
|
return table.GetColumn(table.Deleted)
|
|
}
|
|
|
|
// AddColumn adds a column to table
|
|
func (table *Table) AddColumn(col *Column) {
|
|
table.columnsSeq = append(table.columnsSeq, col.Name)
|
|
table.columns = append(table.columns, col)
|
|
colName := strings.ToLower(col.Name)
|
|
if c, ok := table.columnsMap[colName]; ok {
|
|
table.columnsMap[colName] = append(c, col)
|
|
} else {
|
|
table.columnsMap[colName] = []*Column{col}
|
|
}
|
|
|
|
if col.IsPrimaryKey {
|
|
table.PrimaryKeys = append(table.PrimaryKeys, col.Name)
|
|
}
|
|
if col.IsAutoIncrement {
|
|
table.AutoIncrement = col.Name
|
|
}
|
|
if col.IsCreated {
|
|
table.Created[col.Name] = true
|
|
}
|
|
if col.IsUpdated {
|
|
table.Updated = col.Name
|
|
}
|
|
if col.IsDeleted {
|
|
table.Deleted = col.Name
|
|
}
|
|
if col.IsVersion {
|
|
table.Version = col.Name
|
|
}
|
|
}
|
|
|
|
// AddIndex adds an index or an unique to table
|
|
func (table *Table) AddIndex(index *Index) {
|
|
table.Indexes[index.Name] = index
|
|
}
|
|
|
|
// IDOfV get id from one value of struct
|
|
func (table *Table) IDOfV(rv reflect.Value) (PK, error) {
|
|
v := reflect.Indirect(rv)
|
|
pk := make([]interface{}, len(table.PrimaryKeys))
|
|
for i, col := range table.PKColumns() {
|
|
var err error
|
|
|
|
fieldName := col.FieldName
|
|
for {
|
|
parts := strings.SplitN(fieldName, ".", 2)
|
|
if len(parts) == 1 {
|
|
break
|
|
}
|
|
|
|
v = v.FieldByName(parts[0])
|
|
if v.Kind() == reflect.Ptr {
|
|
v = v.Elem()
|
|
}
|
|
if v.Kind() != reflect.Struct {
|
|
return nil, fmt.Errorf("Unsupported read value of column %s from field %s", col.Name, col.FieldName)
|
|
}
|
|
fieldName = parts[1]
|
|
}
|
|
|
|
pkField := v.FieldByName(fieldName)
|
|
switch pkField.Kind() {
|
|
case reflect.String:
|
|
pk[i], err = col.ConvertID(pkField.String())
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
pk[i], err = col.ConvertID(strconv.FormatInt(pkField.Int(), 10))
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
// id of uint will be converted to int64
|
|
pk[i], err = col.ConvertID(strconv.FormatUint(pkField.Uint(), 10))
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return PK(pk), nil
|
|
}
|