SendToTableOperatorImpl(SendToTableOperatorSpec<K, V> sendToTableOpSpec, Context context) { this.sendToTableOpSpec = sendToTableOpSpec; this.table = context.getTaskContext().getTable(sendToTableOpSpec.getTableId()); }
StreamTableJoinOperatorImpl(StreamTableJoinOperatorSpec<K, M, R, JM> joinOpSpec, Context context) { this.joinOpSpec = joinOpSpec; this.table = context.getTaskContext().getTable(joinOpSpec.getTableId()); }
@Override public Table getTable() { String realTableId = tableSpec.getConfig().get(REAL_TABLE_ID); ReadableTable table = (ReadableTable) this.context.getTaskContext().getTable(realTableId); String cacheTableId = tableSpec.getConfig().get(CACHE_TABLE_ID); ReadWriteTable cache; if (cacheTableId != null) { cache = (ReadWriteTable) this.context.getTaskContext().getTable(cacheTableId); } else { cache = createDefaultCacheTable(realTableId); defaultCaches.add(cache); } boolean isWriteAround = Boolean.parseBoolean(tableSpec.getConfig().get(WRITE_AROUND)); CachingTable cachingTable = new CachingTable(tableSpec.getId(), table, cache, isWriteAround); cachingTable.init(this.context); return cachingTable; }
SendToTableOperatorImpl(SendToTableOperatorSpec<K, V> sendToTableOpSpec, Context context) { this.sendToTableOpSpec = sendToTableOpSpec; this.table = (ReadWriteTable) context.getTaskContext().getTable(sendToTableOpSpec.getTableSpec().getId()); }
StreamTableJoinOperatorImpl(StreamTableJoinOperatorSpec<K, M, R, JM> joinOpSpec, Context context) { this.joinOpSpec = joinOpSpec; this.table = (ReadableTable) context.getTaskContext().getTable(joinOpSpec.getTableSpec().getId()); }
StreamTableJoinOperatorImpl(StreamTableJoinOperatorSpec<K, M, R, JM> joinOpSpec, Context context) { this.joinOpSpec = joinOpSpec; this.table = (ReadableTable) context.getTaskContext().getTable(joinOpSpec.getTableSpec().getId()); }
StreamTableJoinOperatorImpl(StreamTableJoinOperatorSpec<K, M, R, JM> joinOpSpec, Context context) { this.joinOpSpec = joinOpSpec; this.table = (ReadableTable) context.getTaskContext().getTable(joinOpSpec.getTableSpec().getId()); }
StreamTableJoinOperatorImpl(StreamTableJoinOperatorSpec<K, M, R, JM> joinOpSpec, Context context) { this.joinOpSpec = joinOpSpec; this.table = (ReadableTable) context.getTaskContext().getTable(joinOpSpec.getTableSpec().getId()); }
@Override public Table getTable() { String realTableId = tableSpec.getConfig().get(REAL_TABLE_ID); ReadableTable table = (ReadableTable) this.context.getTaskContext().getTable(realTableId); String cacheTableId = tableSpec.getConfig().get(CACHE_TABLE_ID); ReadWriteTable cache; if (cacheTableId != null) { cache = (ReadWriteTable) this.context.getTaskContext().getTable(cacheTableId); } else { cache = createDefaultCacheTable(realTableId); defaultCaches.add(cache); } boolean isWriteAround = Boolean.parseBoolean(tableSpec.getConfig().get(WRITE_AROUND)); CachingTable cachingTable = new CachingTable(tableSpec.getId(), table, cache, isWriteAround); cachingTable.init(this.context); return cachingTable; }
@Override public Table getTable() { String realTableId = tableSpec.getConfig().get(REAL_TABLE_ID); ReadableTable table = (ReadableTable) this.context.getTaskContext().getTable(realTableId); String cacheTableId = tableSpec.getConfig().get(CACHE_TABLE_ID); ReadWriteTable cache; if (cacheTableId != null) { cache = (ReadWriteTable) this.context.getTaskContext().getTable(cacheTableId); } else { cache = createDefaultCacheTable(realTableId); defaultCaches.add(cache); } boolean isWriteAround = Boolean.parseBoolean(tableSpec.getConfig().get(WRITE_AROUND)); CachingTable cachingTable = new CachingTable(tableSpec.getId(), table, cache, isWriteAround); cachingTable.init(this.context); return cachingTable; }
SendToTableOperatorImpl(SendToTableOperatorSpec<K, V> sendToTableOpSpec, Context context) { this.sendToTableOpSpec = sendToTableOpSpec; this.table = (ReadWriteTable) context.getTaskContext().getTable(sendToTableOpSpec.getTableSpec().getId()); }
SendToTableOperatorImpl(SendToTableOperatorSpec<K, V> sendToTableOpSpec, Context context) { this.sendToTableOpSpec = sendToTableOpSpec; this.table = (ReadWriteTable) context.getTaskContext().getTable(sendToTableOpSpec.getTableSpec().getId()); }
SendToTableOperatorImpl(SendToTableOperatorSpec<K, V> sendToTableOpSpec, Context context) { this.sendToTableOpSpec = sendToTableOpSpec; this.table = (ReadWriteTable) context.getTaskContext().getTable(sendToTableOpSpec.getTableSpec().getId()); }
@Override public Table getTable() { String realTableId = tableSpec.getConfig().get(REAL_TABLE_ID); ReadableTable table = (ReadableTable) this.context.getTaskContext().getTable(realTableId); String cacheTableId = tableSpec.getConfig().get(CACHE_TABLE_ID); ReadWriteTable cache; if (cacheTableId != null) { cache = (ReadWriteTable) this.context.getTaskContext().getTable(cacheTableId); } else { cache = createDefaultCacheTable(realTableId); defaultCaches.add(cache); } boolean isWriteAround = Boolean.parseBoolean(tableSpec.getConfig().get(WRITE_AROUND)); CachingTable cachingTable = new CachingTable(tableSpec.getId(), table, cache, isWriteAround); cachingTable.init(this.context); return cachingTable; }
@Override public ReadWriteTable getTable() { Preconditions.checkNotNull(context, String.format("Table %s not initialized", tableId)); JavaTableConfig tableConfig = new JavaTableConfig(context.getJobContext().getConfig()); String realTableId = tableConfig.getForTable(tableId, CachingTableDescriptor.REAL_TABLE_ID); ReadWriteTable table = this.context.getTaskContext().getTable(realTableId); String cacheTableId = tableConfig.getForTable(tableId, CachingTableDescriptor.CACHE_TABLE_ID); ReadWriteTable cache; if (cacheTableId != null) { cache = this.context.getTaskContext().getTable(cacheTableId); } else { cache = createDefaultCacheTable(realTableId, tableConfig); defaultCaches.add(cache); } boolean isWriteAround = Boolean.parseBoolean(tableConfig.getForTable(tableId, CachingTableDescriptor.WRITE_AROUND)); CachingTable cachingTable = new CachingTable(tableId, table, cache, isWriteAround); cachingTable.init(this.context); return cachingTable; }
when(table.get("2")).thenReturn(null); Context context = new MockContext(); when(context.getTaskContext().getTable(tableId)).thenReturn(table);
}).when(context.getTaskContext()).getTable(anyString());