store_kai.erl 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. -module(store_kai).
  2. -author('Maxim Sokhatsky').
  3. -copyright('Synrc Research Center s.r.o.').
  4. -include("config.hrl").
  5. -include("metainfo.hrl").
  6. -include_lib("stdlib/include/qlc.hrl").
  7. -compile(export_all).
  8. -record(data, { key, bucket, last_modified, vector_clocks, checksum, flags, value }).
  9. start() -> kai:start(), ok.
  10. stop() -> kai_store:stop(), ok.
  11. version() -> {version,"KVS KAI"}.
  12. join() -> initialize(), ok.
  13. join(Node) -> initialize(), ok.
  14. initialize() -> ok.
  15. dir() -> [{table,T}||T<-kvs:modules()].
  16. put(Records) when is_list(Records) -> lists:foreach(fun kai_put/1, Records);
  17. put(Record) -> kai_put(Record).
  18. kai_put(Record) ->
  19. Data = #data{key = element(2,Record), bucket = table_to_num(element(1,Record)),
  20. last_modified = now(), checksum = erlang:md5(term_to_binary(Record)),
  21. vector_clocks = vclock:fresh(), value = Record },
  22. kai_store:put(Data).
  23. update(Record, Object) -> ok.
  24. get(Tab, Key) ->
  25. Data = #data{key=Key,bucket=table_to_num(Tab)},
  26. kai_get(Data).
  27. kai_get(Data) ->
  28. case kai_store:get(Data) of
  29. #data{value=Value} -> {ok,Value};
  30. undefined -> {error,not_found};
  31. E -> {error,E} end.
  32. delete(Tab, Key) ->
  33. Data = #data{key=Key,bucket=table_to_num(Tab)},
  34. kai_delete(Data).
  35. kai_delete(Data) ->
  36. case kai_store:delete(Data) of
  37. ok -> ok;
  38. E -> {error,E} end.
  39. key_to_bin(Key) ->
  40. if is_integer(Key) -> erlang:list_to_binary(integer_to_list(Key));
  41. is_list(Key) -> erlang:list_to_binary(Key);
  42. is_atom(Key) -> erlang:list_to_binary(erlang:atom_to_list(Key));
  43. is_binary(Key) -> Key;
  44. true -> [ListKey] = io_lib:format("~p", [Key]), erlang:list_to_binary(ListKey) end.
  45. all(RecordName) ->
  46. {list_of_data,List} = kai_store:list(table_to_num(RecordName)),
  47. [ begin {ok,Val}=kai_get(Data),Val end || Data <- List ].
  48. all_by_index(Tab, IndexId, IndexVal) -> [].
  49. table_to_num(user) -> 10;
  50. table_to_num(subscription) -> 11;
  51. table_to_num(group) -> 12;
  52. table_to_num(id_seq) -> 13;
  53. table_to_num(product) -> 14;
  54. table_to_num(acl) -> 15;
  55. table_to_num(feed) -> 16;
  56. table_to_num(entry) -> 17;
  57. table_to_num(comment) -> 18;
  58. table_to_num(Name) -> case kvs:config(kvs,table_to_num) of
  59. [] -> unknown_table;
  60. Module -> Module:table_to_num(Name) end.