net.officefloor.spring.reactive.ReactiveWoof Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of officespring_webclient Show documentation

OfficeFloor plug-in for Spring WebClient

There is a newer version: 3.40.0

Show newest version

/*-
 * #%L
 * Spring WebClient
 * %%
 * Copyright (C) 2005 - 2020 Daniel Sagenschneider
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see .
 * #L%
 */

package net.officefloor.spring.reactive;

import java.util.function.Consumer;

import org.springframework.web.reactive.function.client.WebClientResponseException;

import net.officefloor.frame.api.function.AsynchronousFlow;
import net.officefloor.server.http.HttpException;
import net.officefloor.server.http.HttpStatus;
import net.officefloor.web.ObjectResponse;
import net.officefloor.woof.WoOF;

/**
 * Utility methods for {@link WoOF} working with Spring Reactive.
 * 
 * @author Daniel Sagenschneider
 */
public class ReactiveWoof {

	/**
	 * Provides handling for on error.
	 * 
	 * @param flow {@link AsynchronousFlow}.
	 * @return {@link Consumer} for error handling.
	 */
	public static Consumer flowError(AsynchronousFlow flow) {
		return (error) -> flow.complete(() -> {
			throw error;
		});
	};

	/**
	 * Provides handling of an error and propagates
	 * 
	 * @param flow {@link AsynchronousFlow}.
	 * @return {@link Consumer} for error handling.
	 */
	public static Consumer propagateHttpError(AsynchronousFlow flow) {
		return (error) -> flow.complete(() -> {

			// Attempt to translate to HTTP exception
			if (error instanceof WebClientResponseException) {
				WebClientResponseException responseException = (WebClientResponseException) error;
				throw new HttpException(HttpStatus.getHttpStatus(responseException.getStatusCode().value()),
						responseException);
			}

			// Propagate the raw error
			throw error;
		});
	}

	/**
	 * Sends the response.
	 * 
	 * @param flow     {@link AsynchronousFlow}.
	 * @param response {@link ObjectResponse}.
	 * @return {@link Consumer} for sending the response.
	 */
	public static  Consumer send(AsynchronousFlow flow, ObjectResponse response) {
		return (payload) -> flow.complete(() -> response.send(payload));
	}

}