@Override public TableProvider getTableProvider(String tableId) { return new RemoteTableProvider(tableId); } }
private TableReadFunction<?, ?> getReadFn(JavaTableConfig tableConfig) { TableReadFunction<?, ?> readFn = deserializeObject(tableConfig, RemoteTableDescriptor.READ_FN); if (readFn != null) { readFn.init(this.context); } return readFn; }
TableReadFunction readFn = getReadFn(tableConfig); RateLimiter rateLimiter = deserializeObject(tableConfig, RemoteTableDescriptor.RATE_LIMITER); if (rateLimiter != null) { rateLimiter.init(this.context); TableRateLimiter.CreditFunction<?, ?> readCreditFn = deserializeObject(tableConfig, RemoteTableDescriptor.READ_CREDIT_FN); TableRateLimiter readRateLimiter = rateLimiter != null && rateLimiter.getSupportedTags().contains(RemoteTableDescriptor.RL_READ_TAG) ? new TableRateLimiter(tableId, rateLimiter, readCreditFn, RemoteTableDescriptor.RL_READ_TAG) : null; TableRetryPolicy readRetryPolicy = deserializeObject(tableConfig, RemoteTableDescriptor.READ_RETRY_POLICY); TableWriteFunction writeFn = getWriteFn(tableConfig); TableRateLimiter writeRateLimiter = null; TableRetryPolicy writeRetryPolicy = null; if (writeFn != null) { TableRateLimiter.CreditFunction<?, ?> writeCreditFn = deserializeObject(tableConfig, RemoteTableDescriptor.WRITE_CREDIT_FN); writeRateLimiter = rateLimiter != null && rateLimiter.getSupportedTags().contains(RemoteTableDescriptor.RL_WRITE_TAG) ? new TableRateLimiter(tableId, rateLimiter, writeCreditFn, RemoteTableDescriptor.RL_WRITE_TAG) : null; writeRetryPolicy = deserializeObject(tableConfig, RemoteTableDescriptor.WRITE_RETRY_POLICY); retryExecutor = createRetryExecutor();
RemoteTableProvider provider = new RemoteTableProvider(desc.getTableId()); provider.init(createMockContext(desc)); Table table = provider.getTable(); Assert.assertTrue(table instanceof RemoteTable); RemoteTable rwTable = (RemoteTable) table;
private TableWriteFunction<?, ?> getWriteFn(JavaTableConfig tableConfig) { TableWriteFunction<?, ?> writeFn = deserializeObject(tableConfig, RemoteTableDescriptor.WRITE_FN); if (writeFn != null) { writeFn.init(this.context); } return writeFn; }