yangjianbo
ae191f72a4
修复软删除拦截器使用错误，确保默认查询过滤已删记录 仓储层改用 ent.Tx 与扫描辅助，避免 sql.Tx 断言问题 同步更新集成测试以覆盖事务与统计变动
34 lines
832 B
Go
34 lines
832 B
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
)
|
|
|
|
type sqlQueryer interface {
|
|
QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error)
|
|
}
|
|
|
|
// scanSingleRow executes a query and scans the first row into dest.
|
|
// If no rows are returned, sql.ErrNoRows is returned.
|
|
// 设计目的：仅依赖 QueryContext，避免 QueryRowContext 对 *sql.Tx 的强绑定，
|
|
// 让 ent.Tx 也能作为 sqlExecutor/Queryer 使用。
|
|
func scanSingleRow(ctx context.Context, q sqlQueryer, query string, args []any, dest ...any) error {
|
|
rows, err := q.QueryContext(ctx, query, args...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer rows.Close()
|
|
|
|
if !rows.Next() {
|
|
if err := rows.Err(); err != nil {
|
|
return err
|
|
}
|
|
return sql.ErrNoRows
|
|
}
|
|
if err := rows.Scan(dest...); err != nil {
|
|
return err
|
|
}
|
|
return rows.Err()
|
|
}
|