File tree Expand file tree Collapse file tree 1 file changed +10
-12
lines changed
core/src/test/java/com/scalar/db/storage/cassandra Expand file tree Collapse file tree 1 file changed +10
-12
lines changed Original file line number Diff line number Diff line change 2222import com .scalar .db .exception .storage .ExecutionException ;
2323import com .scalar .db .exception .storage .NoMutationException ;
2424import com .scalar .db .io .Key ;
25- import java .util .ArrayList ;
2625import java .util .Arrays ;
2726import java .util .List ;
27+ import java .util .stream .Collectors ;
2828import org .junit .jupiter .api .BeforeEach ;
2929import org .junit .jupiter .api .Test ;
3030import org .mockito .Mock ;
@@ -102,17 +102,15 @@ private List<Mutation> prepareNonConditionalPuts() {
102102 }
103103
104104 private List <Mutation > prepareConditionalPuts () {
105- List <Mutation > mutations = prepareNonConditionalPuts ();
106- List <Mutation > conditionalMutations = new ArrayList <>();
107- for (Mutation m : mutations ) {
108- if (m instanceof Put ) {
109- conditionalMutations .add (
110- Put .newBuilder ((Put ) m ).condition (ConditionBuilder .putIfNotExists ()).build ());
111- } else {
112- conditionalMutations .add (m );
113- }
114- }
115- return conditionalMutations ;
105+ return prepareNonConditionalPuts ().stream ()
106+ .map (
107+ m -> {
108+ if (m instanceof Put ) {
109+ return Put .newBuilder ((Put ) m ).condition (ConditionBuilder .putIfNotExists ()).build ();
110+ }
111+ return m ;
112+ })
113+ .collect (Collectors .toList ());
116114 }
117115
118116 private BatchHandler prepareSpiedBatchHandler () {
You can’t perform that action at this time.
0 commit comments