forked from gitea/gitea
1
0
Fork 0
gitea/vendor/xorm.io/core/column.go

167 lines
3.8 KiB
Go
Raw Normal View History

// 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.
2016-11-04 06:16:01 +08:00
package core
import (
"fmt"
"reflect"
"strings"
"time"
)
const (
TWOSIDES = iota + 1
ONLYTODB
ONLYFROMDB
)
// Column defines database column
2016-11-04 06:16:01 +08:00
type Column struct {
Name string
TableName string
FieldName string
SQLType SQLType
IsJSON bool
2016-11-04 06:16:01 +08:00
Length int
Length2 int
Nullable bool
Default string
Indexes map[string]int
IsPrimaryKey bool
IsAutoIncrement bool
MapType int
IsCreated bool
IsUpdated bool
IsDeleted bool
IsCascade bool
IsVersion bool
DefaultIsEmpty bool
EnumOptions map[string]int
SetOptions map[string]int
DisableTimeZone bool
TimeZone *time.Location // column specified time zone
Comment string
2016-11-04 06:16:01 +08:00
}
// NewColumn creates a new column
2016-11-04 06:16:01 +08:00
func NewColumn(name, fieldName string, sqlType SQLType, len1, len2 int, nullable bool) *Column {
return &Column{
Name: name,
TableName: "",
FieldName: fieldName,
SQLType: sqlType,
Length: len1,
Length2: len2,
Nullable: nullable,
Default: "",
Indexes: make(map[string]int),
IsPrimaryKey: false,
IsAutoIncrement: false,
MapType: TWOSIDES,
IsCreated: false,
IsUpdated: false,
IsDeleted: false,
IsCascade: false,
IsVersion: false,
DefaultIsEmpty: false,
EnumOptions: make(map[string]int),
Comment: "",
2016-11-04 06:16:01 +08:00
}
}
// String generate column description string according dialect
2016-11-04 06:16:01 +08:00
func (col *Column) String(d Dialect) string {
sql := d.Quote(col.Name) + " "
2016-11-04 06:16:01 +08:00
sql += d.SqlType(col) + " "
if col.IsPrimaryKey {
sql += "PRIMARY KEY "
if col.IsAutoIncrement {
sql += d.AutoIncrStr() + " "
}
}
2018-01-27 23:20:59 +08:00
if col.Default != "" {
sql += "DEFAULT " + col.Default + " "
}
2016-11-04 06:16:01 +08:00
if d.ShowCreateNull() {
if col.Nullable {
sql += "NULL "
} else {
sql += "NOT NULL "
}
}
return sql
}
// StringNoPk generate column description string according dialect without primary keys
2016-11-04 06:16:01 +08:00
func (col *Column) StringNoPk(d Dialect) string {
sql := d.Quote(col.Name) + " "
2016-11-04 06:16:01 +08:00
sql += d.SqlType(col) + " "
2018-01-27 23:20:59 +08:00
if col.Default != "" {
sql += "DEFAULT " + col.Default + " "
}
2016-11-04 06:16:01 +08:00
if d.ShowCreateNull() {
if col.Nullable {
sql += "NULL "
} else {
sql += "NOT NULL "
}
}
return sql
}
// ValueOf returns column's filed of struct's value
2016-11-04 06:16:01 +08:00
func (col *Column) ValueOf(bean interface{}) (*reflect.Value, error) {
dataStruct := reflect.Indirect(reflect.ValueOf(bean))
return col.ValueOfV(&dataStruct)
}
// ValueOfV returns column's filed of struct's value accept reflevt value
2016-11-04 06:16:01 +08:00
func (col *Column) ValueOfV(dataStruct *reflect.Value) (*reflect.Value, error) {
var fieldValue reflect.Value
fieldPath := strings.Split(col.FieldName, ".")
2016-11-04 06:16:01 +08:00
if dataStruct.Type().Kind() == reflect.Map {
keyValue := reflect.ValueOf(fieldPath[len(fieldPath)-1])
2016-11-04 06:16:01 +08:00
fieldValue = dataStruct.MapIndex(keyValue)
return &fieldValue, nil
} else if dataStruct.Type().Kind() == reflect.Interface {
structValue := reflect.ValueOf(dataStruct.Interface())
dataStruct = &structValue
}
level := len(fieldPath)
fieldValue = dataStruct.FieldByName(fieldPath[0])
2016-11-04 06:16:01 +08:00
for i := 0; i < level-1; i++ {
if !fieldValue.IsValid() {
break
}
if fieldValue.Kind() == reflect.Struct {
fieldValue = fieldValue.FieldByName(fieldPath[i+1])
2016-11-04 06:16:01 +08:00
} else if fieldValue.Kind() == reflect.Ptr {
if fieldValue.IsNil() {
fieldValue.Set(reflect.New(fieldValue.Type().Elem()))
}
fieldValue = fieldValue.Elem().FieldByName(fieldPath[i+1])
2016-11-04 06:16:01 +08:00
} else {
return nil, fmt.Errorf("field %v is not valid", col.FieldName)
2016-11-04 06:16:01 +08:00
}
}
if !fieldValue.IsValid() {
return nil, fmt.Errorf("field %v is not valid", col.FieldName)
2016-11-04 06:16:01 +08:00
}
return &fieldValue, nil
}