kvs/src/store_kai.erl

-module(store_kai).
-author('Maxim Sokhatsky <maxim@synrc.com>').
-copyright('Synrc Research Center s.r.o.').
-include_lib("kai/include/kai.hrl").
-include_lib("kvs/include/config.hrl").
-include_lib("kvs/include/users.hrl").
-include_lib("kvs/include/groups.hrl").
-include_lib("kvs/include/feeds.hrl").
-include_lib("kvs/include/acls.hrl").
-include_lib("kvs/include/invites.hrl").
-include_lib("kvs/include/meetings.hrl").
-include_lib("kvs/include/membership.hrl").
-include_lib("kvs/include/payments.hrl").
-include_lib("kvs/include/purchases.hrl").
-include_lib("kvs/include/accounts.hrl").
-include_lib("stdlib/include/qlc.hrl").
-compile(export_all).

start() -> kai_store:start(), ok.
stop() -> kai_store:stop(), ok.
version() -> {version,"KVS KAI PURE XEN"}.
join() -> initialize(), ok.
join(Node) -> initialize(), ok.
initialize() -> ok.
dir() -> kvs:modules().

put(Records) when is_list(Records) -> lists:foreach(fun kai_put/1, Records);
put(Record) -> kai_put(Record).

kai_put(Record) ->
    Data = #data{key = element(2,Record), bucket = table_to_num(element(1,Record)),
        last_modified = now(), checksum = erlang:md5(term_to_binary(Record)),
        vector_clocks = vclock:fresh(), value = Record },
    kai_store:put(Data).

update(Record, Object) -> ok.

get(Tab, Key) ->
    Data = #data{key=Key,bucket=table_to_num(Tab)},
    kai_get(Data).

kai_get(Data) ->
    case kai_store:get(Data) of
         #data{value=Value} -> Value;
         undefined -> {error,not_found};
         E -> {error,E} end.

delete(Tab, Key) ->
    ok.

key_to_bin(Key) ->
    if is_integer(Key) -> erlang:list_to_binary(integer_to_list(Key));
       is_list(Key) -> erlang:list_to_binary(Key);
       is_atom(Key) -> erlang:list_to_binary(erlang:atom_to_list(Key));
       is_binary(Key) -> Key;
       true ->  [ListKey] = io_lib:format("~p", [Key]), erlang:list_to_binary(ListKey) end.

all(RecordName) ->
    {list_of_data,List} = kai_store:list(table_to_num(RecordName)),
    [ kai_get(Data) || Data <- List ].

all_by_index(Tab, IndexId, IndexVal) -> [].

% index funs

products(UId) -> all_by_index(user_product, <<"user_bin">>, list_to_binary(UId)).
subscriptions(UId) -> all_by_index(subsciption, <<"subs_who_bin">>, list_to_binary(UId)).
subscribed(Who) -> all_by_index(subscription, <<"subs_whom_bin">>, list_to_binary(Who)).
participate(UserName) -> all_by_index(group_subscription, <<"who_bin">>, UserName).
members(GroupName) -> all_by_index(group_subscription, <<"where_bin">>, GroupName).
user_tournaments(UId) -> all_by_index(play_record, <<"play_record_who_bin">>, list_to_binary(UId)).
tournament_users(TId) -> all_by_index(play_record, <<"play_record_tournament_bin">>, list_to_binary(integer_to_list(TId))).
author_comments(Who) ->
    EIDs = [E || #comment{entry_id=E} <- all_by_index(comment,<<"author_bin">>, Who) ],
    lists:flatten([ all_by_index(entry,<<"entry_bin">>,EID) || EID <- EIDs]).

table_to_num(user) -> 10;
table_to_num(user_status) -> 20;
table_to_num(subscription) -> 30;
table_to_num(group) -> 40;
table_to_num(group_subscription) -> 50;
table_to_num(payment) -> 60;
table_to_num(user_payment) -> 70;
table_to_num(account) -> 80;
table_to_num(transaction) -> 90;
table_to_num(id_seq) -> 100;
table_to_num(team) -> 110;
table_to_num(membership) -> 120;
table_to_num(product) -> 130;
table_to_num(product_category) -> 140;
table_to_num(user_product) -> 150;
table_to_num(acl) -> 160;
table_to_num(acl_entry) -> 170;
table_to_num(feed) -> 180;
table_to_num(entry) -> 190;
table_to_num(comment) -> 200.