8cb2d3b352
统一 usage_log_repo 查询的 Close 错误处理，避免\n成功路径吞掉关闭失败 scanSingleRow 使用 errors.Join 合并 Close 错误，\n保留 ErrNoRows 可判定 测试: make -C backend test-unit
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"errors"
|
||
)
|
||
|
||
type sqlQueryer interface {
|
||
QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error)
|
||
}
|
||
|
||
// scanSingleRow 执行查询并扫描第一行到 dest。
|
||
// 若无结果，可通过 errors.Is(err, sql.ErrNoRows) 判断。
|
||
// 如果 Close 失败，会与原始错误合并返回。
|
||
// 设计目的：仅依赖 QueryContext，避免 QueryRowContext 对 *sql.Tx 的强绑定，
|
||
// 让 ent.Tx 也能作为 sqlExecutor/Queryer 使用。
|
||
func scanSingleRow(ctx context.Context, q sqlQueryer, query string, args []any, dest ...any) (err error) {
|
||
rows, err := q.QueryContext(ctx, query, args...)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer func() {
|
||
if closeErr := rows.Close(); closeErr != nil {
|
||
err = errors.Join(err, closeErr)
|
||
}
|
||
}()
|
||
|
||
if !rows.Next() {
|
||
if err = rows.Err(); err != nil {
|
||
return err
|
||
}
|
||
return sql.ErrNoRows
|
||
}
|
||
if err = rows.Scan(dest...); err != nil {
|
||
return err
|
||
}
|
||
if err = rows.Err(); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|