Function NewSqlRowIter

pgserver/iter.go:94–133

(rows *stdsql.Rows, schema sql.Schema)

}

func NewSqlRowIter(rows *stdsql.Rows, schema sql.Schema) (*SqlRowIter, error) {
	columns, err := rows.ColumnTypes()
	if err != nil {
		return nil, err
	}

	width := max(len(columns), len(schema))
	buf := make([]any, width)
	ptrs := make([]any, width)
	for i := range buf {
		ptrs[i] = &buf[i]
	}

	var decimals []int
	for i, c := range columns {
		if strings.HasPrefix(c.DatabaseTypeName(), "DECIMAL") {
			decimals = append(decimals, i)
		}
	}

	var lists []int
	for i, t := range columns {
		if strings.HasSuffix(t.DatabaseTypeName(), "[]") {
			lists = append(lists, i)
		}
	}

	var hugeInts []int
	for i, t := range columns {
		if t.DatabaseTypeName() == "HUGEINT" {
			hugeInts = append(hugeInts, i)
		}
	}

	iter := &SqlRowIter{rows, columns, schema, buf, ptrs, decimals, lists, hugeInts}
	if logrus.GetLevel() >= logrus.DebugLevel {
		logrus.Debug("New " + iter.String())
	}
	return iter, nil
}

func (iter *SqlRowIter) String() string {
	return fmt.Sprintf("SqlRowIter: columns=[%s], schema=[%s]", formatColumnTypes(iter.columns), formatSchema(iter.schema))