mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-11 02:24:45 +08:00
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()
|
|||
|
|
}
|