tikv
/
client-java
mirror of https://github.com/tikv/client-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

support putAtomic and deleteAtomic API (#214) 

Signed-off-by: marsishandsome <marsishandsome@gmail.com>
This commit is contained in:
Liangliang Gu 2021-06-25 11:59:06 +08:00 committed by GitHub
parent 41b24bb877
commit 81d4981c80
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 35 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/main/java/org/tikv/common/region/RegionStoreClient.java
View File

@ -872,7 +872,7 @@ public class RegionStoreClient extends AbstractRegionStoreClient {
return resp.getTtl();
}
public void rawDelete(BackOffer backOffer, ByteString key) {
public void rawDelete(BackOffer backOffer, ByteString key, boolean atomic) {
Histogram.Timer requestTimer =
GRPC_RAW_REQUEST_LATENCY.labels("client_grpc_raw_delete").startTimer();
try {
@ -881,6 +881,7 @@ public class RegionStoreClient extends AbstractRegionStoreClient {
RawDeleteRequest.newBuilder()
.setContext(region.getReplicaContext(storeType))
.setKey(key)
.setForCas(atomic)
.build();
KVErrorHandler<RawDeleteResponse> handler =
@ -908,7 +909,8 @@ public class RegionStoreClient extends AbstractRegionStoreClient {
}
}
public void rawPut(BackOffer backOffer, ByteString key, ByteString value, long ttl) {
public void rawPut(
BackOffer backOffer, ByteString key, ByteString value, long ttl, boolean atomic) {
Histogram.Timer requestTimer =
GRPC_RAW_REQUEST_LATENCY.labels("client_grpc_raw_put").startTimer();
try {
@ -919,6 +921,7 @@ public class RegionStoreClient extends AbstractRegionStoreClient {
.setKey(key)
.setValue(value)
.setTtl(ttl)
.setForCas(atomic)
.build();
KVErrorHandler<RawPutResponse> handler =

32
src/main/java/org/tikv/raw/RawKVClient.java
View File

@ -112,6 +112,21 @@ public class RawKVClient implements AutoCloseable {
* @param ttl the ttl of the key (in seconds), 0 means the key will never be outdated
*/
public void put(ByteString key, ByteString value, long ttl) {
put(key, value, ttl, false);
}
/**
* Put a raw key-value pair to TiKV. This API is atomic.
*
* @param key raw key
* @param value raw value
* @param ttl the ttl of the key (in seconds), 0 means the key will never be outdated
*/
public void putAtomic(ByteString key, ByteString value, long ttl) {
put(key, value, ttl, true);
}
private void put(ByteString key, ByteString value, long ttl, boolean atomic) {
String label = "client_raw_put";
Histogram.Timer requestTimer = RAW_REQUEST_LATENCY.labels(label).startTimer();
try {
@ -119,7 +134,7 @@ public class RawKVClient implements AutoCloseable {
while (true) {
RegionStoreClient client = clientBuilder.build(key);
try {
client.rawPut(backOffer, key, value, ttl);
client.rawPut(backOffer, key, value, ttl, atomic);
RAW_REQUEST_SUCCESS.labels(label).inc();
return;
} catch (final TiKVException e) {
@ -528,6 +543,19 @@ public class RawKVClient implements AutoCloseable {
* @param key raw key to be deleted
*/
public void delete(ByteString key) {
delete(key, false);
}
/**
* Delete a raw key-value pair from TiKV if key exists. This API is atomic.
*
* @param key raw key to be deleted
*/
public void deleteAtomic(ByteString key) {
delete(key, true);
}
private void delete(ByteString key, boolean atomic) {
String label = "client_raw_delete";
Histogram.Timer requestTimer = RAW_REQUEST_LATENCY.labels(label).startTimer();
try {
@ -535,7 +563,7 @@ public class RawKVClient implements AutoCloseable {
while (true) {
RegionStoreClient client = clientBuilder.build(key);
try {
client.rawDelete(defaultBackOff(), key);
client.rawDelete(defaultBackOff(), key, atomic);
RAW_REQUEST_SUCCESS.labels(label).inc();
return;
} catch (final TiKVException e) {