Rxjava背压机制应用于大批量定时任务的Demo

Rxjava背压机制应用于大批量定时任务的Demo

package com.yanbing.demohost.hostdemo;

import io.reactivex.Completable;
import io.reactivex.Emitter;
import io.reactivex.Flowable;
import io.reactivex.Scheduler;
import io.reactivex.functions.Action;
import io.reactivex.functions.Consumer;
import io.reactivex.schedulers.Schedulers;
import org.reactivestreams.Subscriber;
import org.reactivestreams.Subscription;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 项目: host-demo
 *
 * @author 岩冰
 * @version 2020/8/22 9:33
 */
public class RxjavaDemo {
    public static void main(String[] args) throws InterruptedException {
        AtomicInteger aaa = new AtomicInteger(1);
        ExecutorService executor = Executors.newFixedThreadPool(10);
        Scheduler from = Schedulers.from(executor);
        Flowable.generate((Consumer<Emitter<Integer>>) emitter -> {
            int andIncrement = aaa.getAndIncrement();
            if (andIncrement < 100) {
                System.out.println(Thread.currentThread().getName() + "下发");
                emitter.onNext(andIncrement);
            } else {
                emitter.onComplete();
            }
        })
                .observeOn(from)//上游io线程
                .subscribeOn(Schedulers.from(executor))
                .subscribe(new DemoSubscriber(1, executor));
        TimeUnit.HOURS.sleep(1);
        System.out.println("-------");
    }

    public static class DemoSubscriber implements Subscriber<Integer> {
        private int index = 0;
        private ExecutorService executor;

        public DemoSubscriber(int index, ExecutorService executor) {
            this.index = index;
            this.executor = executor;
        }

        private Subscription subscription;

        @Override
        public void onSubscribe(Subscription subscription) {
            this.subscription = subscription;
            subscription.request(10);
        }

        @Override
        public void onNext(Integer integer) {
            Completable.fromAction(() -> {
                TimeUnit.SECONDS.sleep(1);
                System.out.println(integer + "--" + Thread.currentThread().getName() + "bean num" + index);
            }).subscribeOn(Schedulers.from(executor))
                    .subscribe(new Action() {
                        @Override
                        public void run() throws Exception {
                            subscription.request(1);
                        }
                    });
        }

        @Override
        public void onError(Throwable throwable) {

        }

        @Override
        public void onComplete() {
            System.out.println("完成");
        }
    }
}