IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2024-12-26 00:31:07 -06:00
Code Issues Packages Projects Releases Wiki Activity

add FromCSVTo Boxed|Primitive Float|Double Array functions

Browse Source
This commit is contained in:
Jonathan Shook 2024-07-02 19:12:49 -05:00
parent 9e3b6241f1
commit b7e0f35cd2
5 changed files with 202 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
nb-adapters/adapter-cqld4/src/main/java/io/nosqlbench/datamappers/functions/to_cqlvector/from_string/CSVToCqlFloatVector.java Normal file
View File

@ -0,0 +1,42 @@
package io.nosqlbench.datamappers.functions.to_cqlvector.from_string;
/*
* Copyright (c) 2022 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.
*/
import com.datastax.oss.driver.api.core.data.CqlVector;
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Function;
@ThreadSafeMapper
@Categories(Category.conversion)
public class CSVToCqlFloatVector implements Function<String, CqlVector<Float>> {
@Override
public CqlVector<Float> apply(String s) {
String[] split = s.split(",");
List<Float> floats = new ArrayList<>(split.length);
for (String string : split) {
floats.add(Float.parseFloat(string));
}
return com.datastax.oss.driver.api.core.data.CqlVector.newInstance(floats);
}
}

40
nb-virtdata/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/conversions/from_string/FromCSVToBoxedDoubleArray.java Normal file
View File

@ -0,0 +1,40 @@
package io.nosqlbench.virtdata.library.basics.shared.conversions.from_string;
/*
* Copyright (c) 2022 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.
*/
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import java.util.function.Function;
@ThreadSafeMapper
@Categories(Category.conversion)
public class FromCSVToBoxedDoubleArray implements Function<String,Double[]> {
@Override
public Double[] apply(String s) {
String[] split = s.split(",");
Double[] doubles = new Double[split.length];
for (int i = 0; i < split.length; i++) {
doubles[i] = Double.parseDouble(split[i]);
}
return doubles;
}
}

40
nb-virtdata/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/conversions/from_string/FromCSVToBoxedFloatArray.java Normal file
View File

@ -0,0 +1,40 @@
package io.nosqlbench.virtdata.library.basics.shared.conversions.from_string;
/*
* Copyright (c) 2022 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.
*/
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import java.util.function.Function;
@ThreadSafeMapper
@Categories(Category.conversion)
public class FromCSVToBoxedFloatArray implements Function<String,Float[]> {
@Override
public Float[] apply(String s) {
String[] split = s.split(",");
Float[] floats = new Float[split.length];
for (int i = 0; i < split.length; i++) {
floats[i] = Float.parseFloat(split[i]);
}
return floats;
}
}

40
nb-virtdata/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/conversions/from_string/FromCSVToPrimitiveDoubleArray.java Normal file
View File

@ -0,0 +1,40 @@
package io.nosqlbench.virtdata.library.basics.shared.conversions.from_string;
/*
* Copyright (c) 2022 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.
*/
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import java.util.function.Function;
@ThreadSafeMapper
@Categories(Category.conversion)
public class FromCSVToPrimitiveDoubleArray implements Function<String,double[]> {
@Override
public double[] apply(String s) {
String[] split = s.split(",");
double[] doubles = new double[split.length];
for (int i = 0; i < split.length; i++) {
doubles[i] = Double.parseDouble(split[i]);
}
return doubles;
}
}

40
nb-virtdata/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/conversions/from_string/FromCSVToPrimitiveFloatArray.java Normal file
View File

@ -0,0 +1,40 @@
package io.nosqlbench.virtdata.library.basics.shared.conversions.from_string;
/*
* Copyright (c) 2022 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.
*/
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import java.util.function.Function;
@ThreadSafeMapper
@Categories(Category.conversion)
public class FromCSVToPrimitiveFloatArray implements Function<String,float[]> {
@Override
public float[] apply(String s) {
String[] split = s.split(",");
float[] floats = new float[split.length];
for (int i = 0; i < split.length; i++) {
floats[i] = Float.parseFloat(split[i]);
}
return floats;
}
}