bugfix in cache trans queue

Author: mag1c-h

ucm/store/cache/cc/dump_queue.cc

@@ -140,11 +140,11 @@ void DumpQueue::BackendDumpStage()
     dumping_.ConsumerLoop(stop_, [this](auto&& task) {
         if (task.backendTaskHandle > finishedBackendTaskHandle_) {
             auto s = backend_->Wait(task.backendTaskHandle);
+            finishedBackendTaskHandle_ = task.backendTaskHandle;
             if (s.Failure()) {
                 UC_ERROR("Failed({}) to wait backend task({}).", s, task.backendTaskHandle);
                 return;
             }
-            finishedBackendTaskHandle_ = task.backendTaskHandle;
         }
     });
 }

ucm/store/cache/cc/load_queue.cc

@@ -125,14 +125,14 @@ void LoadQueue::TransferOneTask(Trans::Stream* stream, size_t tensorSize, ShardT
 {
     if (task.backendTaskHandle > finishedBackendTaskHandle_) {
         auto s = backend_->Wait(task.backendTaskHandle);
+        finishedBackendTaskHandle_ = task.backendTaskHandle;
         if (s.Failure()) [[unlikely]] {
             failureSet_->Insert(task.taskHandle);
             if (task.waiter) { task.waiter->Done(); }
             UC_ERROR("Failed({}) to wait backend task({}).", s, task.backendTaskHandle);
             return;
         }
         task.bufferHandle.MarkReady();
-        finishedBackendTaskHandle_ = task.backendTaskHandle;
     }
     for (;;) {
         if (failureSet_->Contains(task.taskHandle)) {