From fe5b1834ff0ff48067b9571d1c8aa9cbef530ca6 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 18 Jul 2022 11:51:13 +0800
Subject: [PATCH] fix(mgb/imperative): fix the problem of occasional failure
 during testing of redis

GitOrigin-RevId: da1d55c70dfeee43e091d0fba3c5d1b2e82ed1b7
---
 imperative/src/impl/persistent_cache.cpp | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/imperative/src/impl/persistent_cache.cpp b/imperative/src/impl/persistent_cache.cpp
index 0d0c20a3..0c180af2 100644
--- a/imperative/src/impl/persistent_cache.cpp
+++ b/imperative/src/impl/persistent_cache.cpp
@@ -68,7 +68,14 @@ public:
         encode(category + '@' + key_str, redis_key_str, 24);
         auto result = m_client.get(m_prefix + redis_key_str);
         sync();
-        auto content = result.get();
+        decltype(result.get()) content;
+        int try_number = 3;
+        for (int i = 0; i < try_number; i++) {
+            content = result.get();
+            if (!content.is_null()) {
+                break;
+            }
+        }
         if (content.is_null()) {
             return None;
         }