[ingest-server-grpc] return stamp to GetResponse and ListResponse

Author: je-ik

direct/ingest-server/src/main/java/cz/o2/proxima/direct/server/RetrieveService.java

@@ -160,7 +160,8 @@ public void listAttributes(
               response.addValue(
                   ListResponse.AttrValue.newBuilder()
                       .setAttribute(kv.getAttribute())
-                      .setValue(ByteString.copyFrom(kv.getValue()))));
+                      .setValue(ByteString.copyFrom(kv.getValue()))
+                      .setStamp(kv.getStamp())));
       noticeListResult(request, entity, wildcard, kvs);
       replyLogged(responseObserver, request, response.build());
     } catch (Status s) {
@@ -297,6 +298,7 @@ public void get(Rpc.GetRequest request, StreamObserver<Rpc.GetResponse> response
             Rpc.GetResponse.newBuilder()
                 .setStatus(200)
                 .setValue(ByteString.copyFrom(kv.getValue()))
+                .setStamp(kv.getStamp())
                 .build());
       }
     } catch (Status s) {

direct/ingest-server/src/test/java/cz/o2/proxima/direct/server/RetrieveServiceTest.java

@@ -1091,6 +1091,7 @@ public void testGetValidExtendedScheme() throws InvalidProtocolBufferException {
     String key = "my-fancy-entity-key";
     ExtendedMessage payload = ExtendedMessage.newBuilder().setFirst(1).setSecond(2).build();
 
+    long now = System.currentTimeMillis();
     Optionals.get(server.direct.getWriter(attribute))
         .write(
             StreamElement.upsert(
@@ -1099,7 +1100,7 @@ public void testGetValidExtendedScheme() throws InvalidProtocolBufferException {
                 UUID.randomUUID().toString(),
                 key,
                 attribute.getName(),
-                System.currentTimeMillis(),
+                now,
                 payload.toByteArray()),
             CommitCallback.noop());
     Rpc.GetRequest request =
@@ -1140,6 +1141,7 @@ public void onCompleted() {
         200,
         response.getStatus());
     assertEquals(payload, ExtendedMessage.parseFrom(response.getValue().toByteArray()));
+    assertEquals(now, response.getStamp());
   }
 
   @Test
@@ -1192,14 +1194,15 @@ public void testListWithLongPrefix() {
     String key = "my-fancy-entity-key";
 
     OnlineAttributeWriter writer = Optionals.get(server.direct.getWriter(attribute));
+    long now = System.currentTimeMillis();
     writer.write(
         StreamElement.upsert(
             entity,
             attribute,
             UUID.randomUUID().toString(),
             key,
             attribute.toAttributePrefix() + "non-prefix",
-            System.currentTimeMillis(),
+            now,
             new byte[] {}),
         CommitCallback.noop());
     writer.write(
@@ -1209,7 +1212,7 @@ public void testListWithLongPrefix() {
             UUID.randomUUID().toString(),
             key,
             attribute.toAttributePrefix() + "prefix",
-            System.currentTimeMillis(),
+            now,
             new byte[] {}),
         CommitCallback.noop());
 

rpc/src/main/proto/rpc.proto

@@ -197,6 +197,11 @@ message GetResponse {
    **/
   bytes value = 3;
 
+  /**
+   * Timestamp in ms.
+   **/
+  uint64 stamp = 4;
+
 }
 
 /**
@@ -241,6 +246,9 @@ message ListResponse {
     /** The value of the attribute. */
     bytes value = 2;
 
+    /** Timestamp */
+    uint64 stamp = 3;
+
   }
 
   /**