New Prepare useage #2061
|
@ -93,7 +93,7 @@ func (s *Stmt) ExecContext(ctx context.Context, args ...interface{}) (sql.Result
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
res, err := s.Stmt.ExecContext(ctx, args)
|
res, err := s.Stmt.ExecContext(ctx, args...)
|
||||||
hookCtx.End(ctx, res, err)
|
hookCtx.End(ctx, res, err)
|
||||||
if err := s.db.afterProcess(hookCtx); err != nil {
|
if err := s.db.afterProcess(hookCtx); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
|
@ -933,7 +933,7 @@ func TestGetWithPrepare(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
var v1 GetVarsWithPrepare
|
var v1 GetVarsWithPrepare
|
||||||
has, err := testEngine.Prepare().Get(&v1)
|
has, err := testEngine.Prepare().ID(1).Get(&v1)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, "xlw", v1.Name)
|
assert.EqualValues(t, "xlw", v1.Name)
|
||||||
|
@ -943,16 +943,36 @@ func TestGetWithPrepare(t *testing.T) {
|
||||||
defer sess.Close()
|
defer sess.Close()
|
||||||
|
|
||||||
var v2 GetVarsWithPrepare
|
var v2 GetVarsWithPrepare
|
||||||
has, err = sess.Prepare().Get(&v2)
|
has, err = sess.Prepare().ID(1).Get(&v2)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, "xlw", v2.Name)
|
assert.EqualValues(t, "xlw", v2.Name)
|
||||||
assert.EqualValues(t, 42, v2.Age)
|
assert.EqualValues(t, 42, v2.Age)
|
||||||
|
|
||||||
var v3 GetVarsWithPrepare
|
var v3 GetVarsWithPrepare
|
||||||
has, err = sess.Prepare().Get(&v3)
|
has, err = sess.Prepare().ID(1).Get(&v3)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.True(t, has)
|
assert.True(t, has)
|
||||||
assert.EqualValues(t, "xlw", v3.Name)
|
assert.EqualValues(t, "xlw", v3.Name)
|
||||||
assert.EqualValues(t, 42, v3.Age)
|
assert.EqualValues(t, 42, v3.Age)
|
||||||
|
|
||||||
|
err = sess.Begin()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
cnt, err := sess.Prepare().Insert(&GetVarsWithPrepare{
|
||||||
|
Name: "xlw2",
|
||||||
|
Age: 12,
|
||||||
|
})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
cnt, err = sess.Prepare().Insert(&GetVarsWithPrepare{
|
||||||
|
Name: "xlw3",
|
||||||
|
Age: 13,
|
||||||
|
})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
err = sess.Commit()
|
||||||
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
|
@ -131,6 +131,7 @@ func newSession(engine *Engine) *Session {
|
||||||
afterClosures: make([]func(interface{}), 0),
|
afterClosures: make([]func(interface{}), 0),
|
||||||
afterProcessors: make([]executedProcessor, 0),
|
afterProcessors: make([]executedProcessor, 0),
|
||||||
stmtCache: make(map[uint32]*core.Stmt),
|
stmtCache: make(map[uint32]*core.Stmt),
|
||||||
|
txStmtCache: make(map[uint32]*core.Stmt),
|
||||||
|
|
||||||
lastSQL: "",
|
lastSQL: "",
|
||||||
lastSQLArgs: make([]interface{}, 0),
|
lastSQLArgs: make([]interface{}, 0),
|
||||||
|
@ -151,6 +152,12 @@ func (session *Session) Close() error {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for _, v := range session.txStmtCache {
|
||||||
|
if err := v.Close(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if !session.isClosed {
|
if !session.isClosed {
|
||||||
// When Close be called, if session is a transaction and do not call
|
// When Close be called, if session is a transaction and do not call
|
||||||
// Commit or Rollback, then call Rollback.
|
// Commit or Rollback, then call Rollback.
|
||||||
|
@ -161,6 +168,7 @@ func (session *Session) Close() error {
|
||||||
}
|
}
|
||||||
session.tx = nil
|
session.tx = nil
|
||||||
session.stmtCache = nil
|
session.stmtCache = nil
|
||||||
|
session.txStmtCache = nil
|
||||||
session.isClosed = true
|
session.isClosed = true
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|
|
@ -157,6 +157,13 @@ func (session *Session) exec(sqlStr string, args ...interface{}) (sql.Result, er
|
||||||
session.lastSQLArgs = args
|
session.lastSQLArgs = args
|
||||||
|
|
||||||
if !session.isAutoCommit {
|
if !session.isAutoCommit {
|
||||||
|
if session.prepareStmt {
|
||||||
|
stmt, err := session.doPrepareTx(sqlStr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return stmt.ExecContext(session.ctx, args...)
|
||||||
|
}
|
||||||
return session.tx.ExecContext(session.ctx, sqlStr, args...)
|
return session.tx.ExecContext(session.ctx, sqlStr, args...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -165,12 +172,7 @@ func (session *Session) exec(sqlStr string, args ...interface{}) (sql.Result, er
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
return stmt.ExecContext(session.ctx, args...)
|
||||||
res, err := stmt.ExecContext(session.ctx, args...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return res, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return session.DB().ExecContext(session.ctx, sqlStr, args...)
|
return session.DB().ExecContext(session.ctx, sqlStr, args...)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user