From 9ced678f910f2b56d79ae2f1ecc7e638a2083831 Mon Sep 17 00:00:00 2001
From: yystopf <yystopf@163.com>
Date: Thu, 31 Mar 2022 13:03:54 +0800
Subject: [PATCH] fix: user trace token

---
 app/models/user.rb | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/app/models/user.rb b/app/models/user.rb
index 903b60c38..d661ab591 100644
--- a/app/models/user.rb
+++ b/app/models/user.rb
@@ -783,14 +783,16 @@ class User < Owner
         trace_user.refresh_token
         trace_user.save
       end
+      return trace_user.token
     else
-      trace_user = TraceUser.new
-      trace_user.build_self_data
-      trace_user.build_token 
-      trace_user.save
-    end
+      tu = TraceUser.new
+      tu.user = self
+      tu.build_self_data
+      tu.build_token 
+      tu.save
 
-    trace_user.token
+      return tu.token
+    end
   end
 
   protected