fix(transaction): Fix schedule queues

src/server/list_family_test.cc

@@ -1326,14 +1326,14 @@ TEST_F(ListFamilyTest, LMPopWrongType) {
 // Reproduce a flow that trigerred a wrong DCHECK in the transaction flow.
 TEST_F(ListFamilyTest, AwakeMulti) {
   auto f1 = pp_->at(1)->LaunchFiber(Launch::dispatch, [&] {
-    for (unsigned i = 0; i < 100; ++i) {
+    for (unsigned i = 0; i < 200; ++i) {
       Run("CONSUMER", {"blmove", "src", "dest", "LEFT", "LEFT", "0"});
     };
   });
   auto f2 = pp_->at(1)->LaunchFiber([&] {
-    for (unsigned i = 0; i < 100; ++i) {
+    for (unsigned i = 0; i < 200; ++i) {
       Run("PROD", {"lpush", "src", "a"});
-      ThisFiber::SleepFor(50us);
+      ThisFiber::SleepFor(10us);
     };
   });
 

src/server/transaction.cc

@@ -759,11 +759,7 @@ void Transaction::ScheduleInternal() {
 
     ScheduleContext schedule_ctx{this, optimistic_exec};
 
-    // TODO: this optimization is disabled due to a issue #4648 revealing this code can
-    // lead to transaction not being scheduled.
-    // To reproduce the bug remove the false in the condition and run
-    // ./list_family_test --gtest_filter=*AwakeMulti on alpine machine
-    if (false && unique_shard_cnt_ == 1) {
+    if (unique_shard_cnt_ == 1) {
       // Single shard optimization. Note: we could apply the same optimization
       // to multi-shard transactions as well by creating a vector of ScheduleContext.
       schedule_queues[unique_shard_id_].queue.Push(&schedule_ctx);
@@ -1212,7 +1208,7 @@ void Transaction::ScheduleBatchInShard() {
     // We do this to avoid the situation where we have a data race, where
     // a transaction is added to the queue, we've checked that sq.armed is true and skipped
     // adding the callback that fetches the transaction.
-    sq.armed.store(false, memory_order_release);
+    sq.armed.exchange(false, memory_order_acq_rel);
   }
 }