io.github.icodegarden.nutrient.lang.util.ReactiveUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nutrient-lang Show documentation

The newest version!

package io.github.icodegarden.nutrient.lang.util;

import java.util.concurrent.atomic.AtomicReference;
import java.util.concurrent.locks.LockSupport;

import io.github.icodegarden.nutrient.lang.exception.reactive.ReactiveBlockException;
import reactor.core.publisher.Mono;
import reactor.core.scheduler.Schedulers;

/**
 * 
 * @author Fangfang.Xu
 *
 */
public abstract class ReactiveUtils {

	public static  T block(Mono mono) throws ReactiveBlockException {
		return block(mono, null);
	}

	public static  T block(Mono mono, Long blockTimeoutMs) throws ReactiveBlockException {
		AtomicReference referenceE = new AtomicReference<>();
		AtomicReference reference = new AtomicReference<>();

		Thread thread = Thread.currentThread();

		mono.doOnError(e -> {
			referenceE.set(e);

			if (blockTimeoutMs != null) {
				LockSupport.unpark(thread);
			}
		})//
				.subscribeOn(Schedulers.immediate())//
				.subscribe(next -> {
					reference.set(next);

					if (blockTimeoutMs != null) {
						LockSupport.unpark(thread);
					}
				});

		if (blockTimeoutMs != null) {
			LockSupport.parkUntil(SystemUtils.currentTimeMillis() + blockTimeoutMs);
		}

		if (referenceE.get() != null) {
			throw new ReactiveBlockException(referenceE.get());
		}
		return reference.get();
	}
}