import java.util.*;
import java.util.function.Function;
import java.util.function.Predicate;

public class SimpleStream<T> {
    private final List<T> source;  // 原始數(shù)據(jù)
    private final List<Function<List<T>, List<T>>> operations = new ArrayList<>(); // 用于存儲(chǔ)操作鏈

    public SimpleStream(List<T> source) {
        this.source = source;
    }

    // 中間操作：filter
    public SimpleStream<T> filter(Predicate<T> predicate) {
        operations.add(list -> {
            List<T> result = new ArrayList<>();
            for (T item : list) {
                if (predicate.test(item)) {
                    result.add(item);
                }
            }
            return result;
        });
        return this;
    }

    // 中間操作：map
    public <R> SimpleStream<R> map(Function<T, R> mapper) {
        // 通過 Function 將當(dāng)前 SimpleStream 轉(zhuǎn)換為新類型的 SimpleStream
        SimpleStream<R> newStream = new SimpleStream<>(null);
        operations.add(list -> {
            List<R> result = new ArrayList<>();
            for (T item : list) {
                result.add(mapper.apply(item));
            }
            return result;
        });
        newStream.operations.addAll(this.operations); // 繼承父級(jí)操作鏈
        return newStream;
    }

    // 終端操作：forEach，觸發(fā)延遲執(zhí)行
    public void forEach(java.util.function.Consumer<T> consumer) {
        List<T> result = applyOperations(source); // 在終端操作時(shí)應(yīng)用所有中間操作
        for (T item : result) {
            consumer.accept(item);
        }
    }

    // 終端操作：collect
    public List<T> collect() {
        return applyOperations(source); // 在終端操作時(shí)應(yīng)用所有中間操作
    }

    // 應(yīng)用所有操作
    private List<T> applyOperations(List<T> input) {
        List<T> result = input;
        for (Function<List<T>, List<T>> operation : operations) {
            result = operation.apply(result); // 按順序應(yīng)用所有的操作
        }
        return result;
    }

    // 工廠方法：創(chuàng)建 SimpleStream
    public static <T> SimpleStream<T> of(List<T> source) {
        return new SimpleStream<>(source);
    }

    public static void main(String[] args) {
        // 示例使用
        List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);

        SimpleStream.of(numbers)
                .filter(x -> x > 2) // 延遲執(zhí)行的過濾操作
                .map(x -> x * 2)    // 延遲執(zhí)行的映射操作
                .forEach(System.out::println); // 終端操作觸發(fā)執(zhí)行
    }
}