45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"time"
|
|
|
|
"apigo.cc/go/log"
|
|
)
|
|
|
|
type Stmt struct {
|
|
conn *sql.Stmt
|
|
lastSql *string
|
|
lastArgs []any
|
|
Error error
|
|
logger *dbLogger
|
|
}
|
|
|
|
func (stmt *Stmt) Exec(args ...any) *ExecResult {
|
|
stmt.lastArgs = args
|
|
if stmt.conn == nil {
|
|
return &ExecResult{Sql: stmt.lastSql, Args: stmt.lastArgs, usedTime: -1, logger: stmt.logger, Error: errors.New("operate on a bad connection")}
|
|
}
|
|
startTime := time.Now()
|
|
r, err := stmt.conn.Exec(args...)
|
|
endTime := time.Now()
|
|
usedTime := log.MakeUsedTime(startTime, endTime)
|
|
if err != nil {
|
|
stmt.logger.LogQueryError(err.Error(), *stmt.lastSql, stmt.lastArgs, usedTime)
|
|
return &ExecResult{Sql: stmt.lastSql, Args: stmt.lastArgs, usedTime: usedTime, logger: stmt.logger, Error: err}
|
|
}
|
|
return &ExecResult{Sql: stmt.lastSql, Args: stmt.lastArgs, usedTime: usedTime, logger: stmt.logger, result: r}
|
|
}
|
|
|
|
func (stmt *Stmt) Close() error {
|
|
if stmt.conn == nil {
|
|
return errors.New("operate on a bad connection")
|
|
}
|
|
err := stmt.conn.Close()
|
|
if err != nil {
|
|
stmt.logger.LogQueryError(err.Error(), *stmt.lastSql, stmt.lastArgs, -1)
|
|
}
|
|
return err
|
|
}
|