Question

对 Query 进行包装，用于返回单条记录。

• 没找到记录，返回 ErrNoRows 。
• 方法 Row.Scan 主动释放连接。

package main

import (
	"database/sql"
	"log"

	_ "github.com/mattn/go-sqlite3"
)

func main() {
	log.SetFlags(log.Lshortfile)
    
	db, err := sql.Open("sqlite3", "./test.db")
	if err != nil { log.Fatalln(err) }
	defer db.Close()

    // ------------------
    
	q := `SELECT name FROM user WHERE id > ?`

	var name string
	err = db.QueryRow(q, 10).Scan(&name)

	if err == sql.ErrNoRows { 
		println("norow") 
        return
	} else if err != nil { 
		log.Fatalln(err) 
	}

	println(name)
}

源码剖析

调用 Query 返回 Rows 包装成 Row 单行记录。
在 Row.Scan 操作中，会调用 Rows.Close ，确保归还连接。

// database/sql/sql.go

func (db *DB) QueryRowContext(ctx, query string, args ...any) *Row {
	rows, err := db.QueryContext(ctx, query, args...)
	return &Row{rows: rows, err: err}
}

func (r *Row) Scan(dest ...any) error {
	if r.err != nil {
		return r.err
	}

    // 确保 rows.Close 调用，归还连接。
	defer r.rows.Close()
    
	for _, dp := range dest {
		if _, ok := dp.(*RawBytes); ok {
			return errors.New("sql: RawBytes isn't allowed on Row.Scan")
		}
	}

    // 没有记录。
	if !r.rows.Next() {
        
        // 检查是不是意外导致。
		if err := r.rows.Err(); err != nil {
			return err
		}
        
		return ErrNoRows
	}
    
	err := r.rows.Scan(dest...)
	if err != nil {
		return err
	}
    
	// Make sure the query can be processed to completion with no errors.
	return r.rows.Close()
}