From 3dbf6891c5318e27c502c7516364a52be012083b Mon Sep 17 00:00:00 2001
From: hujiabin <519660157@qq.com>
Date: Wed, 23 Oct 2024 14:16:09 +0800
Subject: [PATCH] 1

---
 webapp/models/UserInvitationCode.php | 3 ++-
 webapp/services/MissionService.php   | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/webapp/models/UserInvitationCode.php b/webapp/models/UserInvitationCode.php
index f2d67c05..cbd396f6 100644
--- a/webapp/models/UserInvitationCode.php
+++ b/webapp/models/UserInvitationCode.php
@@ -19,6 +19,7 @@ class UserInvitationCode extends BaseModel
     }
 
     public static function generateCode(){
+        $user = myself()->_getOrmUserInfo();
         SqlHelper::upsert(
             myself()->_getSelfMysql(),
             't_user_invitation_code',
@@ -30,7 +31,7 @@ class UserInvitationCode extends BaseModel
             ),
             array(
                 'account_id' => myself()->_getAccountId(),
-                'invitation_code' =>rand(100000,999999),
+                'invitation_code' =>$user['idx'],
                 'createtime' =>myself()->_getNowTime(),
                 'modifytime' =>myself()->_getNowTime(),
             )
diff --git a/webapp/services/MissionService.php b/webapp/services/MissionService.php
index 972efcdc..3f3a584f 100644
--- a/webapp/services/MissionService.php
+++ b/webapp/services/MissionService.php
@@ -527,8 +527,10 @@ class MissionService extends BaseService {
         {
             //邀请新人并绑定邀请码(邀请新人数)
             $myCodeDb = UserInvitationCode::findMyCode();
-            $count = UserInvitationCode::getMyCodeBindCount($myCodeDb['invitation_code']);
-            $missionDto['current'] = $count;
+            if ($myCodeDb){
+                $count = UserInvitationCode::getMyCodeBindCount($myCodeDb['invitation_code']);
+                $missionDto['current'] = $count;
+            }
         }
         break;
         case mt\Task::OFFER_REWARD_COND: