more higher order functions

This commit is contained in:
Jonathan Shook 2023-05-18 14:58:28 -05:00
parent 963c481b26
commit e936e35f0d
2 changed files with 83 additions and 0 deletions

View File

@ -0,0 +1,47 @@
/*
* Copyright (c) 2022-2023 nosqlbench
*
* 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 io.nosqlbench.virtdata.library.basics.shared.from_double.to_double;
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import io.nosqlbench.virtdata.api.bindings.VirtDataConversions;
import java.util.List;
import java.util.function.LongToDoubleFunction;
/**
* Compute the sum of a set of functions.
*/
@ThreadSafeMapper
@Categories({Category.general})
public class SumFunctions implements LongToDoubleFunction {
private final List<LongToDoubleFunction> functions;
public SumFunctions(Object... funcs) {
this.functions = VirtDataConversions.adaptFunctionList(funcs, LongToDoubleFunction.class);
}
@Override
public double applyAsDouble(long value) {
double sum = 0.0d;
for (LongToDoubleFunction function : functions) {
sum+=function.applyAsDouble(value);
}
return sum;
}
}

View File

@ -0,0 +1,36 @@
/*
* Copyright (c) 2023 nosqlbench
*
* 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 io.nosqlbench.virtdata.library.basics.shared.from_double.to_double;
import org.assertj.core.data.Offset;
import org.junit.jupiter.api.Test;
import java.util.function.LongToDoubleFunction;
import static org.assertj.core.api.Assertions.assertThat;
public class SumFunctionsTest {
@Test
public void sumFunctionsTest() {
LongToDoubleFunction f1 = d -> d*3.0d;
LongToDoubleFunction f2 = d -> d+5.0d;
SumFunctions ff = new SumFunctions(f1,f2);
assertThat(ff.applyAsDouble(15L)).isEqualTo(65.0d, Offset.offset(0.0002d));
}
}