use of com.cinchapi.concourse.server.storage.Store in project concourse by cinchapi.
the class ConcourseServer method selectKeysCclTimeOrder.
@Override
@TranslateClientExceptions
@VerifyAccessToken
@VerifyReadPermission
public Map<Long, Map<String, Set<TObject>>> selectKeysCclTimeOrder(List<String> keys, String ccl, long timestamp, TOrder order, AccessToken creds, TransactionToken transaction, String environment) throws TException {
try {
AbstractSyntaxTree ast = compiler.parse(ccl);
AtomicSupport store = getStore(transaction, environment);
SortableTable<Set<TObject>> result = emptySortableResultDataset();
AtomicOperations.executeWithRetry(store, atomic -> Operations.selectKeysAstAtomic(keys, ast, timestamp, result, null, $result -> $result.sort(Sorting.byValues(Orders.from(order), atomic), timestamp), atomic));
return result;
} catch (Exception e) {
throw new ParseException(e.getMessage());
}
}
use of com.cinchapi.concourse.server.storage.Store in project concourse by cinchapi.
the class Stores method serialSelect.
/**
* Select the values stored for {@code key} in {@code record} at
* {@code timestamp} from the {@code store}.
* <p>
* If the {@code key} is primitive, the store retrieval is usually a simple
* {@link Store#select(long) select}. However, if the key is a navigation
* key, this method will process it via
* {@link Operations#traverseKeyRecordOptionalAtomic(String, long, long, AtomicOperation)}
* if the {@code store} is an {@link AtomicOperation} or
* {@link AtomicSupport supports} starting one.
* </p>
*
* @param store
* @param key
* @param record
* @param timestamp
*
* @return the values stored for {@code key} in {@code record} at
* {@code timestamp} according to the {@code store}
*/
protected static Set<TObject> serialSelect(Store store, String key, long record, long timestamp) {
Function evalFunc;
if (Keys.isNavigationKey(key)) {
if (store instanceof AtomicOperation || timestamp != Time.NONE) {
return Operations.traverseKeyRecordOptionalAtomic(key, record, timestamp, store);
} else if (store instanceof AtomicSupport) {
AtomicReference<Set<TObject>> value = new AtomicReference<>(ImmutableSet.of());
AtomicOperations.executeWithRetry((AtomicSupport) store, (atomic) -> {
value.set(Operations.traverseKeyRecordOptionalAtomic(key, record, timestamp, atomic));
});
return value.get();
} else {
throw new UnsupportedOperationException("Cannot fetch the current values of a navigation key using a Store that does not support atomic operations");
}
} else if ((evalFunc = Keys.tryParseFunction(key)) != null) {
String method = Calculations.alias(evalFunc.operation()) + "KeyRecordAtomic";
Number value = Reflection.callStatic(Operations.class, method, evalFunc.key(), record, timestamp, store);
return value != null ? ImmutableSet.of(Convert.javaToThrift(value)) : ImmutableSet.of();
} else if (key.equals(Constants.JSON_RESERVED_IDENTIFIER_NAME)) {
return ImmutableSet.of(Convert.javaToThrift(record));
} else {
Source source;
if (Command.isSet()) {
Strategy strategy = new Strategy(Command.current(), store);
source = strategy.source(key, record);
} else {
source = Source.FIELD;
}
Set<TObject> values;
if (source == Source.RECORD) {
// @formatter:off
Map<String, Set<TObject>> data = timestamp == Time.NONE ? store.select(record) : store.select(record, timestamp);
values = data.getOrDefault(key, ImmutableSet.of());
// @formatter:on
} else if (source == Source.FIELD) {
// @formatter:off
values = timestamp == Time.NONE ? store.select(key, record) : store.select(key, record, timestamp);
// @formatter:on
} else {
// source == Source.INDEX
// @formatter:off
values = timestamp == Time.NONE ? store.gather(key, record) : store.gather(key, record, timestamp);
// @formatter:on
}
return values;
}
}
use of com.cinchapi.concourse.server.storage.Store in project concourse by cinchapi.
the class Stores method find.
/**
* Find the records that contain values that are stored for {@code key} and
* satisfy {@code operator} in relation to the specified {@code values} at
* {@code timestamp}.
* <p>
* If the {@code key} is primitive, the store lookup is usually a simple
* {@link Store#find(String, Operator, TObject[]) find}. However, if the key
* is a navigation key, this method will process it by
* {@link #browse(Store, String) browsing} the destination values and
* checking the operator validity of each if and only if the {@code store}
* is an {@link AtomicOperation} or {@link AtomicSupport supports} starting
* one.
* </p>
*
* @param store
* @param timestamp
* @param key
* @param operator
* @param values
* @return the records that satisfy the query
*/
public static Set<Long> find(Store store, long timestamp, String key, Operator operator, TObject... values) {
for (int i = 0; i < values.length; ++i) {
TObject value = values[i];
if (value.getType() == Type.FUNCTION) {
Function function = (Function) Convert.thriftToJava(value);
TemporalFunction func = (TemporalFunction) function;
String method = Calculations.alias(function.operation());
ArrayBuilder<Object> args = ArrayBuilder.builder();
method += "Key";
args.add(function.key());
if (function instanceof KeyRecordsFunction || function instanceof KeyConditionFunction) {
method += "Records";
Collection<Long> records = function instanceof KeyRecordsFunction ? ((KeyRecordsFunction) function).source() : Finder.instance().visit(((KeyConditionFunction) function).source(), store);
args.add(records);
} else if (!(function instanceof IndexFunction)) {
throw new IllegalStateException("Invalid function value");
}
method += "Atomic";
args.add(func.timestamp());
args.add(store);
values[i] = Convert.javaToThrift(Reflection.callStatic(Operations.class, method, args.build()));
}
}
if (Keys.isNavigationKey(key)) {
Map<TObject, Set<Long>> index = timestamp == Time.NONE ? browse(store, key) : browse(store, key, timestamp);
Set<Long> records = index.entrySet().stream().filter(e -> e.getKey().is(operator, values)).map(e -> e.getValue()).flatMap(Set::stream).collect(Collectors.toCollection(LinkedHashSet::new));
return records;
} else if (Keys.isFunctionKey(key)) {
Set<Long> records = Sets.newLinkedHashSet();
for (long record : store.getAllRecords()) {
Set<TObject> aggregate = select(store, key, record, timestamp);
for (TObject tobject : aggregate) {
if (tobject.is(operator, values)) {
records.add(record);
break;
}
}
}
return records;
} else {
return timestamp == Time.NONE ? store.find(key, operator, values) : store.find(timestamp, key, operator, values);
}
}
use of com.cinchapi.concourse.server.storage.Store in project concourse by cinchapi.
the class Operations method browseNavigationKeyOptionalAtomic.
/**
* Perform "browse" functionality on a navigation key.
*
* @param key
* @param timestamp
* @param store
* @return a mapping from each possible destination value for a given
* navigation {@code key} to the records where the navigation could
* begin to retrieve the value by selecting the navigation
* {@code key}
*/
public static Map<TObject, Set<Long>> browseNavigationKeyOptionalAtomic(String key, long timestamp, Store store) {
String[] toks = key.split("\\.");
if (toks.length == 1) {
return timestamp == Time.NONE ? store.browse(key) : store.browse(key, timestamp);
} else {
String start = toks[0];
StringBuilder $key = new StringBuilder();
for (int i = 1; i < toks.length - 1; ++i) {
$key.append(toks[i]).append('.');
}
$key.append(toks[toks.length - 1]);
Map<TObject, Set<Long>> root = timestamp == Time.NONE ? store.browse(start) : store.browse(start, timestamp);
Map<TObject, Set<Long>> index = Maps.newLinkedHashMap();
root.entrySet().stream().filter(e -> e.getKey().getType() == Type.LINK).forEach(entry -> {
Link link = (Link) Convert.thriftToJava(entry.getKey());
Set<Long> nodes = entry.getValue();
for (long node : nodes) {
Set<TObject> values = traverseKeyRecordOptionalAtomic($key.toString(), link.longValue(), timestamp, store);
for (TObject value : values) {
index.computeIfAbsent(value, ignore -> Sets.newLinkedHashSet()).add(node);
}
}
});
return index;
}
}
use of com.cinchapi.concourse.server.storage.Store in project concourse by cinchapi.
the class ConcourseServer method findCriteria.
@Override
@TranslateClientExceptions
@VerifyAccessToken
@VerifyReadPermission
public Set<Long> findCriteria(TCriteria criteria, AccessToken creds, TransactionToken transaction, String environment) throws TException {
AbstractSyntaxTree ast = compiler.parse(criteria);
AtomicSupport store = getStore(transaction, environment);
Function<Store, Set<Long>> function = $store -> ast.accept(Finder.instance(), $store);
try {
return function.apply(store);
} catch (InsufficientAtomicityException e) {
return AtomicOperations.supplyWithRetry(store, atomic -> function.apply(atomic));
}
}
Aggregations