From 6817b5be3a21163db680a59b91e790b47645cb5c Mon Sep 17 00:00:00 2001
From: tison <wander4096@gmail.com>
Date: Sun, 14 May 2023 20:47:11 +0800
Subject: [PATCH] feat: add Compare protos

Signed-off-by: tison <wander4096@gmail.com>
---
 zeronos-proto/src/main/proto/rpc.proto        | 32 +++++++++++++
 .../server/state/ZeroStateMachine.java        | 47 +++++++++++++++++++
 2 files changed, 79 insertions(+)

diff --git a/zeronos-proto/src/main/proto/rpc.proto b/zeronos-proto/src/main/proto/rpc.proto
index 179e9b1..2bac900 100644
--- a/zeronos-proto/src/main/proto/rpc.proto
+++ b/zeronos-proto/src/main/proto/rpc.proto
@@ -77,8 +77,40 @@ message ResponseOp {
   }
 }
 
+message Compare {
+  enum CompareResult {
+    EQUAL = 0;
+    GREATER = 1;
+    LESS = 2;
+    NOT_EQUAL = 3;
+    GREATER_OR_EQUAL = 4;
+    LESS_OR_EQUAL = 5;
+  }
+
+  enum CompareTarget {
+    VERSION = 0;
+    VALUE = 1;
+  }
+
+  // result is logical comparison operation for this comparison.
+  CompareResult result = 1;
+  // target is the key-value field to inspect for the comparison.
+  CompareTarget target = 2;
+  // key is the subject key for the comparison operation.
+  bytes key = 3;
+
+  oneof target_union {
+    // version is the version of the given key
+    int64 version = 4;
+    // value is the value of the given key, in bytes.
+    bytes value = 5;
+  }
+}
+
 message TxnRequest {
+  repeated Compare compare = 1;
   repeated RequestOp success = 2;
+  repeated RequestOp failure = 3;
 }
 
 message TxnResponse {
diff --git a/zeronos-server/src/main/java/io/korandoru/zeronos/server/state/ZeroStateMachine.java b/zeronos-server/src/main/java/io/korandoru/zeronos/server/state/ZeroStateMachine.java
index 5456861..e0507f8 100644
--- a/zeronos-server/src/main/java/io/korandoru/zeronos/server/state/ZeroStateMachine.java
+++ b/zeronos-server/src/main/java/io/korandoru/zeronos/server/state/ZeroStateMachine.java
@@ -16,6 +16,7 @@
 
 package io.korandoru.zeronos.server.state;
 
+import io.korandoru.zeronos.proto.Compare;
 import io.korandoru.zeronos.proto.DeleteRangeRequest;
 import io.korandoru.zeronos.proto.DeleteRangeResponse;
 import io.korandoru.zeronos.proto.KeyBytes;
@@ -42,6 +43,7 @@
 import java.io.File;
 import java.io.IOException;
 import java.util.ArrayList;
+import java.util.Comparator;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.atomic.AtomicLong;
@@ -156,6 +158,51 @@ private DeleteRangeResponse deleteRange(WriteTxn writeTxn, DeleteRangeRequest re
         return DeleteRangeResponse.newBuilder().setDeleted(r.getTotal()).build();
     }
 
+    private boolean applyCompares(ReadTxn readTxn, List<Compare> compareList, long revision)
+            throws InvalidProtocolBufferException {
+        for (Compare compare : compareList) {
+            final RangeRequest req = RangeRequest.newBuilder().setKey(compare.getKey()).build();
+            final RangeResponse resp = range(readTxn, req, revision);
+            if (!applyCompare(compare, resp.getKvsList())) {
+                return false;
+            }
+        }
+        return true;
+    }
+
+    private boolean applyCompare(Compare compare, List<KeyValue> kvs) {
+        if (kvs.isEmpty()) {
+            // always fail if comparing a value on a key/keys that doesn't exist;
+            // null == empty string in gRPC; no way to represent missing value
+            return false;
+        }
+
+        final Comparator<ByteString> comparator = ByteString.unsignedLexicographicalComparator();
+        for (KeyValue kv : kvs) {
+            final int result = switch (compare.getTarget()) {
+                case VERSION -> Long.compare(kv.getVersion(), compare.getVersion());
+                case VALUE -> comparator.compare(kv.getValue(), compare.getValue());
+                default -> throw new UnsupportedOperationException(compare.getTarget().name());
+            };
+
+            final boolean match = switch (compare.getResult()) {
+                case EQUAL -> result == 0;
+                case GREATER -> result > 0;
+                case LESS -> result < 0;
+                case NOT_EQUAL -> result != 0;
+                case GREATER_OR_EQUAL -> result >= 0;
+                case LESS_OR_EQUAL -> result <= 0;
+                default -> throw new UnsupportedOperationException(compare.getResult().name());
+            };
+
+            if (!match) {
+                return false;
+            }
+        }
+
+        return true;
+    }
+
     @Override
     public CompletableFuture<Message> query(Message request) {
         final List<RequestOp> requestList;