graph

type Definitions struct {
	FeatureSets     []*FeatureSet
	StreamResolvers []*StreamResolver
}

type FeatureBuilder interface {
	// contains filtered or unexported methods
}

type FeatureSet struct {
	// metadata fields
	Name               string
	Features           []*graphv1.FeatureType
	IsSingleton        bool
	Tags               []string
	Owner              string
	Doc                string
	EtlOfflineToOnline bool
	MaxStaleness       time.Duration
	// contains filtered or unexported fields
}

type FeatureTimeBuilder struct{}

type Features map[string]FeatureBuilder

type HasManyFeatureBuilder struct {
	ForeignNamespace string
	MaxStaleness     time.Duration
}

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

type MaterializationOptions struct {
	// map window -> bucket duration
	// NOTE: OPPOSITE ORDER AS chalkpy if construction is being done via struct
	BucketDurations       map[time.Duration]time.Duration
	DefaultBucketDuration time.Duration
	// The period for which to use the continuous resolver, instead
	// of relying upon the last backfill. If not provided, and a continuous
	// resolver is provided, this will be set to backfill_lookback_duration.
	ContinuousBufferDuration time.Duration
	// A crontab or duration string to specify the schedule for back filling the
	// materialized aggregate.
	BackfillSchedule string
	// The lower bound of the first bucket. All buckets are aligned to this time.
	BucketStart time.Time

	// technically not part of materialization kwarg, but still useful to expose (for now)
	// The 'k' arg of approx_top_k.
	ApproxTopKArgK int64
	// The resolver to use for back-filling the materialized aggregate.
	// If not provided, the data will be back filled using the resolver
	// that would run for an offline query.
	BackfillResolver string
	// The amount of time before the start of the previous backfill
	// to consider when running the backfill resolver. Set this parameter
	// to the be equal to the latest arriving data in the backfill window.
	BackfillLookbackDuration time.Duration
	// The time at which to start back filling the materialized aggregate.
	// If not provided, the backfill consider the earliest available data returned
	// by the `backfill_resolver`.
	BackfillStartTime time.Time
	// The resolver to use for continuous updates to the materialized aggregate.
	// If not provided, the data will be updated using the resolver that would run
	// for an online query.
	ContinuousResolver string
}

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

type StreamListMessage struct {
	ValueType StreamStructMessage
}

type StreamMessageType interface {
	ToArrowType() *arrowv1.ArrowType
	// contains filtered or unexported methods
}

type StreamResolver struct {
	Name  string
	Owner string
	Doc   string

	// must match up to a UI-defined Data Source
	StreamSourceName string
	// "kafka" or "kinesis" or "pubsub"
	StreamSourceType string

	OutputFeatureSet string
	OutputFeatures   map[string]expr.Expr
	// defines the structure of message
	// should be a mapping from JSON key name to value type ("int" or "float" or "bool" or "str")
	MessageType StreamMessageType
	// if provided, only run on this machine type (defaults to all)
	MachineType string
	// if provided, only run in these environments (defaults to all)
	RunInEnvironments []string
	// if provided, evaluate this expression against incoming messages to either parse or filter
	Parse expr.Expr
}

type StreamStructMessage struct {
	Fields map[string]string
}

type WindowedFeatureBuilder struct {
	Default      expr.Expr
	Expression   expr.Expr
	MaxStaleness time.Duration

	Materialization MaterializationOptions
	// contains filtered or unexported fields
}