Add Truncate.
This commit is contained in:
parent
52855dae32
commit
2643312baa
|
@ -1233,12 +1233,20 @@ func (engine *Engine) Update(bean interface{}, condiBeans ...interface{}) (int64
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete records, bean's non-empty fields are conditions
|
// Delete records, bean's non-empty fields are conditions
|
||||||
|
// At least one condition must be set.
|
||||||
func (engine *Engine) Delete(beans ...interface{}) (int64, error) {
|
func (engine *Engine) Delete(beans ...interface{}) (int64, error) {
|
||||||
session := engine.NewSession()
|
session := engine.NewSession()
|
||||||
defer session.Close()
|
defer session.Close()
|
||||||
return session.Delete(beans...)
|
return session.Delete(beans...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Truncate records, bean's non-empty fields are conditions
|
||||||
|
func (engine *Engine) Truncate(beans ...interface{}) (int64, error) {
|
||||||
|
session := engine.NewSession()
|
||||||
|
defer session.Close()
|
||||||
|
return session.Truncate(beans...)
|
||||||
|
}
|
||||||
|
|
||||||
// Get retrieve one record from table, bean's non-empty fields
|
// Get retrieve one record from table, bean's non-empty fields
|
||||||
// are conditions
|
// are conditions
|
||||||
func (engine *Engine) Get(beans ...interface{}) (bool, error) {
|
func (engine *Engine) Get(beans ...interface{}) (bool, error) {
|
||||||
|
|
|
@ -266,3 +266,28 @@ func TestDelete2(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.False(t, has)
|
assert.False(t, has)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestTruncate(t *testing.T) {
|
||||||
|
assert.NoError(t, PrepareEngine())
|
||||||
|
|
||||||
|
type User struct {
|
||||||
|
Uid int64 `xorm:"id pk not null autoincr"`
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, testEngine.Sync(new(User)))
|
||||||
|
|
||||||
|
cnt, err := testEngine.Insert(&User{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
_, err = testEngine.Delete(&User{})
|
||||||
|
assert.Error(t, err)
|
||||||
|
|
||||||
|
_, err = testEngine.Truncate(&User{})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
user2 := User{}
|
||||||
|
has, err := testEngine.ID(1).Get(&user2)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.False(t, has)
|
||||||
|
}
|
||||||
|
|
|
@ -31,6 +31,7 @@ type Interface interface {
|
||||||
Decr(column string, arg ...interface{}) *Session
|
Decr(column string, arg ...interface{}) *Session
|
||||||
Desc(...string) *Session
|
Desc(...string) *Session
|
||||||
Delete(...interface{}) (int64, error)
|
Delete(...interface{}) (int64, error)
|
||||||
|
Truncate(...interface{}) (int64, error)
|
||||||
Distinct(columns ...string) *Session
|
Distinct(columns ...string) *Session
|
||||||
DropIndexes(bean interface{}) error
|
DropIndexes(bean interface{}) error
|
||||||
Exec(sqlOrArgs ...interface{}) (sql.Result, error)
|
Exec(sqlOrArgs ...interface{}) (sql.Result, error)
|
||||||
|
|
|
@ -91,7 +91,17 @@ func (session *Session) cacheDelete(table *schemas.Table, tableName, sqlStr stri
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete records, bean's non-empty fields are conditions
|
// Delete records, bean's non-empty fields are conditions
|
||||||
|
// At least one condition must be set.
|
||||||
func (session *Session) Delete(beans ...interface{}) (int64, error) {
|
func (session *Session) Delete(beans ...interface{}) (int64, error) {
|
||||||
|
return session.delete(beans, true)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Truncate records, bean's non-empty fields are conditions
|
||||||
|
func (session *Session) Truncate(beans ...interface{}) (int64, error) {
|
||||||
|
return session.delete(beans, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (session *Session) delete(beans []interface{}, mustHaveConditions bool) (int64, error) {
|
||||||
if session.isAutoClose {
|
if session.isAutoClose {
|
||||||
defer session.Close()
|
defer session.Close()
|
||||||
}
|
}
|
||||||
|
@ -127,7 +137,7 @@ func (session *Session) Delete(beans ...interface{}) (int64, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
pLimitN := session.statement.LimitN
|
pLimitN := session.statement.LimitN
|
||||||
if condWriter.Len() == 0 && (pLimitN == nil || *pLimitN == 0) {
|
if mustHaveConditions && condWriter.Len() == 0 && (pLimitN == nil || *pLimitN == 0) {
|
||||||
return 0, ErrNeedDeletedCond
|
return 0, ErrNeedDeletedCond
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue