Refine search
static Map<ByteBuffer, List<ColumnOrSuperColumn>> getColsByKey(List<KeySlice> firstPage) { Map<ByteBuffer, List<ColumnOrSuperColumn>> ret = Maps.newHashMapWithExpectedSize(firstPage.size()); for (KeySlice e : firstPage) { ret.put(ByteBuffer.wrap(e.getKey()), e.getColumns()); } return ret; } }
public KeySlice prefixKeySlice(KeySlice unprefixed) { KeySlice prefixed = new KeySlice(); prefixed.key = ps.toByteBuffer(unprefixed.key); prefixed.columns = unprefixed.columns; return prefixed; }
protected List<KeySlice> checkFreshSlices(List<KeySlice> ks) { if (0 == ks.size()) { seenEnd = true; return Collections.emptyList(); } nextStartKey = ks.get(ks.size() - 1).bufferForKey(); nextStartToken = partitioner.getToken(nextStartKey); if (nextStartToken.equals(endToken)) { seenEnd = true; if (omitEndToken) ks.remove(ks.size() - 1); } return ks; }
@Override public boolean apply(@Nullable KeySlice row) { return (row != null) && row.getColumns().size() > 0; } }
public void setFieldValue(_Fields field, Object value) { switch (field) { case KEY: if (value == null) { unsetKey(); } else { setKey((ByteBuffer)value); } break; case COLUMNS: if (value == null) { unsetColumns(); } else { setColumns((List<ColumnOrSuperColumn>)value); } break; } }
@Test public void getSizeForKeySliceWithKeyAndColumns() { assertThat(ThriftObjectSizeUtils.getKeySliceSize(new KeySlice() .setKey(TEST_NAME.getBytes()) .setColumns(ImmutableList.of(EMPTY_COLUMN_OR_SUPERCOLUMN)))) .isEqualTo(TEST_NAME.getBytes().length + EMPTY_COLUMN_OR_SUPERCOLUMN_SIZE); }
counterColumns = new ArrayList<CounterSuperColumn>(slice.getColumnsSize()); for (ColumnOrSuperColumn column : slice.columns) { counterColumns.add(column.counter_super_column); results.put(slice.getKey(), counterColumns); counterColumns = new ArrayList<CounterColumn>(slice.getColumnsSize()); for (ColumnOrSuperColumn column : slice.columns) { counterColumns.add(column.counter_column); results.put(slice.getKey(), counterColumns);
/** * Fetches Row keys from a {@link List} of {@link KeySlice}. * * @param keySlices the key slices * @param metadata the metadata * @return the row keys */ public static List<Object> getRowKeys(List<KeySlice> keySlices, EntityMetadata metadata) { PropertyAccessor<?> accessor = PropertyAccessorFactory.getPropertyAccessor((Field) metadata.getIdAttribute() .getJavaMember()); List<Object> rowKeys = new ArrayList<Object>(); for (KeySlice keySlice : keySlices) { byte[] key = keySlice.getKey(); Object rowKey = accessor.fromBytes(metadata.getIdAttribute().getJavaType(), key); rowKeys.add(rowKey); } return rowKeys; }
@Test public void getSizeForKeySliceWithKeyNotSetButColumnsSet() { assertThat(ThriftObjectSizeUtils.getKeySliceSize(new KeySlice() .setColumns(ImmutableList.of(EMPTY_COLUMN_OR_SUPERCOLUMN)))) .isEqualTo(NULL_SIZE + EMPTY_COLUMN_OR_SUPERCOLUMN_SIZE); }
Iterator<KeySlice> iter = ks.iterator(); while (iter.hasNext()) { if (iter.next().getColumnsSize() == 0) iter.remove(); String token = partitioner.getTokenForKey(lastRow.bufferForKey()).toString(); checkpointManager.trackCheckpoint(tokenPair.left, token); if (getRepeatLastToken()) {
public KeySlice setKey(byte[] key) { setKey(key == null ? (ByteBuffer)null : ByteBuffer.wrap(key)); return this; }
if (ks.getColumnsSize() == 0)
@Override public RecordIterator<Entry> getEntries() { ensureOpen(); return new RecordIterator<Entry>() { final Iterator<Entry> columns = CassandraHelper.makeEntryIterator(mostRecentRow.getColumns(), entryGetter, columnSlice.getSliceEnd(), columnSlice.getLimit()); @Override public boolean hasNext() { ensureOpen(); return columns.hasNext(); } @Override public Entry next() { ensureOpen(); return columns.next(); } @Override public void close() { closeIterator(); } @Override public void remove() { throw new UnsupportedOperationException(); } }; }
public void setFieldValue(_Fields field, Object value) { switch (field) { case KEY: if (value == null) { unsetKey(); } else { setKey((ByteBuffer)value); } break; case COLUMNS: if (value == null) { unsetColumns(); } else { setColumns((List<ColumnOrSuperColumn>)value); } break; } }
@Test public void getKeySlicesSize() { List<KeySlice> slices = ImmutableList.of( new KeySlice() .setKey(TEST_NAME_BYTES) .setColumns(ImmutableList.of(EMPTY_COLUMN_OR_SUPERCOLUMN))); long expectedSize = TEST_NAME_BYTES_SIZE + EMPTY_COLUMN_OR_SUPERCOLUMN_SIZE; assertThat(ThriftObjectSizeUtils.getApproximateSizeOfKeySlices(slices)).isEqualTo(expectedSize); }
/** * Fetches Row keys from a {@link List} of {@link KeySlice} */ public static List<Object> getRowKeys(List<KeySlice> keySlices, EntityMetadata metadata) { PropertyAccessor<?> accessor = PropertyAccessorFactory.getPropertyAccessor((Field) metadata.getIdAttribute() .getJavaMember()); List<Object> rowKeys = new ArrayList<Object>(); for (KeySlice keySlice : keySlices) { byte[] key = keySlice.getKey(); Object rowKey = accessor.fromBytes(metadata.getIdAttribute().getJavaType(), key); rowKeys.add(rowKey); } return rowKeys; }
Iterator<KeySlice> iter = ks.iterator(); while (iter.hasNext()) { if (iter.next().getColumnsSize() == 0) iter.remove(); String token = partitioner.getTokenForKey(lastRow.bufferForKey()).toString(); checkpointManager.trackCheckpoint(tokenPair.left, token); if (getRepeatLastToken()) {
public KeySlice setKey(byte[] key) { setKey(key == null ? (ByteBuffer)null : ByteBuffer.wrap(key)); return this; }