use of com.ibm.etcd.api.PutRequest in project etcd-java by IBM.
the class KvTest method testKvOps.
@Test
public void testKvOps() throws Exception {
proxy.start();
try (KvStoreClient directClient = EtcdClient.forEndpoint("localhost", 2379).withPlainText().build();
KvStoreClient client = EtcdClient.forEndpoint("localhost", 2391).withPlainText().build()) {
KvClient kvc = client.getKvClient();
assertEquals(0L, kvc.delete(bs("notthere")).sync().getDeleted());
ByteString a = bs("a"), b = bs("b"), v1 = bs("v1"), v2 = bs("v2");
// basic put
assertTrue(kvc.put(a, v1).sync().getHeader().getRevision() > 0);
// basic get
RangeResponse rr = kvc.get(bs("a")).sync();
assertEquals(1L, rr.getCount());
assertEquals(v1, rr.getKvs(0).getValue());
// basic delete
assertEquals(v1, kvc.delete(a).prevKv().sync().getPrevKvs(0).getValue());
assertEquals(0, kvc.get(bs("a")).sync().getCount());
PutRequest pr1 = kvc.put(a, v1).asRequest(), pr2 = kvc.put(b, v2).asRequest();
// batch put
assertEquals(2, kvc.batch().put(pr1).put(pr2).sync().getResponsesCount());
assertEquals(v1, kvc.get(a).sync().getKvs(0).getValue());
assertEquals(v2, kvc.get(b).sync().getKvs(0).getValue());
// basic transaction
ListenableFuture<TxnResponse> tresp = kvc.txnIf().cmpEqual(a).value(v1).and().cmpNotEqual(b).version(10).then().put(kvc.put(bs("new"), bs("newval")).asRequest()).async();
assertNotNull(tresp.get().getResponses(0).getResponsePut().getHeader());
// test disconnected behaviour
proxy.kill();
Thread.sleep(200L);
// should fail
ListenableFuture<RangeResponse> rrFut1 = kvc.get(bs("new")).async();
ListenableFuture<RangeResponse> rrFut2 = kvc.get(bs("new")).backoffRetry().async();
try {
rrFut1.get(1000, TimeUnit.SECONDS);
fail("expected get to fail while disconnected");
} catch (Exception e) {
// TODO
System.out.println("failed with: " + e);
}
// this one should still be retrying
assertFalse(rrFut2.isDone());
// reconnect
proxy.start();
// should succeed once network path is there again
long before = System.nanoTime();
RangeResponse rr2 = rrFut2.get(2000, TimeUnit.SECONDS);
long took = (System.nanoTime() - before) / 1000_000L;
assertEquals(bs("newval"), rr2.getKvs(0).getValue());
System.out.println("took " + took + "ms after network was reestablished");
} finally {
proxy.close();
}
}
Aggregations