Method setAutoIncrementValue

catalog/table.go:871–931

setAutoIncrementValue is a helper function to update the sequence value

(ctx *sql.Context, value uint64)

	}

	return val, nil
}

// GetNextAutoIncrementValue implements sql.AutoIncrementTable.
func (t *Table) GetNextAutoIncrementValue(ctx *sql.Context, insertVal any) (uint64, error) {
	t.mu.Lock()
	defer t.mu.Unlock()

	if t.comment.Meta.Sequence == "" {
		return 0, sql.ErrNoAutoIncrementCol
	}

	nextVal, err := t.getNextAutoIncrementValue(ctx)
	if err != nil {
		return 0, err
	}

	// If insertVal is provided and greater than the next sequence value, update sequence
	if insertVal != nil {
		var start uint64
		switch v := insertVal.(type) {
		case uint64:
			start = v
		case int64:
			if v > 0 {
				start = uint64(v)
			}
		}
		if start > 0 && start > nextVal {
			err := t.setAutoIncrementValue(ctx, start)
			if err != nil {
				return 0, err
			}
			return start, nil
		}
	}

	// Get next value from sequence
	var val uint64
	err = adapter.QueryRowCatalog(ctx, `SELECT nextval('`+t.comment.Meta.Sequence+`')`).Scan(&val)
	if err != nil {
		return 0, ErrDuckDB.New(err)
	}

	return val, nil
}

// AutoIncrementSetter implements sql.AutoIncrementTable.
func (t *Table) AutoIncrementSetter(ctx *sql.Context) sql.AutoIncrementSetter {
	if t.comment.Meta.Sequence == "" {
		return nil
	}
	return &autoIncrementSetter{t: t}
}

// setAutoIncrementValue is a helper function to update the sequence value
func (t *Table) setAutoIncrementValue(ctx *sql.Context, value uint64) error {
	// DuckDB does not support setting the sequence value directly,