add email_in_limit to store email address that send count less than limit value
add email_out_limit to store email address that send count beyond limit value
the email queues email_in_limit/email_out_limit will be reset everyday
at 00:00, every time send mail to an email address, the value for the
email address in email_in_limit will +1, until the value up to the
limited value, the email address will be moved to email_out_limit, and
the email send to the address will be throw away.
Signed-off-by: Luan Shengde <shdluan(a)163.com>
---
.../email_redis_queue_limit.rb | 48 +++++++++++++++++++
1 file changed, 48 insertions(+)
create mode 100755 container/send-internet-mail/email_redis_queue_limit.rb
diff --git a/container/send-internet-mail/email_redis_queue_limit.rb b/container/send-internet-mail/email_redis_queue_limit.rb
new file mode 100755
index 0000000..0999983
--- /dev/null
+++ b/container/send-internet-mail/email_redis_queue_limit.rb
@@ -0,0 +1,48 @@
+#!/usr/bin/env ruby
+# SPDX-License-Identifier: MulanPSL-2.0+
+# Copyright (c) 2020 Huawei Technologies Co., Ltd. All rights reserved.
+# frozen_string_literal: true
+
+require 'json'
+require 'redis'
+
+# to check if the emails count has beyond the limit
+class EmailRateLimit
+ def initialize(mail_info)
+ @mail_info = mail_info
+ @redis = Redis.new('host' => REDIS_HOST, 'port' => REDIS_PORT)
+ end
+
+ def check_email_counts
+ email_to = @mail_info['to'].clone
+ email_cc = @mail_info['cc'].clone
+ @mail_info['to'] = check_emails(email_to)
+ @mail_info['cc'] = check_emails(email_cc)
+
+ return @mail_info
+ end
+
+ def check_emails(mail_list)
+ return if mail_list.nil? || mail_list.empty?
+
+ mail_list.clone.each do |email|
+ if @redis.hexists 'email_out_limit', email
+ mail_list -= [email]
+ next
+ end
+
+ if @redis.hexists 'email_in_limit', email
+ email_account = @redis.hget 'email_in_limit', email
+ @redis.hset 'email_in_limit', email, email_account.to_i + 1
+ else
+ @redis.hset 'email_in_limit', email, 1
+ end
+
+ if (@redis.hget 'email_in_limit', email).to_i >= 10
+ @redis.hdel 'email_in_limit', email
+ @redis.hset 'email_out_limit', email, 10
+ end
+ end
+ return mail_list
+ end
+end
--
2.23.0