From ee7bca3ccf1820ac932225950d71e91974545720 Mon Sep 17 00:00:00 2001
From: daiao <358551898@qq.com>
Date: Wed, 17 Jul 2019 19:27:15 +0800
Subject: [PATCH] 1

---
 app/controllers/competition_teams_controller.rb | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/app/controllers/competition_teams_controller.rb b/app/controllers/competition_teams_controller.rb
index 3efaa1cf..bcb04e25 100644
--- a/app/controllers/competition_teams_controller.rb
+++ b/app/controllers/competition_teams_controller.rb
@@ -114,7 +114,9 @@ class CompetitionTeamsController < ApplicationController
         new_team.team_members.create!(user_id: user_id, role: 3, competition_id: @competition.id, is_teacher: 1)
       end
       # 创建时，记录是否是引流用户
-      record_agent_user_action
+      ip = request.remote_ip
+      ua = UserAgent.find_by_ip(ip)
+      ua.update_column(:type, UserAgent::COMPETITION) if ua
     end
   end
 
@@ -229,7 +231,9 @@ class CompetitionTeamsController < ApplicationController
 
       team.team_members.create!(user_id: User.current.id, role: 2, competition_id: @competition.id)
       # 创建时，记录是否是引流用户
-      record_agent_user_action
+      ip = request.remote_ip
+      ua = UserAgent.find_by_ip(ip)
+      ua.update_column(:type, UserAgent::COMPETITION) if ua
     end
   end