loki/pkg/dataobj/planner/schema/column.go

// Package schema provides types and utilities for working with dataset schemas
package schema

import (
	"github.com/grafana/loki/v3/pkg/dataobj/internal/metadata/datasetmd"
)

// ColumnSchema describes a column in a dataset, including its name and data type.
type ColumnSchema struct {
	// Name is the identifier of the column
	Name string
	// Type represents the data type of values in this column
	Type datasetmd.ValueType
}

// Schema represents the complete structure of a dataset, containing all column definitions.
type Schema struct {
	// Columns is a slice of ColumnSchema defining all columns in the dataset
	Columns []ColumnSchema
}

// Filter returns a new Schema with only the columns specified in projection
func (s *Schema) Filter(projection []string) Schema {
	filteredColumns := make([]ColumnSchema, 0, len(projection))
	for _, col := range s.Columns {
		for _, pcol := range projection {
			if col.Name == pcol {
				filteredColumns = append(filteredColumns, col)
			}
		}
	}
	return Schema{
		Columns: filteredColumns,
	}
}

func FromColumns(columns []ColumnSchema) Schema {
	return Schema{
		Columns: columns,
	}
}