anlicheng/iot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add spec

Browse Source
This commit is contained in:
安礼成 2023-02-20 11:37:55 +08:00
parent 3adb2d33ec
commit 1ad849c058
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/iot/src/model/host_model.erl
View File

@ -12,7 +12,7 @@
-include_lib("stdlib/include/qlc.hrl").
%% API
-export([get_host/1, get_hosts/2, add_host/1, change_status/2, delete/1, table_size/0, filter_hosts/3]).
-export([get_host/1, get_hosts/2, add_host/1, change_status/2, delete/1, table_size/0, find_hosts/3]).
get_host(HostId) when is_binary(HostId) ->
case mnesia:dirty_read(host, HostId) of
@ -23,6 +23,9 @@ get_host(HostId) when is_binary(HostId) ->
end.
%% 获取app信息
-spec get_hosts(Start :: integer(), Limit :: integer()) ->
{ok, Items :: list(), Stat :: maps()} |
{error, Reason :: any()}.
get_hosts(Start, Limit) when is_integer(Limit), is_integer(Start), Start >= 0, Limit > 0 ->
Fun = fun() ->
Q = qlc:q([E || E <- mnesia:table(host)]),
@ -46,7 +49,10 @@ get_hosts(Start, Limit) when is_integer(Limit), is_integer(Start), Start >= 0, L
Error
end.
filter_hosts(Pred, Start, Limit) when is_function(Pred, 1), is_integer(Limit), is_integer(Start), Start >= 0, Limit > 0 ->
-spec find_hosts(Pred :: fun((#host{}) -> boolean()), Start :: integer(), Limit :: integer()) ->
{ok, Items :: [#host{}], Num :: integer()} |
{error, Reason :: any()}.
find_hosts(Pred, Start, Limit) when is_function(Pred, 1), is_integer(Limit), is_integer(Start), Start >= 0, Limit > 0 ->
Fun = fun() ->
Q = qlc:q([E || E <- mnesia:table(host)]),
qlc:fold(fun(Host, Acc) ->
@ -60,11 +66,12 @@ filter_hosts(Pred, Start, Limit) when is_function(Pred, 1), is_integer(Limit), i
case mnesia:transaction(Fun) of
{atomic, Items} when is_list(Items) ->
Items1 = lists:reverse(Items),
{ok, lists:sublist(Items1, Start + 1, Limit)};
{ok, lists:sublist(Items1, Start + 1, Limit), length(Items1)};
{aborted, Error} ->
Error
{error, Error}
end.
-spec add_host(Host :: #host{}) -> ok | {error, Reason :: any()}.
add_host(Host = #host{}) ->
case mnesia:transaction(fun() -> mnesia:write(host, Host, write) end) of
{atomic, _} ->
@ -73,6 +80,7 @@ add_host(Host = #host{}) ->
{error, Error}
end.
-spec change_status(HostId :: binary(), Status :: integer()) -> ok | {error, Reason :: any()}.
change_status(HostId, Status) when is_binary(HostId), is_integer(Status) ->
Fun = fun() ->
case mnesia:read(host, HostId) of
@ -90,6 +98,7 @@ change_status(HostId, Status) when is_binary(HostId), is_integer(Status) ->
{error, Reason}
end.
-spec delete(HostId :: binary()) -> ok | {error, Reason :: any()}.
delete(HostId) when is_binary(HostId) ->
case mnesia:transaction(fun() -> mnesia:delete(host, HostId, write) end) of
{atomic, ok} ->