From 8039e9c484cad9ca0b1ad69f8d38c783585138c8 Mon Sep 17 00:00:00 2001
From: Rustam Safargalin <rustam.safargalin@sifox.ru>
Date: Tue, 21 Aug 2018 13:35:15 +0300
Subject: [PATCH] Prevent silent dropping of messages when hwm is exceeded

---
 src/error_logger_lager_h.erl | 4 ++--
 src/lager_file_backend.erl   | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/error_logger_lager_h.erl b/src/error_logger_lager_h.erl
index a3613e2..0e7efe2 100644
--- a/src/error_logger_lager_h.erl
+++ b/src/error_logger_lager_h.erl
@@ -121,8 +121,8 @@ handle_event(Event, #state{sink=Sink, shaper=Shaper} = State) ->
                 "lager_error_logger_h dropped ~p messages in the last second that exceeded the limit of ~p messages/sec",
                 [Drop, Hwm]),
             eval_gl(Event, State#state{shaper=NewShaper});
-        {false, _, NewShaper} ->
-            {ok, State#state{shaper=NewShaper}}
+        {false, _, #lager_shaper{dropped=D} = NewShaper} ->
+            {ok, State#state{shaper=NewShaper#lager_shaper{dropped=D+1}}}
     end.
 
 handle_info({shaper_expired, ?MODULE}, #state{sink=Sink, shaper=Shaper} = State) ->
diff --git a/src/lager_file_backend.erl b/src/lager_file_backend.erl
index f861717..a9d4dc4 100644
--- a/src/lager_file_backend.erl
+++ b/src/lager_file_backend.erl
@@ -179,8 +179,8 @@ handle_event({log, Message},
                     {ok,write(NewState#state{shaper=NewShaper},
                         lager_msg:timestamp(Message), lager_msg:severity_as_int(Message),
                         Formatter:format(Message,FormatConfig))};
-                {false, _, NewShaper} ->
-                    {ok, State#state{shaper=NewShaper}}
+                {false, _, #lager_shaper{dropped=D} = NewShaper} ->
+                    {ok, State#state{shaper=NewShaper#lager_shaper{dropped=D+1}}}
             end;
         false ->
             {ok, State}