Fix errors after database connection interruption (drogonframework#2350)

Author: an-tao

orm_lib/src/mysql_impl/MysqlConnection.cc

@@ -433,7 +433,14 @@ void MysqlConnection::execSqlInLoop(
     assert(rcb);
     assert(!isWorking_);
     assert(!sql.empty());
-
+    if (status_ != ConnectStatus::Ok)
+    {
+        LOG_ERROR << "Connection is not ready";
+        auto exceptPtr =
+            std::make_exception_ptr(drogon::orm::BrokenConnection());
+        exceptCallback(exceptPtr);
+        return;
+    }
     callback_ = std::move(rcb);
     isWorking_ = true;
     exceptionCallback_ = std::move(exceptCallback);

orm_lib/src/postgresql_impl/PgBatchConnection.cc

@@ -241,6 +241,14 @@ void PgConnection::execSqlInLoop(
     std::function<void(const std::exception_ptr &)> &&exceptCallback)
 {
     LOG_TRACE << sql;
+    if (status_ != ConnectStatus::Ok)
+    {
+        LOG_ERROR << "Connection is not ready";
+        auto exceptPtr =
+            std::make_exception_ptr(drogon::orm::BrokenConnection());
+        exceptCallback(exceptPtr);
+        return;
+    }
     isWorking_ = true;
     batchSqlCommands_.emplace_back(
         std::make_shared<SqlCmd>(std::move(sql),

orm_lib/src/postgresql_impl/PgConnection.cc

@@ -230,6 +230,14 @@ void PgConnection::execSqlInLoop(
     assert(rcb);
     assert(!isWorking_);
     assert(!sql.empty());
+    if (status_ != ConnectStatus::Ok)
+    {
+        LOG_ERROR << "Connection is not ready";
+        auto exceptPtr =
+            std::make_exception_ptr(drogon::orm::BrokenConnection());
+        exceptCallback(exceptPtr);
+        return;
+    }
     sql_ = std::move(sql);
     callback_ = std::move(rcb);
     isWorking_ = true;

orm_lib/src/sqlite3_impl/Sqlite3Connection.cc

@@ -128,6 +128,7 @@ void Sqlite3Connection::init()
         else
         {
             sqlite3_extended_result_codes(tmp, true);
+            status_ = ConnectStatus::Ok;
             okCallback_(thisPtr);
         }
     });
@@ -167,6 +168,14 @@ void Sqlite3Connection::execSqlInQueue(
     const std::function<void(const std::exception_ptr &)> &exceptCallback)
 {
     LOG_TRACE << "sql:" << sql;
+    if (status_ != ConnectStatus::Ok)
+    {
+        LOG_ERROR << "Connection is not ready";
+        auto exceptPtr =
+            std::make_exception_ptr(drogon::orm::BrokenConnection());
+        exceptCallback(exceptPtr);
+        return;
+    }
     std::shared_ptr<sqlite3_stmt> stmtPtr;
     bool newStmt = false;
     if (paraNum > 0)
@@ -375,6 +384,7 @@ void Sqlite3Connection::disconnect()
             auto thisPtr = weakPtr.lock();
             if (!thisPtr)
                 return;
+            thisPtr->status_ = ConnectStatus::Bad;
             thisPtr->connectionPtr_.reset();
         }
         pro.set_value(1);