func checkResult(c *C, se tidb.Session, affectedRows uint64, insertID uint64) {
gotRows := se.AffectedRows()
c.Assert(gotRows, Equals, affectedRows)
gotID := se.LastInsertID()
c.Assert(gotID, Equals, insertID)
}
func sessionExec(c *C, se tidb.Session, sql string) ([]rset.Recordset, error) {
se.Execute("BEGIN;")
r, err := se.Execute(sql)
c.Assert(err, IsNil)
se.Execute("COMMIT;")
return r, err
}
func exec(se tidb.Session, sql string, args ...interface{}) (ast.RecordSet, error) {
if len(args) == 0 {
rs, err := se.Execute(sql)
if err == nil && len(rs) > 0 {
return rs[0], nil
}
return nil, err
}
stmtID, _, _, err := se.PrepareStmt(sql)
if err != nil {
return nil, err
}
rs, err := se.ExecutePreparedStmt(stmtID, args...)
if err != nil {
return nil, err
}
return rs, nil
}
func mustExecSQL(c *C, se tidb.Session, sql string) {
_, err := se.Execute(sql)
c.Assert(err, IsNil)
}