Allow to chain ContextConsumer implementations

Closes gh-26723
This commit is contained in:
Stephane Nicoll
2021-06-01 14:04:35 +02:00
parent 79b0da555a
commit 9e46061aa6
2 changed files with 95 additions and 1 deletions

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2012-2019 the original author or authors.
* Copyright 2012-2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -17,6 +17,7 @@
package org.springframework.boot.test.context.runner;
import org.springframework.context.ApplicationContext;
import org.springframework.util.Assert;
/**
* Callback interface used to process an {@link ApplicationContext} with the ability to
@@ -38,4 +39,20 @@ public interface ContextConsumer<C extends ApplicationContext> {
*/
void accept(C context) throws Throwable;
/**
* Returns a composed {@code ContextConsumer} that performs, in sequence, this
* operation followed by the {@code after} operation.
* @param after the operation to perform after this operation
* @return a composed {@code ContextConsumer} that performs in sequence this operation
* followed by the {@code after} operation
* @since 2.6.0
*/
default ContextConsumer<C> andThen(ContextConsumer<? super C> after) {
Assert.notNull(after, "After must not be null");
return (context) -> {
accept(context);
after.accept(context);
};
}
}