参考文献校对升级

application/common/mq/ReferenceCheckArticleWorker.php

@@ -3,6 +3,7 @@
 namespace app\common\mq;
 
 use think\Db;
+use app\common\DbReconnectHelper;
 use app\common\ReferenceCheckService;
 
 /**
@@ -25,6 +26,7 @@ class ReferenceCheckArticleWorker
 
   public function handleMessage(array $payload)
   {
+    DbReconnectHelper::ensure();
     $pArticleId = intval(isset($payload['p_article_id']) ? $payload['p_article_id'] : 0);
     $batchId = intval(isset($payload['batch_id']) ? $payload['batch_id'] : 0);
     if ($pArticleId <= 0 || $batchId <= 0) {
@@ -115,6 +117,7 @@ class ReferenceCheckArticleWorker
    */
   private function processOneRow($checkId, array $row)
   {
+    DbReconnectHelper::ensure();
     $claimed = Db::name('article_reference_check_result')
       ->where('id', intval($checkId))
       ->where('queue_status', ReferenceCheckService::QUEUE_PENDING)
@@ -134,6 +137,7 @@ class ReferenceCheckArticleWorker
       return 'ok';
     } catch (\Exception $e) {
       $this->svc->log('ReferenceCheckArticleWorker check_id=' . $checkId . ' err=' . $e->getMessage());
+      DbReconnectHelper::ensure();
       if ($retryCount < ReferenceCheckService::QUEUE_MAX_RETRY) {
         $this->svc->markQueueRuntime($checkId, ReferenceCheckService::QUEUE_PENDING, $retryCount + 1);
         return $this->processOneRow($checkId, array_merge($row, ['retry_count' => $retryCount + 1]));