IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

pull sequence type up

Browse Source
This commit is contained in:
Jonathan Shook 2020-09-10 09:38:33 -05:00
parent 5f5c08d77d
commit ab5538d1c2
2 changed files with 51 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
engine-api/src/main/java/io/nosqlbench/engine/api/activityapi/planning/Sequence.java Normal file
View File

@ -0,0 +1,49 @@
package io.nosqlbench.engine.api.activityapi.planning;
import java.util.Arrays;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;
public class Sequence<T> implements OpSequence<T> {
private final SequencerType type;
private final List<T> elems;
private final int[] seq;
Sequence(SequencerType type, List<T> elems, int[] seq) {
this.type = type;
this.elems = elems;
this.seq = seq;
}
@Override
public T get(long selector) {
int index = (int) (selector % seq.length);
index = seq[index];
return elems.get(index);
}
@Override
public List<T> getOps() {
return elems;
}
@Override
public int[] getSequence() {
return seq;
}
public SequencerType getSequencerType() {
return type;
}
@Override
public <U> Sequence<U> transform(Function<T, U> func) {
return new Sequence<U>(type, elems.stream().map(func).collect(Collectors.toList()), seq);
}
@Override
public String toString() {
return Arrays.toString(this.seq);
}
}

45
engine-api/src/main/java/io/nosqlbench/engine/api/activityapi/planning/SequencePlanner.java
View File

@ -22,15 +22,13 @@ import org.slf4j.LoggerFactory;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.function.Function;
import java.util.function.ToLongFunction; import java.util.function.ToLongFunction;
import java.util.stream.Collectors;
public class SequencePlanner<T> { public class SequencePlanner<T> {
private final static Logger logger = LoggerFactory.getLogger(SequencePlanner.class); private final static Logger logger = LoggerFactory.getLogger(SequencePlanner.class);
private SequencerType sequencerType; private final SequencerType sequencerType;
private List<T> elements = new ArrayList<>(); private List<T> elements = new ArrayList<>();
private List<Long> ratios = new ArrayList<>(); private final List<Long> ratios = new ArrayList<>();
private int[] elementIndex; private int[] elementIndex;
public SequencePlanner(SequencerType sequencerType) { public SequencePlanner(SequencerType sequencerType) {
@ -65,43 +63,4 @@ public class SequencePlanner<T> {
return new Sequence<>(sequencerType, elements, elementIndex); return new Sequence<>(sequencerType, elements, elementIndex);
} }
public static class Sequence<T> implements OpSequence<T> {
private final SequencerType type;
private final List<T> elems;
private final int[] seq;
Sequence(SequencerType type, List<T> elems, int[] seq) {
this.type = type;
this.elems = elems;
this.seq = seq;
}
@Override
public T get(long selector) {
int index = (int) (selector % seq.length);
index = seq[index];
return elems.get(index);
}
@Override
public List<T> getOps() {
return elems;
}
@Override
public int[] getSequence() {
return seq;
}
public SequencerType getSequencerType() {
return type;
}
@Override
public <U> Sequence<U> transform(Function<T,U> func) {
return new Sequence<U>(type, elems.stream().map(func).collect(Collectors.toList()), seq);
}
}
} }