io.github.shanqiang.window.TumbleWindow Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jstream Show documentation
The newest version!
package io.github.shanqiang.window;

import io.github.shanqiang.function.AggTimeWindowFunction;
import io.github.shanqiang.function.ReduceTimeWindowFunction;
import io.github.shanqiang.function.TimeWindowFunction;
import io.github.shanqiang.table.Table;

import java.time.Duration;

public class TumbleWindow {
    private final SlideWindow slideWindow;

    public TumbleWindow(Duration windowSize,
                     String[] partitionByColumnNames,
                     String timeColumnName,
                     ReduceTimeWindowFunction reduceTimeWindowFunction,
                     String... columnNames) {
        this.slideWindow = new SlideWindow(windowSize,
                windowSize,
                partitionByColumnNames,
                timeColumnName,
                reduceTimeWindowFunction,
                columnNames);
    }

    public TumbleWindow(Duration windowSize,
                        String[] partitionByColumnNames,
                        String timeColumnName,
                        AggTimeWindowFunction aggTimeWindowFunction,
                        String... columnNames) {
        this.slideWindow = new SlideWindow(windowSize,
                windowSize,
                partitionByColumnNames,
                timeColumnName,
                aggTimeWindowFunction,
                columnNames);
    }

    public TumbleWindow(Duration windowSize,
                     String[] partitionByColumnNames,
                     String timeColumnName,
                     TimeWindowFunction windowFunction,
                     String... addedColumnNames) {
        this.slideWindow = new SlideWindow(windowSize,
                windowSize,
                partitionByColumnNames,
                timeColumnName,
                windowFunction,
                addedColumnNames);
    }

    public void setNoDataDelay(Duration noDataDelay) {
        slideWindow.setNoDataDelay(noDataDelay);
    }

    public void setWatermark(Duration watermark) {
        slideWindow.setWatermark(watermark);
    }

    public Table tumble(Table hashed) {
        return slideWindow.slide(hashed);
    }
}