HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

resolved #389

This commit is contained in:
Lunny Xiao 2016-07-09 11:36:23 +08:00
parent 5612916fea
commit 811443400f
2 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
rows.go
View File

@ -35,6 +35,9 @@ func newRows(session *Session, bean interface{}) (*Rows, error) {
var args []interface{} var args []interface{}
rows.session.Statement.setRefValue(rValue(bean)) rows.session.Statement.setRefValue(rValue(bean))
if len(session.Statement.TableName()) <= 0 {
return nil, ErrTableNotFound
}
if rows.session.Statement.RawSQL == "" { if rows.session.Statement.RawSQL == "" {
sqlStr, args = rows.session.Statement.genGetSql(bean) sqlStr, args = rows.session.Statement.genGetSql(bean)

20
session.go
View File

@ -1014,6 +1014,10 @@ func (session *Session) Get(bean interface{}) (bool, error) {
} }
session.Statement.setRefValue(rValue(bean)) session.Statement.setRefValue(rValue(bean))
if len(session.Statement.TableName()) <= 0 {
return false, ErrTableNotFound
}
session.Statement.Limit(1) session.Statement.Limit(1)
var sqlStr string var sqlStr string
@ -1220,6 +1224,10 @@ func (session *Session) Find(rowsSlicePtr interface{}, condiBean ...interface{})
} }
} }
if len(session.Statement.TableName()) <= 0 {
return ErrTableNotFound
}
var table = session.Statement.RefTable var table = session.Statement.RefTable
var addedTableName = (len(session.Statement.JoinStr) > 0) var addedTableName = (len(session.Statement.JoinStr) > 0)
@ -2219,6 +2227,10 @@ func (session *Session) innerInsertMulti(rowsSlicePtr interface{}) (int64, error
bean := sliceValue.Index(0).Interface() bean := sliceValue.Index(0).Interface()
elementValue := rValue(bean) elementValue := rValue(bean)
session.Statement.setRefValue(elementValue) session.Statement.setRefValue(elementValue)
if len(session.Statement.TableName()) <= 0 {
return 0, ErrTableNotFound
}
table := session.Statement.RefTable table := session.Statement.RefTable
size := sliceValue.Len() size := sliceValue.Len()
@ -3056,6 +3068,10 @@ func (session *Session) value2Interface(col *core.Column, fieldValue reflect.Val
func (session *Session) innerInsert(bean interface{}) (int64, error) { func (session *Session) innerInsert(bean interface{}) (int64, error) {
session.Statement.setRefValue(rValue(bean)) session.Statement.setRefValue(rValue(bean))
if len(session.Statement.TableName()) <= 0 {
return 0, ErrTableNotFound
}
table := session.Statement.RefTable table := session.Statement.RefTable
// handle BeforeInsertProcessor // handle BeforeInsertProcessor
@ -3453,6 +3469,10 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
if isStruct { if isStruct {
session.Statement.setRefValue(v) session.Statement.setRefValue(v)
if len(session.Statement.TableName()) <= 0 {
return 0, ErrTableNotFound
}
if session.Statement.ColumnStr == "" { if session.Statement.ColumnStr == "" {
colNames, args = buildUpdates(session.Engine, session.Statement.RefTable, bean, false, false, colNames, args = buildUpdates(session.Engine, session.Statement.RefTable, bean, false, false,
false, false, session.Statement.allUseBool, session.Statement.useAllCols, false, false, session.Statement.allUseBool, session.Statement.useAllCols,