ompute.logical-plan.0.2.1.source-code.Aggregate.kt Maven / Gradle / Ivy

Go to download

package org.ballistacompute.logical

import org.ballistacompute.datatypes.Schema

/**
 * Logical plan representing an aggregate query against an input.
 */
class Aggregate(val input: LogicalPlan,
                val groupExpr: List,
                val aggregateExpr: List) : LogicalPlan {

    override fun schema(): Schema {
        return Schema(
                groupExpr.map { it.toField(input) } +
                aggregateExpr.map { it.toField(input) }
        )
    }

    override fun children(): List {
        return listOf(input)
    }

    override fun toString(): String {
        return "Aggregate: groupExpr=$groupExpr, aggregateExpr=$aggregateExpr"
    }
}