sql

type BinaryType struct {
}

type BooleanType struct {
}

type ByteType struct {
}

type DataFrame interface {
	// Show prints out data frame data.
	Show(numRows int, truncate bool) error
	// Schema returns the schema for the current data frame.
	Schema() (*StructType, error)
	// Collect returns the data rows of the current data frame.
	Collect() ([]Row, error)
	// Write returns a data frame writer, which could be used to save data frame to supported storage.
	Write() DataFrameWriter
	// CreateTempView creates or replaces a temporary view.
	CreateTempView(viewName string, replace bool, global bool) error
	// Repartition re-partitions a data frame.
	Repartition(numPartitions int, columns []string) (DataFrame, error)
	// RepartitionByRange re-partitions a data frame by range partition.
	RepartitionByRange(numPartitions int, columns []RangePartitionColumn) (DataFrame, error)
}

type DataFrameReader interface {
	// Format specifies data format (data source type) for the underlying data, e.g. parquet.
	Format(source string) DataFrameReader
	// Load reads the underlying data and returns a data frame.
	Load(path string) (DataFrame, error)
}

type DataFrameWriter interface {
	// Mode specifies saving mode for the data, e.g. Append, Overwrite, ErrorIfExists.
	Mode(saveMode string) DataFrameWriter
	// Format specifies data format (data source type) for the underlying data, e.g. parquet.
	Format(source string) DataFrameWriter
	// Save writes data frame to the given path.
	Save(path string) error
}

type DataType interface {
	TypeName() string
}

type DateType struct {
}

type DecimalType struct {
}

type DoubleType struct {
}

type FloatType struct {
}

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

type IntegerType struct {
}

type LongType struct {
}

type RangePartitionColumn struct {
	Name       string
	Descending bool
}

type Row interface {
	Schema() (*StructType, error)
	Values() ([]any, error)
}

type ShortType struct {
}

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

type StringType struct {
}

type StructField struct {
	Name     string
	DataType DataType
	Nullable bool // default should be true
}

type StructType struct {
	TypeName string
	Fields   []StructField
}

type TimestampNtzType struct {
}

type TimestampType struct {
}

type UnsupportedType struct {
	TypeInfo any
}