support different text type with JSON type

This commit is contained in:
Lunny Xiao 2017-07-20 20:07:54 +08:00
parent 14a0c19a0c
commit b42320686f
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
13 changed files with 73 additions and 16 deletions

View File

@ -244,7 +244,7 @@ func (db *mssql) SqlType(c *core.Column) string {
c.Length = 7 c.Length = 7
case core.MediumInt: case core.MediumInt:
res = core.Int res = core.Int
case core.Text, core.MediumText, core.TinyText, core.LongText, core.Json: case core.Text, core.MediumText, core.TinyText, core.LongText:
res = core.Varchar + "(MAX)" res = core.Varchar + "(MAX)"
case core.Double: case core.Double:
res = core.Real res = core.Real

View File

@ -243,8 +243,6 @@ func (db *mysql) SqlType(c *core.Column) string {
case core.Uuid: case core.Uuid:
res = core.Varchar res = core.Varchar
c.Length = 40 c.Length = 40
case core.Json:
res = core.Text
default: default:
res = t res = t
} }

View File

@ -519,7 +519,7 @@ func (db *oracle) SqlType(c *core.Column) string {
res = "TIMESTAMP WITH TIME ZONE" res = "TIMESTAMP WITH TIME ZONE"
case core.Float, core.Double, core.Numeric, core.Decimal: case core.Float, core.Double, core.Numeric, core.Decimal:
res = "NUMBER" res = "NUMBER"
case core.Text, core.MediumText, core.LongText, core.Json: case core.Text, core.MediumText, core.LongText:
res = "CLOB" res = "CLOB"
case core.Char, core.Varchar, core.TinyText: case core.Char, core.Varchar, core.TinyText:
res = "VARCHAR2" res = "VARCHAR2"

View File

@ -165,7 +165,7 @@ func (db *sqlite3) SqlType(c *core.Column) string {
case core.TimeStampz: case core.TimeStampz:
return core.Text return core.Text
case core.Char, core.Varchar, core.NVarchar, core.TinyText, case core.Char, core.Varchar, core.NVarchar, core.TinyText,
core.Text, core.MediumText, core.LongText, core.Json: core.Text, core.MediumText, core.LongText:
return core.Text return core.Text
case core.Bit, core.TinyInt, core.SmallInt, core.MediumInt, core.Int, core.Integer, core.BigInt: case core.Bit, core.TinyInt, core.SmallInt, core.MediumInt, core.Int, core.Integer, core.BigInt:
return core.Integer return core.Integer

View File

@ -1076,7 +1076,6 @@ func (engine *Engine) mapType(v reflect.Value) (*core.Table, error) {
} }
table.AddColumn(col) table.AddColumn(col)
} // end for } // end for
if idFieldColName != "" && len(table.PrimaryKeys) == 0 { if idFieldColName != "" && len(table.PrimaryKeys) == 0 {

View File

@ -34,7 +34,7 @@ func (engine *Engine) buildConds(table *core.Table, bean interface{},
if engine.dialect.DBType() == core.MSSQL && (col.SQLType.Name == core.Text || col.SQLType.IsBlob() || col.SQLType.Name == core.TimeStampz) { if engine.dialect.DBType() == core.MSSQL && (col.SQLType.Name == core.Text || col.SQLType.IsBlob() || col.SQLType.Name == core.TimeStampz) {
continue continue
} }
if col.SQLType.IsJson() { if col.IsJSON || col.SQLType.IsJson() {
continue continue
} }
@ -144,7 +144,7 @@ func (engine *Engine) buildConds(table *core.Table, bean interface{},
continue continue
} }
} else { } else {
if col.SQLType.IsJson() { if col.IsJSON || col.SQLType.IsJson() {
if col.SQLType.IsText() { if col.SQLType.IsText() {
bytes, err := DefaultJSONHandler.Marshal(fieldValue.Interface()) bytes, err := DefaultJSONHandler.Marshal(fieldValue.Interface())
if err != nil { if err != nil {

View File

@ -484,7 +484,7 @@ func (session *Session) slice2Bean(scanResults []interface{}, fields []string, b
fieldType := fieldValue.Type() fieldType := fieldValue.Type()
hasAssigned := false hasAssigned := false
if col.SQLType.IsJson() { if col.IsJSON || col.SQLType.IsJson() {
var bs []byte var bs []byte
if rawValueType.Kind() == reflect.String { if rawValueType.Kind() == reflect.String {
bs = []byte(vv.String()) bs = []byte(vv.String())
@ -664,7 +664,7 @@ func (session *Session) slice2Bean(scanResults []interface{}, fields []string, b
session.engine.logger.Error("sql.Sanner error:", err.Error()) session.engine.logger.Error("sql.Sanner error:", err.Error())
hasAssigned = false hasAssigned = false
} }
} else if col.SQLType.IsJson() { } else if col.IsJSON || col.SQLType.IsJson() {
if rawValueType.Kind() == reflect.String { if rawValueType.Kind() == reflect.String {
hasAssigned = true hasAssigned = true
x := reflect.New(fieldType) x := reflect.New(fieldType)

View File

@ -595,7 +595,7 @@ func (session *Session) value2Interface(col *core.Column, fieldValue reflect.Val
return t.Float64, nil return t.Float64, nil
} }
if !col.SQLType.IsJson() { if !(col.IsJSON || col.SQLType.IsJson()) {
// !<winxxp>! 增加支持driver.Valuer接口的结构如sql.NullString // !<winxxp>! 增加支持driver.Valuer接口的结构如sql.NullString
if v, ok := fieldValue.Interface().(driver.Valuer); ok { if v, ok := fieldValue.Interface().(driver.Valuer); ok {
return v.Value() return v.Value()

View File

@ -8,8 +8,8 @@ import (
"testing" "testing"
"time" "time"
"xorm.io/core"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"xorm.io/core"
) )
func TestDelete(t *testing.T) { func TestDelete(t *testing.T) {
@ -29,7 +29,7 @@ func TestDelete(t *testing.T) {
if testEngine.Dialect().DBType() == core.MSSQL { if testEngine.Dialect().DBType() == core.MSSQL {
err = session.Begin() err = session.Begin()
assert.NoError(t, err) assert.NoError(t, err)
_, err = session.Exec("SET IDENTITY_INSERT userinfo_delete ON") _, err = session.Exec("SET IDENTITY_INSERT " + testEngine.TableMapper.Obj2Table("UserinfoDelete") + " ON")
assert.NoError(t, err) assert.NoError(t, err)
} }

View File

@ -397,7 +397,7 @@ func (statement *Statement) buildUpdates(bean interface{},
} else if nulType, ok := fieldValue.Interface().(driver.Valuer); ok { } else if nulType, ok := fieldValue.Interface().(driver.Valuer); ok {
val, _ = nulType.Value() val, _ = nulType.Value()
} else { } else {
if !col.SQLType.IsJson() { if !(col.IsJSON || col.SQLType.IsJson()) {
engine.autoMapType(fieldValue) engine.autoMapType(fieldValue)
if table, ok := engine.Tables[fieldValue.Type()]; ok { if table, ok := engine.Tables[fieldValue.Type()]; ok {
if len(table.PrimaryKeys) == 1 { if len(table.PrimaryKeys) == 1 {

15
tag.go
View File

@ -55,14 +55,18 @@ var (
"CACHE": CacheTagHandler, "CACHE": CacheTagHandler,
"NOCACHE": NoCacheTagHandler, "NOCACHE": NoCacheTagHandler,
"COMMENT": CommentTagHandler, "COMMENT": CommentTagHandler,
"JSON": JSONTagHandler,
} }
) )
func init() { func init() {
for k := range core.SqlTypes { for k := range core.SqlTypes {
// don't overwrite
if _, ok := defaultTagHandlers[k]; !ok {
defaultTagHandlers[k] = SQLTypeTagHandler defaultTagHandlers[k] = SQLTypeTagHandler
} }
} }
}
// IgnoreTagHandler describes ignored tag handler // IgnoreTagHandler describes ignored tag handler
func IgnoreTagHandler(ctx *tagContext) error { func IgnoreTagHandler(ctx *tagContext) error {
@ -242,8 +246,19 @@ func SQLTypeTagHandler(ctx *tagContext) error {
return nil return nil
} }
// JSONTagHandler handle json tag
func JSONTagHandler(ctx *tagContext) error {
fmt.Println("fdsfafadfs")
ctx.col.IsJSON = true
if len(ctx.params) == 0 {
ctx.col.SQLType = core.SQLType{Name: core.Text}
}
return nil
}
// ExtendsTagHandler describes extends tag handler // ExtendsTagHandler describes extends tag handler
func ExtendsTagHandler(ctx *tagContext) error { func ExtendsTagHandler(ctx *tagContext) error {
ctx.ignoreNext = true
var fieldValue = ctx.fieldValue var fieldValue = ctx.fieldValue
var isPtr = false var isPtr = false
switch fieldValue.Kind() { switch fieldValue.Kind() {

View File

@ -598,3 +598,48 @@ func TestTagPrimarykey(t *testing.T) {
assert.True(t, col2.IsPrimaryKey) assert.True(t, col2.IsPrimaryKey)
assert.False(t, col2.IsAutoIncrement) assert.False(t, col2.IsAutoIncrement)
} }
func TestTagJSON(t *testing.T) {
assert.NoError(t, prepareEngine())
type TestTagJSON1 struct {
Stings []string `xorm:"json"`
}
assert.NoError(t, testEngine.Sync2(new(TestTagJSON1)))
table := testEngine.TableInfo(new(TestTagJSON1))
assert.NotNil(t, table)
assert.EqualValues(t, 1, len(table.Columns()))
assert.True(t, table.Columns()[0].IsJSON)
assert.EqualValues(t, "TEXT", table.Columns()[0].SQLType.Name)
tables, err := testEngine.DBMetas()
assert.NoError(t, err)
assert.EqualValues(t, 1, len(tables))
assert.EqualValues(t, 1, len(tables[0].Columns()))
if testEngine.dialect.DBType() != core.MSSQL {
assert.EqualValues(t, "TEXT", tables[0].Columns()[0].SQLType.Name)
}
assert.NoError(t, testEngine.DropTables(new(TestTagJSON1)))
type TestTagJSON2 struct {
Stings []string `xorm:"json MediumText"`
}
assert.NoError(t, testEngine.Sync2(new(TestTagJSON2)))
table = testEngine.TableInfo(new(TestTagJSON2))
assert.NotNil(t, table)
assert.EqualValues(t, 1, len(table.Columns()))
assert.True(t, table.Columns()[0].IsJSON)
assert.EqualValues(t, "MEDIUMTEXT", table.Columns()[0].SQLType.Name)
tables, err = testEngine.DBMetas()
assert.NoError(t, err)
assert.EqualValues(t, 1, len(tables))
assert.EqualValues(t, 1, len(tables[0].Columns()))
if testEngine.dialect.DBType() == core.MYSQL {
assert.EqualValues(t, "MEDIUMTEXT", tables[0].Columns()[0].SQLType.Name)
}
}