mirror of
https://github.com/astaxie/beego.git
synced 2024-11-22 15:30:55 +00:00
orm fix when use uint as pk
This commit is contained in:
parent
dc8f932009
commit
8f5ca303b5
@ -304,8 +304,12 @@ func (d *dbBase) Delete(q dbQuerier, mi *modelInfo, ind reflect.Value, tz *time.
|
|||||||
|
|
||||||
if num > 0 {
|
if num > 0 {
|
||||||
if mi.fields.pk.auto {
|
if mi.fields.pk.auto {
|
||||||
|
if mi.fields.pk.fieldType&IsPostiveIntegerField > 0 {
|
||||||
|
ind.Field(mi.fields.pk.fieldIndex).SetUint(0)
|
||||||
|
} else {
|
||||||
ind.Field(mi.fields.pk.fieldIndex).SetInt(0)
|
ind.Field(mi.fields.pk.fieldIndex).SetInt(0)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
err := d.deleteRels(q, mi, []interface{}{pkValue}, tz)
|
err := d.deleteRels(q, mi, []interface{}{pkValue}, tz)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -10,7 +10,11 @@ func getExistPk(mi *modelInfo, ind reflect.Value) (column string, value interfac
|
|||||||
fi := mi.fields.pk
|
fi := mi.fields.pk
|
||||||
|
|
||||||
v := ind.Field(fi.fieldIndex)
|
v := ind.Field(fi.fieldIndex)
|
||||||
if fi.fieldType&IsIntegerField > 0 {
|
if fi.fieldType&IsPostiveIntegerField > 0 {
|
||||||
|
vu := v.Uint()
|
||||||
|
exist = vu > 0
|
||||||
|
value = vu
|
||||||
|
} else if fi.fieldType&IsIntegerField > 0 {
|
||||||
vu := v.Int()
|
vu := v.Int()
|
||||||
exist = vu > 0
|
exist = vu > 0
|
||||||
value = vu
|
value = vu
|
||||||
|
@ -61,6 +61,7 @@ type DataNull struct {
|
|||||||
// only for mysql
|
// only for mysql
|
||||||
type UserBig struct {
|
type UserBig struct {
|
||||||
Id uint64
|
Id uint64
|
||||||
|
Name string
|
||||||
}
|
}
|
||||||
|
|
||||||
type User struct {
|
type User struct {
|
||||||
|
@ -70,9 +70,13 @@ func (o *orm) Insert(md interface{}) (int64, error) {
|
|||||||
}
|
}
|
||||||
if id > 0 {
|
if id > 0 {
|
||||||
if mi.fields.pk.auto {
|
if mi.fields.pk.auto {
|
||||||
|
if mi.fields.pk.fieldType&IsPostiveIntegerField > 0 {
|
||||||
|
ind.Field(mi.fields.pk.fieldIndex).SetUint(uint64(id))
|
||||||
|
} else {
|
||||||
ind.Field(mi.fields.pk.fieldIndex).SetInt(id)
|
ind.Field(mi.fields.pk.fieldIndex).SetInt(id)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return id, nil
|
return id, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -93,9 +97,13 @@ func (o *orm) Delete(md interface{}) (int64, error) {
|
|||||||
}
|
}
|
||||||
if num > 0 {
|
if num > 0 {
|
||||||
if mi.fields.pk.auto {
|
if mi.fields.pk.auto {
|
||||||
|
if mi.fields.pk.fieldType&IsPostiveIntegerField > 0 {
|
||||||
|
ind.Field(mi.fields.pk.fieldIndex).SetUint(0)
|
||||||
|
} else {
|
||||||
ind.Field(mi.fields.pk.fieldIndex).SetInt(0)
|
ind.Field(mi.fields.pk.fieldIndex).SetInt(0)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return num, nil
|
return num, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -34,9 +34,13 @@ func (o *insertSet) Insert(md interface{}) (int64, error) {
|
|||||||
}
|
}
|
||||||
if id > 0 {
|
if id > 0 {
|
||||||
if o.mi.fields.pk.auto {
|
if o.mi.fields.pk.auto {
|
||||||
|
if o.mi.fields.pk.fieldType&IsPostiveIntegerField > 0 {
|
||||||
|
ind.Field(o.mi.fields.pk.fieldIndex).SetUint(uint64(id))
|
||||||
|
} else {
|
||||||
ind.Field(o.mi.fields.pk.fieldIndex).SetInt(id)
|
ind.Field(o.mi.fields.pk.fieldIndex).SetInt(id)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return id, nil
|
return id, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
115
orm/orm_test.go
115
orm/orm_test.go
@ -211,6 +211,7 @@ func TestRegisterModels(t *testing.T) {
|
|||||||
RegisterModel(new(Post))
|
RegisterModel(new(Post))
|
||||||
RegisterModel(new(Tag))
|
RegisterModel(new(Tag))
|
||||||
RegisterModel(new(Comment))
|
RegisterModel(new(Comment))
|
||||||
|
RegisterModel(new(UserBig))
|
||||||
|
|
||||||
BootStrap()
|
BootStrap()
|
||||||
|
|
||||||
@ -231,8 +232,7 @@ func TestModelSyntax(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDataTypes(t *testing.T) {
|
var Data_Values = map[string]interface{}{
|
||||||
values := map[string]interface{}{
|
|
||||||
"Boolean": true,
|
"Boolean": true,
|
||||||
"Char": "char",
|
"Char": "char",
|
||||||
"Text": "text",
|
"Text": "text",
|
||||||
@ -253,11 +253,13 @@ func TestDataTypes(t *testing.T) {
|
|||||||
"Float32": float32(100.1234),
|
"Float32": float32(100.1234),
|
||||||
"Float64": float64(100.1234),
|
"Float64": float64(100.1234),
|
||||||
"Decimal": float64(100.1234),
|
"Decimal": float64(100.1234),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDataTypes(t *testing.T) {
|
||||||
d := Data{}
|
d := Data{}
|
||||||
ind := reflect.Indirect(reflect.ValueOf(&d))
|
ind := reflect.Indirect(reflect.ValueOf(&d))
|
||||||
|
|
||||||
for name, value := range values {
|
for name, value := range Data_Values {
|
||||||
e := ind.FieldByName(name)
|
e := ind.FieldByName(name)
|
||||||
e.Set(reflect.ValueOf(value))
|
e.Set(reflect.ValueOf(value))
|
||||||
}
|
}
|
||||||
@ -272,7 +274,7 @@ func TestDataTypes(t *testing.T) {
|
|||||||
|
|
||||||
ind = reflect.Indirect(reflect.ValueOf(&d))
|
ind = reflect.Indirect(reflect.ValueOf(&d))
|
||||||
|
|
||||||
for name, value := range values {
|
for name, value := range Data_Values {
|
||||||
e := ind.FieldByName(name)
|
e := ind.FieldByName(name)
|
||||||
vu := e.Interface()
|
vu := e.Interface()
|
||||||
switch name {
|
switch name {
|
||||||
@ -376,6 +378,17 @@ func TestCRUD(t *testing.T) {
|
|||||||
u = &User{Id: 100}
|
u = &User{Id: 100}
|
||||||
err = dORM.Read(u)
|
err = dORM.Read(u)
|
||||||
throwFail(t, AssertIs(err, T_Equal, ErrNoRows))
|
throwFail(t, AssertIs(err, T_Equal, ErrNoRows))
|
||||||
|
|
||||||
|
ub := UserBig{}
|
||||||
|
ub.Name = "name"
|
||||||
|
id, err = dORM.Insert(&ub)
|
||||||
|
throwFail(t, err)
|
||||||
|
throwFail(t, AssertIs(id, T_Equal, 1))
|
||||||
|
|
||||||
|
ub = UserBig{Id: 1}
|
||||||
|
err = dORM.Read(&ub)
|
||||||
|
throwFail(t, err)
|
||||||
|
throwFail(t, AssertIs(ub.Name, T_Equal, "name"))
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestInsertTestData(t *testing.T) {
|
func TestInsertTestData(t *testing.T) {
|
||||||
@ -823,7 +836,15 @@ func TestPrepareInsert(t *testing.T) {
|
|||||||
throwFail(t, AssertIs(err, T_Equal, ErrStmtClosed))
|
throwFail(t, AssertIs(err, T_Equal, ErrStmtClosed))
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRaw(t *testing.T) {
|
func TestRawQueryRow(t *testing.T) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRawQueryRows(t *testing.T) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRawValues(t *testing.T) {
|
||||||
switch {
|
switch {
|
||||||
case IsMysql || IsSqlite:
|
case IsMysql || IsSqlite:
|
||||||
|
|
||||||
@ -860,42 +881,7 @@ func TestRaw(t *testing.T) {
|
|||||||
if num == 3 {
|
if num == 3 {
|
||||||
throwFail(t, AssertIs(list[0], T_Equal, "2"))
|
throwFail(t, AssertIs(list[0], T_Equal, "2"))
|
||||||
throwFail(t, AssertIs(list[1], T_Equal, "3"))
|
throwFail(t, AssertIs(list[1], T_Equal, "3"))
|
||||||
throwFail(t, AssertIs(list[2], T_Equal, ""))
|
throwFail(t, AssertIs(list[2], T_Equal, nil))
|
||||||
}
|
|
||||||
|
|
||||||
pre, err := dORM.Raw("INSERT INTO tag (name) VALUES (?)").Prepare()
|
|
||||||
throwFail(t, err)
|
|
||||||
if pre != nil {
|
|
||||||
r, err := pre.Exec("name1")
|
|
||||||
throwFail(t, err)
|
|
||||||
|
|
||||||
tid, err := r.LastInsertId()
|
|
||||||
throwFail(t, err)
|
|
||||||
throwFail(t, AssertIs(tid, T_Large, 0))
|
|
||||||
|
|
||||||
r, err = pre.Exec("name2")
|
|
||||||
throwFail(t, err)
|
|
||||||
|
|
||||||
id, err := r.LastInsertId()
|
|
||||||
throwFail(t, err)
|
|
||||||
throwFail(t, AssertIs(id, T_Equal, tid+1))
|
|
||||||
|
|
||||||
r, err = pre.Exec("name3")
|
|
||||||
throwFail(t, err)
|
|
||||||
|
|
||||||
id, err = r.LastInsertId()
|
|
||||||
throwFail(t, err)
|
|
||||||
throwFail(t, AssertIs(id, T_Equal, tid+2))
|
|
||||||
|
|
||||||
err = pre.Close()
|
|
||||||
throwFail(t, err)
|
|
||||||
|
|
||||||
res, err := dORM.Raw("DELETE FROM tag WHERE name IN (?, ?, ?)", []string{"name1", "name2", "name3"}).Exec()
|
|
||||||
throwFail(t, err)
|
|
||||||
|
|
||||||
num, err := res.RowsAffected()
|
|
||||||
throwFail(t, err)
|
|
||||||
throwFail(t, AssertIs(num, T_Equal, 3))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
case IsPostgres:
|
case IsPostgres:
|
||||||
@ -933,8 +919,51 @@ func TestRaw(t *testing.T) {
|
|||||||
if num == 3 {
|
if num == 3 {
|
||||||
throwFail(t, AssertIs(list[0], T_Equal, "2"))
|
throwFail(t, AssertIs(list[0], T_Equal, "2"))
|
||||||
throwFail(t, AssertIs(list[1], T_Equal, "3"))
|
throwFail(t, AssertIs(list[1], T_Equal, "3"))
|
||||||
throwFail(t, AssertIs(list[2], T_Equal, ""))
|
throwFail(t, AssertIs(list[2], T_Equal, nil))
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRawPrepare(t *testing.T) {
|
||||||
|
switch {
|
||||||
|
case IsMysql || IsSqlite:
|
||||||
|
|
||||||
|
pre, err := dORM.Raw("INSERT INTO tag (name) VALUES (?)").Prepare()
|
||||||
|
throwFail(t, err)
|
||||||
|
if pre != nil {
|
||||||
|
r, err := pre.Exec("name1")
|
||||||
|
throwFail(t, err)
|
||||||
|
|
||||||
|
tid, err := r.LastInsertId()
|
||||||
|
throwFail(t, err)
|
||||||
|
throwFail(t, AssertIs(tid, T_Large, 0))
|
||||||
|
|
||||||
|
r, err = pre.Exec("name2")
|
||||||
|
throwFail(t, err)
|
||||||
|
|
||||||
|
id, err := r.LastInsertId()
|
||||||
|
throwFail(t, err)
|
||||||
|
throwFail(t, AssertIs(id, T_Equal, tid+1))
|
||||||
|
|
||||||
|
r, err = pre.Exec("name3")
|
||||||
|
throwFail(t, err)
|
||||||
|
|
||||||
|
id, err = r.LastInsertId()
|
||||||
|
throwFail(t, err)
|
||||||
|
throwFail(t, AssertIs(id, T_Equal, tid+2))
|
||||||
|
|
||||||
|
err = pre.Close()
|
||||||
|
throwFail(t, err)
|
||||||
|
|
||||||
|
res, err := dORM.Raw("DELETE FROM tag WHERE name IN (?, ?, ?)", []string{"name1", "name2", "name3"}).Exec()
|
||||||
|
throwFail(t, err)
|
||||||
|
|
||||||
|
num, err := res.RowsAffected()
|
||||||
|
throwFail(t, err)
|
||||||
|
throwFail(t, AssertIs(num, T_Equal, 3))
|
||||||
|
}
|
||||||
|
|
||||||
|
case IsPostgres:
|
||||||
|
|
||||||
pre, err := dORM.Raw(`INSERT INTO "tag" ("name") VALUES (?) RETURNING "id"`).Prepare()
|
pre, err := dORM.Raw(`INSERT INTO "tag" ("name") VALUES (?) RETURNING "id"`).Prepare()
|
||||||
throwFail(t, err)
|
throwFail(t, err)
|
||||||
|
Loading…
Reference in New Issue
Block a user