net.dv8tion.jda.internal.requests.restaction.operator.FlatMapRestAction Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of JDA Show documentation
Show all versions of JDA Show documentation
Java wrapper for the popular chat & VOIP service: Discord https://discord.com
/*
* Copyright 2015 Austin Keener, Michael Ritter, Florian Spieß, and the JDA contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package net.dv8tion.jda.internal.requests.restaction.operator;
import net.dv8tion.jda.api.exceptions.RateLimitedException;
import net.dv8tion.jda.api.requests.RestAction;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.concurrent.CancellationException;
import java.util.concurrent.CompletableFuture;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Predicate;
public class FlatMapRestAction extends RestActionOperator
{
private final Function> function;
private final Predicate condition;
public FlatMapRestAction(RestAction action, Predicate condition,
Function> function)
{
super(action);
this.function = function;
this.condition = condition;
}
private RestAction supply(I input)
{
return applyContext(function.apply(input));
}
@Override
public void queue(@Nullable Consumer success, @Nullable Consumer failure)
{
Consumer catcher = contextWrap(failure);
handle(action, catcher, (result) -> {
if (condition != null && !condition.test(result))
return;
RestAction then = supply(result);
if (then == null) // caught by handle try/catch abstraction
throw new IllegalStateException("FlatMap operand is null");
then.queue(success, catcher);
});
}
@Override
public O complete(boolean shouldQueue) throws RateLimitedException
{
I complete = action.complete(shouldQueue);
if (this.condition != null && !this.condition.test(complete))
{
throw new CancellationException("FlatMap condition failed");
}
return supply(complete).complete(shouldQueue);
}
@Nonnull
@Override
public CompletableFuture submit(boolean shouldQueue)
{
return action.submit(shouldQueue)
.thenCompose((result) ->
{
if (condition != null && !condition.test(result))
{
CompletableFuture future = new CompletableFuture<>();
future.cancel(true);
return future;
}
return supply(result).submit(shouldQueue);
});
}
}