新增：发送邮件及短信次数控制

app/services/info_risk_control_service.rb

@@ -0,0 +1,77 @@
+class InfoRiskControlService < ApplicationService
+
+  attr_reader :receiver, :remote_ip 
+  attr_accessor :status, :message
+
+
+  def initialize(receiver="", remote_ip="0.0.0.0")
+    @receiver = receiver
+    @remote_ip = remote_ip 
+    @status = 1
+    @message = ""
+  end
+
+  def call
+    if receiver == ""
+      remote_ip_minute_risk_control
+      remote_ip_risk_control if @status = 1
+    else  
+      remote_ip_minute_risk_control
+      remote_ip_risk_control if @status = 1
+      minute_risk_control 
+      day_risk_control if @status = 1
+    end
+
+    return @status, @message
+  end
+
+  private 
+  def remote_ip_minute_risk_control 
+    result = Rails.cache.read("InfoRiskControlService-RemoteIp-Minute-#{remote_ip}")
+    if result.present? 
+      @status = 0 
+      @message = "您的请求过于频繁，请稍后再试"
+    else 
+      Rails.cache.write("InfoRiskControlService-RemoteIp-Minute-#{remote_ip}", 1, expires_in: 1.minute)
+    end
+  end 
+
+  def remote_ip_risk_control 
+    result = Rails.cache.read("InfoRiskControlService-RemoteIp-#{remote_ip}")
+    if result.present? 
+      if result.to_i > 20
+        @status = 0 
+        @message = "暂时无法请求，请稍后再试"
+      else 
+        Rails.cache.write("InfoRiskControlService-RemoteIp-#{remote_ip}", result.to_i + 1)
+      end
+    else 
+      Rails.cache.write("InfoRiskControlService-RemoteIp-#{remote_ip}", 1, expires_in: 1.day)
+    end
+  end
+
+  def minute_risk_control 
+    result = Rails.cache.read("InfoRiskControlService-Minute-#{receiver}")
+    if result.present?  
+      @status = 0 
+      @message = "您的请求过于频繁，请稍后再试" 
+    else 
+      Rails.cache.write("InfoRiskControlService-Minute-#{receiver}", 1, expires_in: 1.minute)
+    end
+  end
+
+  def day_risk_control 
+    result = Rails.cache.read("InfoRiskControlService-Day-#{receiver}")
+    if result.present? 
+      if result.to_i > 10
+        @status = 0 
+        @message = "您的请求过于频繁，请稍后再试"
+      else 
+        Rails.cache.write("InfoRiskControlService-Day-#{receiver}", result.to_i + 1)
+      end
+    else 
+      Rails.cache.write("InfoRiskControlService-Day-#{receiver}", 1, expires_in: 1.days)
+    end 
+  end
+
+end