kvs/src/kvs_subscription.erl

-module(kvs_subscription).
%%-copyright('Synrc Research Center s.r.o.').

-include_lib("kvs/include/subscription.hrl").
-include_lib("kvs/include/metainfo.hrl").


-export([
  metainfo/0,
  subscribe/2,
  unsubscribe/2,
  subscriptions/1,
  subscribed/1,
  subscribed/2
]).


metainfo() ->
    #schema{name=kvs,tables=[
        #table{name=subscription,fields=record_info(fields,subscription),keys=[id,whom,who]}
    ]}.

subscribe(Who, Whom) -> kvs:put(#subscription{key={Who,Whom},who = Who, whom = Whom}).
unsubscribe(Who, Whom) ->
    case subscribed(Who, Whom) of
        true  -> kvs:delete(subscription, {Who, Whom});
        false -> skip end.

subscriptions(UId) -> kvs:index(subscription, who, UId).
subscribed(Who) -> kvs:index(subscription, whom, Who).
subscribed(Who, Whom) ->
    case kvs:get(subscription, {Who, Whom}) of
        {ok, _} -> true;
        _ -> false end.