Method GetNextAutoIncrementValue

catalog/table.go:819–860

GetNextAutoIncrementValue implements sql.AutoIncrementTable.

(ctx *sql.Context, insertVal any)

		columnInfo := &ColumnInfo{
			ColumnName:    columnName,
			ColumnIndex:   columnIndex,
			DataType:      dataType,
			IsNullable:    isNullable,
			ColumnDefault: columnDefault,
			Comment:       comment,
		}
		columns = append(columns, columnInfo)
	}

	if err = rows.Err(); err != nil {
		return nil, err
	}

	return columns, nil
}

func (t *IndexedTable) LookupPartitions(ctx *sql.Context, lookup sql.IndexLookup) (sql.PartitionIter, error) {
	return nil, fmt.Errorf("unimplemented(LookupPartitions) (table: %s, query: %s)", t.name, ctx.Query())
}

// PeekNextAutoIncrementValue implements sql.AutoIncrementTable.
func (t *Table) PeekNextAutoIncrementValue(ctx *sql.Context) (uint64, error) {
	t.mu.RLock()
	defer t.mu.RUnlock()

	if t.comment.Meta.Sequence == "" {
		return 0, sql.ErrNoAutoIncrementCol
	}
	return t.getNextAutoIncrementValue(ctx)
}

func (t *Table) getNextAutoIncrementValue(ctx *sql.Context) (uint64, error) {
	// For PeekNextAutoIncrementValue, we want to see what the next value would be
	// without actually incrementing. We can do this by getting currval + 1.
	var val uint64
	err := adapter.QueryRowCatalog(ctx, `SELECT currval('`+t.comment.Meta.Sequence+`') + 1`).Scan(&val)
	if err != nil {
		// https://duckdb.org/docs/sql/statements/create_sequence.html#selecting-the-current-value
		// > Note that the nextval function must have already been called before calling currval,
		// > otherwise a Serialization Error (sequence is not yet defined in this session) will be thrown.
		if !strings.Contains(err.Error(), "sequence is not yet defined in this session") {
			return 0, ErrDuckDB.New(err)
		}
		// If the sequence has not been used yet, we can get the start value from the sequence.