Split the pr to support named blob ttl update only

ambry-api/src/main/java/com/github/ambry/named/NamedBlobDb.java

@@ -50,6 +50,15 @@ default CompletableFuture<NamedBlobRecord> get(String accountName, String contai
     return get(accountName, containerName, blobName, GetOption.None);
   }
 
+  /**
+   * Support ttl update for {@link NamedBlobRecord}
+   * @param record the {@link NamedBlobRecord}
+   * @param state the {@link NamedBlobState}
+   * @return a {@link CompletableFuture} that will eventually contain either the {@link NamedBlobRecord} for the named
+   *         blob or an exception if an error occurred.
+   */
+  CompletableFuture<PutResult> ttlUpdate(NamedBlobRecord record, NamedBlobState state);
+
   /**
    * List blobs that start with a provided prefix in a container. This returns paginated results. If there are
    * additional pages to read, {@link Page#getNextPageToken()} will be non null.

ambry-api/src/main/java/com/github/ambry/named/NamedBlobRecord.java

@@ -25,9 +25,9 @@ public class NamedBlobRecord {
   private final String accountName;
   private final String containerName;
   private final String blobName;
-  private final String blobId;
   private final long expirationTimeMs;
   private long version;
+  private String blobId;
 
   /**
    * @param accountName the account name.
@@ -98,6 +98,10 @@ public void setVersion(long newVersion) {
     version = newVersion;
   }
 
+  public void setBlobId(String blobId) {
+    this.blobId = blobId;
+  }
+
   /**
    * @return the expiration time in milliseconds since epoch, or -1 if the blob should be permanent.
    */

ambry-frontend/src/test/java/com/github/ambry/frontend/TestNamedBlobDb.java

@@ -25,6 +25,7 @@
 import com.github.ambry.rest.RestServiceException;
 import com.github.ambry.utils.Pair;
 import com.github.ambry.utils.Time;
+import com.github.ambry.utils.Utils;
 import java.io.IOException;
 import java.sql.Timestamp;
 import java.util.ArrayList;
@@ -70,7 +71,7 @@ public CompletableFuture<NamedBlobRecord> get(String accountName, String contain
     } else if (recordWithDelete.getSecond() != 0 && recordWithDelete.getSecond() < time.milliseconds()
         && !includeDeletedOptions.contains(option)) {
       future.completeExceptionally(new RestServiceException("Deleted", RestServiceErrorCode.Deleted));
-    } else if (recordWithDelete.getFirst().getExpirationTimeMs() != 0
+    } else if (recordWithDelete.getFirst().getExpirationTimeMs() != Utils.Infinite_Time
         && recordWithDelete.getFirst().getExpirationTimeMs() < time.milliseconds() && !includeExpiredOptions.contains(
         option)) {
       future.completeExceptionally(new RestServiceException("Deleted", RestServiceErrorCode.Deleted));
@@ -140,6 +141,12 @@ public CompletableFuture<PutResult> put(NamedBlobRecord record, NamedBlobState s
     return future;
   }
 
+  @Override
+  public CompletableFuture<PutResult> ttlUpdate(NamedBlobRecord record, NamedBlobState state) {
+    //TODO: implement later
+    return null;
+  }
+
   @Override
   public CompletableFuture<PutResult> updateBlobStateToReady(NamedBlobRecord record) {
     CompletableFuture<PutResult> future = new CompletableFuture<>();
@@ -278,6 +285,6 @@ private void setException(Exception exception) {
 
   @Override
   public void close() throws IOException {
-    
+
   }
 }

ambry-named-mysql/src/integration-test/java/com/github/ambry/named/MySqlNamedBlobDbIntegrationTest.java

@@ -199,6 +199,31 @@ public void testPutGetListDeleteSequence() throws Exception {
     }
   }
 
+  /**
+   * Test ttl update for named blob.
+   * @throws Exception
+   */
+  @Test
+  public void testUpdateTtlForNamedBlob() throws Exception {
+    Account account = accountService.getAllAccounts().iterator().next();
+    Container container = account.getAllContainers().iterator().next();
+    String blobId = getBlobId(account, container);
+    String blobName = "blobName";
+    long expirationTime = System.currentTimeMillis() + TimeUnit.HOURS.toMillis(1);
+    NamedBlobRecord record =
+        new NamedBlobRecord(account.getName(), container.getName(), blobName, blobId, expirationTime);
+    namedBlobDb.put(record).get();
+    time.setCurrentMilliseconds(System.currentTimeMillis());
+    PutResult ttlUpdateResult = namedBlobDb.ttlUpdate(record, NamedBlobState.IN_PROGRESS).get();
+    record.setVersion(ttlUpdateResult.getInsertedRecord().getVersion());
+    time.setCurrentMilliseconds(System.currentTimeMillis());
+    namedBlobDb.updateBlobStateToReady(record).get();
+    time.setCurrentMilliseconds(System.currentTimeMillis());
+    NamedBlobRecord recordFromDb = namedBlobDb.get(account.getName(), container.getName(), blobName).get();
+    assertEquals("Mismatch on blob version", ttlUpdateResult.getInsertedRecord().getVersion(), recordFromDb.getVersion());
+    assertEquals("Mismatch on blob expiration time", Utils.Infinite_Time, recordFromDb.getExpirationTimeMs());
+  }
+
   /**
    * Test behavior with expired blobs
    */

ambry-named-mysql/src/main/java/com/github/ambry/named/MySqlNamedBlobDb.java

@@ -764,6 +764,30 @@ private void applySoftDelete(short accountId, short containerId, String blobName
     }
   }
 
