Mirror/dhtcrawler2
1
0
Fork 0
mirror of https://github.com/btdig/dhtcrawler2.git synced 2025-01-19 12:41:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

see master branch

Browse Source
This commit is contained in:
Kevin Lynx 2013-07-06 09:42:32 +08:00
parent 8f04efed5d
commit 1a9f38df49
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/hash_reader/db_hash_reader_sup.erl
View File

@ -69,4 +69,5 @@ config_default() ->
[{save_torrent, true},
{save_to_db, false},
{save_to_file, true},
{load_from_db, false},
{torrent_path, "torrents/"}].

14
src/local_torrent/loc_torrent_cache.erl
View File

@ -17,16 +17,20 @@ save(Conn, MagHash, Content) when is_list(MagHash), length(MagHash) == 40 ->
ok.
load(Conn, MagHash) when is_list(MagHash), length(MagHash) == 40 ->
LoadDB = config:get(load_from_db, false),
case load_from_file(MagHash) of
not_found ->
not_found when LoadDB ->
db_loc_torrent:load(Conn, MagHash);
not_found ->
not_found;
Content ->
Content
end.
%% TODO: put these file-realted codes to another module
save_to_file(MagHash, Content) ->
FileName = torrent_file_name(MagHash),
{Path, FileName} = torrent_file_name(MagHash),
filelib:ensure_dir(Path),
case file:write_file(FileName, Content) of
ok -> ok;
{error, Reason} ->
@ -34,7 +38,7 @@ save_to_file(MagHash, Content) ->
end.
load_from_file(MagHash) ->
FileName = torrent_file_name(MagHash),
{_, FileName} = torrent_file_name(MagHash),
case file:read_file(FileName) of
{ok, Content} ->
Content;
@ -47,5 +51,5 @@ torrent_file_name(MagHash) ->
Path = config:get(torrent_path, "torrents/"),
FullPath = Path ++ lists:sublist(MagHash, 1, 2) ++ "/" ++
lists:sublist(MagHash, 3, 2) ++ "/",
filelib:ensure_dir(FullPath),
FullPath ++ MagHash ++ ".torrent".
{FullPath, FullPath ++ MagHash ++ ".torrent"}.