Retry if failed

enums · Aug 11, 2021 · 0141897 · 0141897
1 parent 4351b3a
commit 0141897
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/Sources/Heze/Model/HezeMySQLDatabase.swift b/Sources/Heze/Model/HezeMySQLDatabase.swift
@@ -60,7 +60,7 @@ open class HezeMySQLDatabase: HezeObject, HezeDatabaseImpl {
     }
 
     public func query(_ sql: HezeSQLStatement) -> [HezeDatabaseRecord]? {
-        guard mysql.query(statement: sql) else {
+        guard queryWithRetryIfFailed(sql) else {
             HezeLogger.shared.debug("[SQL][Failed] \(sql)")
             return nil
         }
@@ -77,7 +77,7 @@ open class HezeMySQLDatabase: HezeObject, HezeDatabaseImpl {
 
     public func query(_ sqls: [HezeSQLStatement]) -> [[HezeDatabaseRecord]]? {
         let sql = sqls.joined(separator: " ")
-        guard mysql.query(statement: sql) else {
+        guard queryWithRetryIfFailed(sql) else {
             HezeLogger.shared.debug("[SQL][Failed] \(sql)")
             return nil
         }
@@ -175,5 +175,16 @@ open class HezeMySQLDatabase: HezeObject, HezeDatabaseImpl {
         return query(sql) != nil
     }
 
+    private func queryWithRetryIfFailed(_ statement: String) -> Bool {
+        if mysql.query(statement: statement) {
+            return true
+        }
+
+        guard mysql.ping() else {
+            return false
+        }
+
+        return mysql.query(statement: statement)
+    }
 
 }