ggs_db.erl only init and stop left to do.

ggs_db.test.erl only init and stop left todo.
Everything else is done with these two files.
This commit is contained in:
Kallfaktorn 2011-02-22 01:09:55 +01:00
parent 32d9622da8
commit bf4d743923
3 changed files with 1 additions and 1 deletions

View file

@ -1,86 +0,0 @@
%%%%----------------------------------------------------
%%% @author Mattias Pettersson <mattiaspgames@gmail.com>
%%% @copyright 2011 Mattias Pettersson
%%% @doc Database for runtime game variable storage.
%%% @end
-module(ggs_db).
-import(mnesia).
%-compile({no_auto_import,[length/2]}).
-export([init/0,setItem/4,getItem/3,removeItem/3,key/3,clear/2,length/2]).
-include("ggs_db.hrl").
%%-----------------------------------------------------
%% Creation
%%-----------------------------------------------------
init() ->
mnesia:create_table(data, [{attributes, record_info(fields, data)}]).
%%-----------------------------------------------------
%% Test
%%-----------------------------------------------------
%test_data() ->
% set(0, "Hello"),
% get(0).
%%-----------------------------------------------------
%% Insertions
%%-----------------------------------------------------
setItem(Db,Ns,Key,Value) ->
Fun = fun() ->
Data = #data{key = {Db,Ns,Key}, value = Value},
mnesia:write(Data)
end,
mnesia:transaction(Fun).
%%-----------------------------------------------------
%% Deletions
%%-----------------------------------------------------
removeItem(Db,Ns,Key) ->
Fun = fun() ->
mnesia:delete({data,{Db,Ns,Key}})
end,
mnesia:transaction(Fun).
clear(Db,Ns) ->
Fun = fun() ->
Keys = mnesia:all_keys(data),
Rest = lists:filter(fun({A,B,_}) -> ((A==Db) and (B==Ns)) end, Keys),
lists:map(fun({A,B,C}) -> removeItem(A,B,C) end, Rest)
end,
{atomic, Ret} = mnesia:transaction(Fun),
Ret.
%%-----------------------------------------------------
%% Querries
%%-----------------------------------------------------
getItem(Db,Ns,Key) ->
Fun = fun() ->
[Data] = mnesia:read(data, {Db,Ns,Key}),
Data#data.value
end,
mnesia:transaction(Fun).
length(Db,Ns) ->
Fun = fun() ->
Keys = mnesia:all_keys(data),
length(lists:filter(fun({A,B,_}) -> ((A==Db) and (B==Ns)) end, Keys))
end,
{atomic, Ret} = mnesia:transaction(Fun),
Ret.
key(Db,Ns,Position) ->
Fun = fun() ->
Keys = mnesia:all_keys(data),
Rest = lists:filter(fun({A,B,_}) -> ((A==Db) and (B==Ns)) end, Keys),
lists:nth(Position, Rest)
end,
{atomic, Ret} = mnesia:transaction(Fun),
Ret.

View file

@ -1,44 +0,0 @@
-module(ggs_db_test).
%-compile({no_auto_import,[get/1,set/2]}).
-include_lib("eunit/include/eunit.hrl").
%ggs_db_test_() ->
% {spawn,
% {setup, fun setup/0, fun cleanup/1,[ fun ggs_db_test/0 ]}
% }.
%Key should be a tuple of two elements
getItem_setItem_test() ->
mnesia:start(),
ggs_db:init(),
ggs_db:setItem("dbname","nsname","keyname","Hello"),
?assertMatch({atomic,"Hello"},ggs_db:getItem("dbname","nsname","keyname")).
length_test() ->
ggs_db:setItem(1,1,2,"112"),
ggs_db:setItem(1,2,2,"122"),
ggs_db:setItem(1,1,3,"113"),
ggs_db:setItem(1,1,4,"114"),
?assertEqual(ggs_db:length(1,1), 3).
removeItem_test() ->
ggs_db:removeItem(1,1,4),
?assertNot(ggs_db:getItem(1,1,4) =:= {atomic,"114"}).
key_test() ->
?assert(ggs_db:key(1,1,2) =:= {1,1,3}).
clear_test() ->
ggs_db:clear(1,1),
?assert(ggs_db:length(1,1) =:= 0),
mnesia:stop().
%tests() ->
% ggs_db_test().
%ggs_db_test() ->
% ggs_db:set(0, "Hello"),
%io:format("~s",[Val]),
% ?assertMatch({atomic,"Hello"},ggs_db:get(0)).