summaryrefslogtreecommitdiff
path: root/src/couch_log/test/eunit/couch_log_writer_ets.erl
blob: 385aa96762e530060464207f264444cbbbba29bd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
% Licensed under the Apache License, Version 2.0 (the "License"); you may not
% use this file except in compliance with the License. You may obtain a copy of
% the License at
%
%   http://www.apache.org/licenses/LICENSE-2.0
%
% Unless required by applicable law or agreed to in writing, software
% distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
% WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
% License for the specific language governing permissions and limitations under
% the License.

-module(couch_log_writer_ets).
-behaviour(couch_log_writer).

-export([
    init/0,
    terminate/2,
    write/2
]).

-include("couch_log.hrl").

init() ->
    ets:new(?COUCH_LOG_TEST_TABLE, [named_table, public, ordered_set]),
    {ok, 0}.

terminate(_, _St) ->
    ets:delete(?COUCH_LOG_TEST_TABLE),
    ok.

write(Entry0, St) ->
    Entry1 = couch_log_util:maybe_format_type(Entry0),
    Entry = Entry1#log_entry{
        msg = lists:flatten(Entry1#log_entry.msg),
        time_stamp = lists:flatten(Entry1#log_entry.time_stamp)
    },
    Ignored = application:get_env(couch_log, ignored_pids, []),
    case lists:member(Entry#log_entry.pid, Ignored) of
        true ->
            {ok, St};
        false ->
            ets:insert(?COUCH_LOG_TEST_TABLE, {St, Entry}),
            {ok, St + 1}
    end.