sqlx：ScanRows 和 ScanAll方法 about ekit HOT 9 CLOSED

type Scanner struct {
	sqlRows      *sql.Rows
	cachedValues []any
}

func NewScanner(r *sql.Rows) (*Scanner, error) {
        if r == nil {
            // .....
        }
	colsInfo, err := r.ColumnTypes()
	if err != nil {
		return nil, err
	}
	values := make([]any, len(colsInfo))
	for i, colInfo := range colsInfo {
		typ := colInfo.ScanType()
		for typ.Kind() == reflect.Pointer {
			typ = typ.Elem()
		}
		values[i] = reflect.New(typ).Interface()
	}
	return &Scanner{sqlRows: r, cachedValues: values}, nil
}

func (s *Scanner) ScanOne() ([]any, error) {
        // 每次都用同一个列集合来获取数据
	err := s.sqlRows.Scan(s.cachedValues...)
	if err != nil {
		return nil, err
	}
        // 获取当前行的各个列值
	values := make([]any, len(s.cachedValues))
	for i := 0; i < len(s.cachedValues); i++ {
		values[i] = reflect.ValueOf(s.cachedValues[i]).Elem().Interface()
	}
	return values, nil
}

func (s *Scanner) ScanAll() ([][]any, error) {
	res := make([][]any, 0, 32)
	for s.sqlRows.Next() {
		cols, err := s.ScanOne()
		if err != nil {
			return nil, err
		}
		res = append(res, cols)
	}
	return res, nil
}

// 用v1版Scnner重构下面的代码通过了测试集合
func ScanRows(rows *sql.Rows) ([]any, error) {
	scanner, err := NewScanner(rows)
	if err != nil {
		return nil, err
	}
	return scanner.ScanOne()
}

func ScanAll(rows *sql.Rows) ([][]any, error) {
	scanner, err := NewScanner(rows)
	if err != nil {
		return nil, err
	}
	return scanner.ScanAll()
}

defer rows.Close()
var all
for rows.Next() {
     var values
     if err := rows.Scan(values...); err != nil {
         // 扫描当前values出错
     }
     // 代码走到这,可以安全地使用values
     all := append(all, values)
}
if rows.Err() != nil {
    // 迭代期间出错 ....可能导致当前结果集读取不完整
}

// 走到这可以安全地使用all

all, err := ScanAll(rows)
if err != nil {
    // Scan出错
}
if rows.Err() != nil {
   // 迭代期间出错 ....可能导致当前结果集读取不完整
}
// 走到这可以安全地使用all

defer rows.Close()
var all
for rows.Next() {
     values, err := ScanRows(rows)
     if err != nil {
         // Scan报错
     }
     // 代码走到这,可以安全地使用values
     all := append(all, values)
}
if rows.Err() != nil {
    // 迭代期间出错 ....可能导致当前结果集读取不完整
}

var ErrNoMoreRows = errors.New("ekit: no more rows in result set")

type Scanner struct {
	sqlRows      *sql.Rows
	cachedValues []any
}

func NewScanner(r *sql.Rows) (*Scanner, error) {
	// 读取列的元数据缓存
	colsInfo, err := r.ColumnTypes()
	if err != nil {
		return nil, err
	}
	values := make([]any, len(colsInfo))
	for i, colInfo := range colsInfo {
		typ := colInfo.ScanType()
		for typ.Kind() == reflect.Pointer {
			typ = typ.Elem()
		}
		values[i] = reflect.New(typ).Interface()
	}
	return &Scanner{sqlRows: r, cachedValues: values}, nil
}

func (s *Scanner) ScanOne() ([]any, error) {
	// 调用Next
	if !s.sqlRows.Next() {
                 // 检测迭代错误
		if err := s.sqlRows.Err(); err != nil {
			return nil, err
		}
		// 数据读取完
		return nil, fmt.Errorf("%w", ErrNoMoreRows)
	}
	err := s.sqlRows.Scan(s.cachedValues...)
	if err != nil {
		return nil, err
	}
	values := make([]any, len(s.cachedValues))
	for i := 0; i < len(s.cachedValues); i++ {
		values[i] = reflect.ValueOf(s.cachedValues[i]).Elem().Interface()
	}
	return values, nil
}

func (s *Scanner) ScanAll() ([][]any, error) {
	res := make([][]any, 0, 32)
	for {
		cols, err := s.ScanOne()
		if err != nil {
                         // 需要区分正常结束,只是没有数据的情况
			if errors.Is(err, ErrNoMoreRows) {
				break
			}
			return nil, err
		}
		res = append(res, cols)
	}
	return res, nil
}

func ScanRow[T any](row sql.Row) ([]T, error) {
	// 列的个数如何定？
	res := make([]T, 0, 8)
	err := row.Scan(res)
	if err != nil {
		return nil, err
	}
	return res, nil
}