+  @Override
+  public CompletableFuture<PutResult> ttlUpdate(NamedBlobRecord record, NamedBlobState state) {
+    return executeTransactionAsync(record.getAccountName(), record.getContainerName(), true,
+        (accountId, containerId, connection) -> {
+          long startTime = this.time.milliseconds();
+          logger.trace("NamedBlobPutInfo: accountId='{}', containerId='{}', blobName='{}'", accountId, containerId,
+              record.getBlobName());
+          //for ttl update, get the blob id first before insert a new record.
+          NamedBlobRecord recordCurrent;
+          try {
+            recordCurrent =
+                run_get_v2(record.getAccountName(), record.getContainerName(), record.getBlobName(), GetOption.None,
+                    accountId, containerId, connection);
+          } catch (RestServiceException e) {
+            throw buildException("Failed to do ttl update due to not able to found existing record", e.getErrorCode(),
+                record.getAccountName(), record.getContainerName(), record.getBlobName());
+          }
+          record.setBlobId(recordCurrent.getBlobId());
+          PutResult putResult = run_put_v2(record, state, accountId, containerId, connection);
+          metricsRecoder.namedBlobPutTimeInMs.update(this.time.milliseconds() - startTime);
+          return putResult;
+        }, null);
+  }
+
   /**
    * Build the version for Named Blob row based on timestamp and uuid postfix.
    * @return a long number whose rightmost 5 digits are uuid postfix, and the remaining digits are current timestamp

ambry-named-mysql/src/test/java/com/github/ambry/named/MySqlNamedBlobDbTest.java

@@ -29,6 +29,7 @@
 import com.github.ambry.rest.RestServiceErrorCode;
 import com.github.ambry.rest.RestServiceException;
 import com.github.ambry.utils.TestUtils;
+import com.github.ambry.utils.Utils;
 import java.io.IOException;
 import java.sql.Connection;
 import java.sql.PreparedStatement;
@@ -164,6 +165,25 @@ public void testUpdateBlobStateToReady() throws Exception {
     assertEquals("Blob Id is not matched with the record", id, namedBlobRecord.getBlobId());
   }
 
+  /**
+   * Test update ttl for named blob.
+   * @throws Exception
+   */
+  @Test
+  public void testUpdateTtlForNamedBlob() throws Exception {
+    dataSourceFactory.setLocalDatacenter(foundDatacenter);
+    dataSourceFactory.triggerDataResultSet(datacenters);
+    long expirationTimeMs = System.currentTimeMillis() + 3600;
+    NamedBlobRecord record = new NamedBlobRecord(account.getName(), container.getName(), "blobName", id, expirationTimeMs);
+    namedBlobDb.put(record, NamedBlobState.READY, true).get();
+    assertEquals("Mismatch on expiration time", expirationTimeMs , record.getExpirationTimeMs());
+    PutResult putResult = namedBlobDb.ttlUpdate(record, NamedBlobState.IN_PROGRESS).get();
+    record.setVersion(putResult.getInsertedRecord().getVersion());
+    namedBlobDb.updateBlobStateToReady(record).get();
+    record = namedBlobDb.get(account.getName(), container.getName(), "blobName").get();
+    assertEquals("Mismatch on expiration time", Utils.Infinite_Time , record.getExpirationTimeMs());
+  }
+
   /**
    * @param callable an async call, where the {@link Future} is expected to be completed with an exception.
    * @param errorCode the expected {@link RestServiceErrorCode}.