integrations

package

v0.1.1 Latest Latest Go to latest Published: Aug 31, 2024 License: MIT Imports: 11 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type DuckDBExtension struct {
	Name          string
	LoadByDefault bool
}

DuckDBExtension represents a DuckDB extension with its name and load preference.

func DefaultExtensions() []DuckDBExtension

DefaultExtensions returns the default extensions to be loaded in DuckDB.

type DuckDBReadOptions struct {
	Extensions []DuckDBExtension
	Query      string
}

DuckDBReadOptions defines options for reading from DuckDB.

type DuckDBReader struct {
	// contains filtered or unexported fields
}

DuckDBReader reads records from DuckDB and implements the Reader interface.

func NewDuckDBReader(ctx context.Context, dbURL string, opts *DuckDBReadOptions) (*DuckDBReader, error)

NewDuckDBReader creates a new DuckDB reader.

func (d *DuckDBReader) Close() error

Close releases resources associated with the DuckDB reader.

func (d *DuckDBReader) Read() (arrow.Record, error)

Read reads the next record from DuckDB.

func (r *DuckDBReader) RunSQL(sql string) ([]arrow.Record, error)

RunSQL runs a SQL query on DuckDB and returns the results as Arrow records.

func (d *DuckDBReader) Schema() *arrow.Schema

Schema returns the schema of the records being read from DuckDB.

type DuckDBWriter struct {
	// contains filtered or unexported fields
}

DuckDBWriter writes records to DuckDB and implements the Writer interface.

func NewDuckDBWriter(ctx context.Context, dbURL string, tableName string, extensions []DuckDBExtension) (*DuckDBWriter, error)

NewDuckDBWriter creates a new DuckDB writer.

func (w *DuckDBWriter) Close() error

Close closes the DuckDB writer and releases resources.

func (w *DuckDBWriter) Write(record arrow.Record) error

Write writes a record to DuckDB.