@Override public TableProvider getTableProvider(TableSpec tableSpec) { Preconditions.checkNotNull(tableSpec, "null table spec"); return new RemoteTableProvider(tableSpec); } }
private TableWriteFunction<?, ?> getWriteFn() { TableWriteFunction<?, ?> writeFn = deserializeObject(WRITE_FN); if (writeFn != null) { writeFn.init(this.context); } return writeFn; } }
@Test public void testSerializeWithLimiterAndReadWriteCredFns() { doTestSerialize(createMockRateLimiter(), (key, value) -> 1, (key, value) -> 1); }
@Test public void testDeserializeReadFunctionAndLimiterRateOnlyWrite() { doTestDeserializeReadFunctionAndLimiter(true, false, true); }
@Test public void testSerializeSimple() { doTestSerialize(null, null, null); }
private TableWriteFunction<?, ?> getWriteFn() { TableWriteFunction<?, ?> writeFn = deserializeObject(WRITE_FN); if (writeFn != null) { writeFn.init(this.context); } return writeFn; } }
@Override public TableProvider getTableProvider(TableSpec tableSpec) { Preconditions.checkNotNull(tableSpec, "null table spec"); return new RemoteTableProvider(tableSpec); } }
@Test public void testSerializeWithLimiter() { doTestSerialize(createMockRateLimiter(), null, null); }
@Test public void testDeserializeReadFunctionAndLimiterRateOnlyReadWrite() { doTestDeserializeReadFunctionAndLimiter(true, true, true); }
private TableReadFunction<?, ?> getReadFn() { TableReadFunction<?, ?> readFn = deserializeObject(READ_FN); if (readFn != null) { readFn.init(this.context); } return readFn; }
@Override public TableProvider getTableProvider(TableSpec tableSpec) { Preconditions.checkNotNull(tableSpec, "null table spec"); return new RemoteTableProvider(tableSpec); } }
@Test public void testSerializeWithLimiterAndReadCredFn() { doTestSerialize(createMockRateLimiter(), (k, v) -> 1, null); }
@Test public void testDeserializeReadFunctionAndLimiterWrite() { doTestDeserializeReadFunctionAndLimiter(false, false, true); }
private TableReadFunction<?, ?> getReadFn() { TableReadFunction<?, ?> readFn = deserializeObject(READ_FN); if (readFn != null) { readFn.init(this.context); } return readFn; }
@Test public void testSerializeWithLimiterAndWriteCredFn() { doTestSerialize(createMockRateLimiter(), null, (k, v) -> 1); }
@Test public void testDeserializeReadFunctionAndLimiterRateOnlyRead() { doTestDeserializeReadFunctionAndLimiter(true, true, false); }
private TableWriteFunction<?, ?> getWriteFn() { TableWriteFunction<?, ?> writeFn = deserializeObject(WRITE_FN); if (writeFn != null) { writeFn.init(this.context); } return writeFn; } }
@Test public void testDeserializeReadFunctionAndLimiterReadWrite() { doTestDeserializeReadFunctionAndLimiter(false, true, true); }
private TableReadFunction<?, ?> getReadFn() { TableReadFunction<?, ?> readFn = deserializeObject(READ_FN); if (readFn != null) { readFn.init(this.context); } return readFn; }
private TableWriteFunction<?, ?> getWriteFn() { TableWriteFunction<?, ?> writeFn = deserializeObject(WRITE_FN); if (writeFn != null) { writeFn.init(this.context); } return writeFn; } }