From 1e8049788d09177594b9804f16218fd38dd545d0 Mon Sep 17 00:00:00 2001
From: mayue <mayue@easycorp.ltd>
Date: Thu, 23 Jun 2022 09:35:54 +0800
Subject: [PATCH] * Optimize code.

---
 module/upgrade/model.php | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/module/upgrade/model.php b/module/upgrade/model.php
index 5cbf184e6b..519a6c500c 100644
--- a/module/upgrade/model.php
+++ b/module/upgrade/model.php
@@ -6449,10 +6449,11 @@ class upgradeModel extends model
      */
     public function addStoryViewPriv()
     {
-        $groupIdList = $this->dao->select('id')->from(TABLE_GROUPPRIV)
+        $groupIdList = $this->dao->select('group')->from(TABLE_GROUPPRIV)
             ->where('module')->eq('story')
             ->andWhere('method')->eq('view')
-            ->fetchPairs('id');
+            ->fetchPairs('group');
+
         foreach($groupIdList as $groupID)
         {
             $this->dao->insert(TABLE_GROUPPRIV)
@@ -6461,5 +6462,7 @@ class upgradeModel extends model
                 ->set('method')->eq('storyView')
                 ->exec();
         }
+
+        return true;
     }
 }