Fix DBMetas returned unsigned tinyint

This commit is contained in:
Lunny Xiao 2021-07-28 23:05:57 +08:00
parent 5240459858
commit c88bb03dd8
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
4 changed files with 77 additions and 20 deletions

View File

@ -493,6 +493,45 @@ func TestUnsignedUint32(t *testing.T) {
assert.EqualValues(t, uint64(math.MaxUint32), v.Id) assert.EqualValues(t, uint64(math.MaxUint32), v.Id)
} }
func TestUnsignedTinyInt(t *testing.T) {
type MyUnsignedTinyIntStruct struct {
Id uint8 `xorm:"unsigned tinyint"`
}
assert.NoError(t, PrepareEngine())
assertSync(t, new(MyUnsignedTinyIntStruct))
tables, err := testEngine.DBMetas()
assert.NoError(t, err)
assert.EqualValues(t, 1, len(tables))
assert.EqualValues(t, 1, len(tables[0].Columns()))
switch testEngine.Dialect().URI().DBType {
case schemas.SQLITE:
assert.EqualValues(t, "INTEGER", tables[0].Columns()[0].SQLType.Name)
case schemas.MYSQL:
assert.EqualValues(t, "UNSIGNED TINYINT", tables[0].Columns()[0].SQLType.Name)
case schemas.POSTGRES:
assert.EqualValues(t, "INT", tables[0].Columns()[0].SQLType.Name)
case schemas.MSSQL:
assert.EqualValues(t, "INT", tables[0].Columns()[0].SQLType.Name)
default:
assert.False(t, true, "Unsigned is not implemented")
}
cnt, err := testEngine.Insert(&MyUnsignedTinyIntStruct{
Id: math.MaxUint8,
})
assert.NoError(t, err)
assert.EqualValues(t, 1, cnt)
var v MyUnsignedTinyIntStruct
has, err := testEngine.Get(&v)
assert.NoError(t, err)
assert.True(t, has)
assert.EqualValues(t, uint64(math.MaxUint32), v.Id)
}
type MyDecimal big.Int type MyDecimal big.Int
func (d *MyDecimal) FromDB(data []byte) error { func (d *MyDecimal) FromDB(data []byte) error {

View File

@ -92,16 +92,19 @@ func (s *SQLType) IsXML() bool {
// enumerates all the database column types // enumerates all the database column types
var ( var (
Bit = "BIT" Bit = "BIT"
UnsignedBit = "UNSIGNED BIT" UnsignedBit = "UNSIGNED BIT"
TinyInt = "TINYINT" TinyInt = "TINYINT"
SmallInt = "SMALLINT" UnsignedTinyInt = "UNSIGNED TINYINT"
MediumInt = "MEDIUMINT" SmallInt = "SMALLINT"
Int = "INT" UnsignedSmallInt = "UNSIGNED SMALLINT"
UnsignedInt = "UNSIGNED INT" MediumInt = "MEDIUMINT"
Integer = "INTEGER" UnsignedMediumInt = "UNSIGNED MEDIUMINT"
BigInt = "BIGINT" Int = "INT"
UnsignedBigInt = "UNSIGNED BIGINT" UnsignedInt = "UNSIGNED INT"
Integer = "INTEGER"
BigInt = "BIGINT"
UnsignedBigInt = "UNSIGNED BIGINT"
Enum = "ENUM" Enum = "ENUM"
Set = "SET" Set = "SET"
@ -158,16 +161,19 @@ var (
Array = "ARRAY" Array = "ARRAY"
SqlTypes = map[string]int{ SqlTypes = map[string]int{
Bit: NUMERIC_TYPE, Bit: NUMERIC_TYPE,
UnsignedBit: NUMERIC_TYPE, UnsignedBit: NUMERIC_TYPE,
TinyInt: NUMERIC_TYPE, TinyInt: NUMERIC_TYPE,
SmallInt: NUMERIC_TYPE, UnsignedTinyInt: NUMERIC_TYPE,
MediumInt: NUMERIC_TYPE, SmallInt: NUMERIC_TYPE,
Int: NUMERIC_TYPE, UnsignedSmallInt: NUMERIC_TYPE,
UnsignedInt: NUMERIC_TYPE, MediumInt: NUMERIC_TYPE,
Integer: NUMERIC_TYPE, UnsignedMediumInt: NUMERIC_TYPE,
BigInt: NUMERIC_TYPE, Int: NUMERIC_TYPE,
UnsignedBigInt: NUMERIC_TYPE, UnsignedInt: NUMERIC_TYPE,
Integer: NUMERIC_TYPE,
BigInt: NUMERIC_TYPE,
UnsignedBigInt: NUMERIC_TYPE,
Enum: TEXT_TYPE, Enum: TEXT_TYPE,
Set: TEXT_TYPE, Set: TEXT_TYPE,

View File

@ -217,6 +217,10 @@ func (parser *Parser) parseFieldWithTags(table *schemas.Table, fieldIndex int, f
if col.SQLType.Name == "" { if col.SQLType.Name == "" {
col.SQLType = schemas.Type2SQLType(field.Type) col.SQLType = schemas.Type2SQLType(field.Type)
} }
if ctx.isUnsigned && col.SQLType.IsNumeric() && !strings.HasPrefix(col.SQLType.Name, "UNSIGNED") {
col.SQLType.Name = col.SQLType.Name + " UNSIGNED"
}
parser.dialect.SQLType(col) parser.dialect.SQLType(col)
if col.Length == 0 { if col.Length == 0 {
col.Length = col.SQLType.DefaultLength col.Length = col.SQLType.DefaultLength

View File

@ -93,6 +93,7 @@ type Context struct {
hasCacheTag bool hasCacheTag bool
hasNoCacheTag bool hasNoCacheTag bool
ignoreNext bool ignoreNext bool
isUnsigned bool
} }
// Handler describes tag handler for XORM // Handler describes tag handler for XORM
@ -122,6 +123,7 @@ var (
"NOCACHE": NoCacheTagHandler, "NOCACHE": NoCacheTagHandler,
"COMMENT": CommentTagHandler, "COMMENT": CommentTagHandler,
"EXTENDS": ExtendsTagHandler, "EXTENDS": ExtendsTagHandler,
"UNSIGNED": UnsignedTagHandler,
} }
) )
@ -268,6 +270,12 @@ func UniqueTagHandler(ctx *Context) error {
return nil return nil
} }
// UnsignedTagHandler represents the column is unsigned
func UnsignedTagHandler(ctx *Context) error {
ctx.isUnsigned = true
return nil
}
// CommentTagHandler add comment to column // CommentTagHandler add comment to column
func CommentTagHandler(ctx *Context) error { func CommentTagHandler(ctx *Context) error {
if len(ctx.params) > 0 { if len(ctx.params) > 0 {