aboutsummaryrefslogtreecommitdiff
path: root/src/mod_pubsub/pubsub_index.erl
blob: 607b646eae45fcf0496539ec78392159d62a5231 (plain) (blame)
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
%%% ====================================================================
%%% ``The contents of this file are subject to the Erlang Public License,
%%% Version 1.1, (the "License"); you may not use this file except in
%%% compliance with the License. You should have received a copy of the
%%% Erlang Public License along with this software. If not, it can be
%%% retrieved via the world wide web at http://www.erlang.org/.
%%% 
%%% Software distributed under the License is distributed on an "AS IS"
%%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%%% the License for the specific language governing rights and limitations
%%% under the License.
%%% 
%%% The Initial Developer of the Original Code is ProcessOne.
%%% Portions created by ProcessOne are Copyright 2006-2011, ProcessOne
%%% All Rights Reserved.''
%%% This software is copyright 2006-2011, ProcessOne.
%%%
%%%
%%% @copyright 2006-2011 ProcessOne
%%% @author Christophe Romain <christophe.romain@process-one.net>
%%%   [http://www.process-one.net/]
%%% @version {@vsn}, {@date} {@time}
%%% @end
%%% ====================================================================

%% important note:
%% new/1 and free/2 MUST be called inside a transaction bloc

-module(pubsub_index).
-author('christophe.romain@process-one.net').

-include("pubsub.hrl").

-export([
	 init/3,
	 new/1,
	 free/2
	]).


-spec(init/3 ::
      (
	     Host       :: string(),
	     ServerHost :: string(),
	     Opts       :: [{Key::atom(), Value::term()}])
      -> 'ok'
	    ).

init(_Host, _ServerHost, _Opts) ->
    mnesia:create_table(pubsub_index,
			[{disc_copies, [node()]},
			 {attributes, record_info(fields, pubsub_index)}]).


-spec(new/1 ::
      (
	    Index::atom())
      -> Idx::integer()
	    ).

new(Index) ->
    case mnesia:read({pubsub_index, Index}) of
	[#pubsub_index{free = [], last = Last} = PubsubIndex] ->
	    Idx = Last + 1,
	    mnesia:write(PubsubIndex#pubsub_index{last = Idx}),
	    Idx;
	[#pubsub_index{free = [Idx|Free]} = PubsubIndex] ->
	    mnesia:write(PubsubIndex#pubsub_index{free = Free}),
	    Idx;
	_ ->
	    mnesia:write(#pubsub_index{index = Index, last = 1, free = []}),
	    1
    end.


-spec(free/2 ::
      (
	     Index :: atom(),
	     Idx   :: integer())
      -> 'ok'
	    ).

free(Index, Idx) ->
    case mnesia:read({pubsub_index, Index}) of
	[#pubsub_index{free = Free} = PubsubIndex] ->
	    mnesia:write(PubsubIndex#pubsub_index{free = [Idx|Free]});
	_ -> ok
    end.