如何doOnNext只消耗PublishSubject的前n项？

PublishSubject<Integer> subject = PublishSubject.create();

subject
    .doOnNext(data -> {
        // 在消耗数据之前执行的操作
        System.out.println("do something before consuming data: " + data);
    })
    .take(n) // 只消耗前n项数据
    .subscribe(data -> {
        // 订阅者接收数据的逻辑
        System.out.println("consume data: " + data);
    });

// 发送数据给PublishSubject
for (int i = 1; i <= 10; i++) {
    subject.onNext(i);
}