From ef95c59e5962027dadf26299672817872342c5ef Mon Sep 17 00:00:00 2001
From: jingquan huang <huang.jingquan@163.com>
Date: Tue, 16 Jul 2019 18:00:54 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=B0=E5=BD=95=E7=94=A8=E6=88=B7=E6=B5=81?=
 =?UTF-8?q?=E9=87=8F=E8=A1=8C=E6=96=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/mobile/apis/memos.rb                    |  8 ++++++++
 app/models/user_agent.rb                        |  3 +++
 db/migrate/20190716073605_create_user_agents.rb | 12 ++++++++++++
 spec/factories/user_agents.rb                   |  7 +++++++
 spec/models/user_agent_spec.rb                  |  5 +++++
 5 files changed, 35 insertions(+)
 create mode 100644 app/models/user_agent.rb
 create mode 100644 db/migrate/20190716073605_create_user_agents.rb
 create mode 100644 spec/factories/user_agents.rb
 create mode 100644 spec/models/user_agent_spec.rb

diff --git a/app/api/mobile/apis/memos.rb b/app/api/mobile/apis/memos.rb
index c8c0041a..04191f53 100644
--- a/app/api/mobile/apis/memos.rb
+++ b/app/api/mobile/apis/memos.rb
@@ -29,6 +29,14 @@ module Mobile
           requires :id, type: Integer, desc: "帖子ID"
         end
         get ':id' do
+          # 记录用户流量源
+          if params[:edu]
+            ip = request.host
+            ua = UserAgent.find_by_ip(ip)
+            unless ua
+              UserAgent.create(:key => params[:edu].strip, :ip => request.host)
+            end
+          end
           MemosService.new.show params, current_user
         end
 
diff --git a/app/models/user_agent.rb b/app/models/user_agent.rb
new file mode 100644
index 00000000..b71c8449
--- /dev/null
+++ b/app/models/user_agent.rb
@@ -0,0 +1,3 @@
+class UserAgent < ActiveRecord::Base
+
+end
diff --git a/db/migrate/20190716073605_create_user_agents.rb b/db/migrate/20190716073605_create_user_agents.rb
new file mode 100644
index 00000000..22bc181e
--- /dev/null
+++ b/db/migrate/20190716073605_create_user_agents.rb
@@ -0,0 +1,12 @@
+class CreateUserAgents < ActiveRecord::Migration
+  def change
+    create_table :user_agents do |t|
+      t.string :type
+      t.string :key
+      t.string :ip
+
+      t.timestamps
+    end
+    add_index(:user_agents, :ip, :unique => true)
+  end
+end
diff --git a/spec/factories/user_agents.rb b/spec/factories/user_agents.rb
new file mode 100644
index 00000000..c8afcb41
--- /dev/null
+++ b/spec/factories/user_agents.rb
@@ -0,0 +1,7 @@
+FactoryGirl.define do
+  factory :user_agent do
+    type 1
+    key "MyString"
+    ip "MyString"
+  end
+end
diff --git a/spec/models/user_agent_spec.rb b/spec/models/user_agent_spec.rb
new file mode 100644
index 00000000..20fd5cf3
--- /dev/null
+++ b/spec/models/user_agent_spec.rb
@@ -0,0 +1,5 @@
+require 'rails_helper'
+
+RSpec.describe UserAgent, :type => :model do
+  pending "add some examples to (or delete) #{__FILE__}"
+end