121 lines
4.4 KiB
Erlang
121 lines
4.4 KiB
Erlang
%% -------------------------------------------------------------------
|
|
%%
|
|
%% bdberl: Port Driver Stress tests
|
|
%% Copyright (c) 2008 The Hive. All rights reserved.
|
|
%%
|
|
%% Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
%% of this software and associated documentation files (the "Software"), to deal
|
|
%% in the Software without restriction, including without limitation the rights
|
|
%% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
%% copies of the Software, and to permit persons to whom the Software is
|
|
%% furnished to do so, subject to the following conditions:
|
|
%%
|
|
%% The above copyright notice and this permission notice shall be included in
|
|
%% all copies or substantial portions of the Software.
|
|
%%
|
|
%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
%% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
%% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
%% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
%% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
%% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
%% THE SOFTWARE.
|
|
%%
|
|
%% -------------------------------------------------------------------
|
|
-module(stress_SUITE).
|
|
-compile(export_all).
|
|
-include_lib("common_test/include/ct.hrl").
|
|
|
|
%% NOTE: all of the tests are set for a low number of iterations to guarantee
|
|
%% that they all pass and run in a reasonable amount of time. That kinda defeats
|
|
%% the purpose of the test, tho. Work is ongoing to make this a useful test suite.
|
|
|
|
all() ->
|
|
[rewrite_array_test,
|
|
rewrite_bytes_test,
|
|
write_array_test,
|
|
write_bytes_test].
|
|
|
|
dbconfig(Config) ->
|
|
Cfg = [
|
|
{set_data_dir, ?config(priv_dir, Config)},
|
|
{set_flags, 'DB_TXN_WRITE_NOSYNC'},
|
|
{set_cachesize, '0 536870912 1'},
|
|
{set_lg_max, '1048576000'},
|
|
{set_lg_bsize, '5368709120'},
|
|
{set_log_config, 'DB_LOG_IN_MEMORY'}
|
|
],
|
|
list_to_binary(lists:flatten([io_lib:format("~s ~s\n", [K,V]) || {K, V} <- Cfg])).
|
|
|
|
init_per_suite(Config) ->
|
|
DbHome = ?config(priv_dir, Config),
|
|
os:putenv("DB_HOME", DbHome),
|
|
ok = file:write_file(DbHome ++ "DB_CONFIG", dbconfig(Config)),
|
|
Config.
|
|
|
|
end_per_suite(_Config) ->
|
|
ok.
|
|
|
|
init_per_testcase(TestCase, Config) ->
|
|
Size = 1024 * 1024,
|
|
Chunk = crypto:rand_bytes(Size),
|
|
Name = io_lib:format("~p.db", [TestCase]),
|
|
{ok, Db} = bdberl:open(Name, hash),
|
|
[{size, Size}, {chunk, Chunk}, {db, Db}|Config].
|
|
|
|
end_per_testcase(_TestCase, Config) ->
|
|
bdberl:close(?config(db, Config)),
|
|
ok.
|
|
|
|
%%---------------------------------------------------------------------------
|
|
|
|
rewrite_array_test(Config) ->
|
|
%% If you try to run this one for more than 2K iterations than the Erlang
|
|
%% VM will die with a memory allocation error when creating the binary.
|
|
ct:print("Running rewrite_array test for 2000 iterations..."),
|
|
Chunk = ?config(chunk, Config),
|
|
rewrite_array(?config(db, Config), Chunk, [Chunk], 20).
|
|
|
|
rewrite_array(_Db, _Block, _Bytes, 0) ->
|
|
ok;
|
|
rewrite_array(Db, Block, Bytes, Iter) ->
|
|
bdberl:put(Db, 1, Bytes),
|
|
rewrite_array(Db, Block, [Block|Bytes], Iter - 1).
|
|
|
|
%%---------------------------------------------------------------------------
|
|
|
|
rewrite_bytes_test(Config) ->
|
|
ct:print("Running rewrite_bytes test for 2500 iterations..."),
|
|
rewrite_bytes(?config(db, Config), ?config(chunk, Config), 25).
|
|
|
|
rewrite_bytes(_Db, _Bytes, 0) ->
|
|
ok;
|
|
rewrite_bytes(Db, Bytes, Iter) ->
|
|
bdberl:put(Db, 1, Bytes),
|
|
rewrite_bytes(Db, Bytes, Iter - 1).
|
|
|
|
%%---------------------------------------------------------------------------
|
|
|
|
write_array_test(Config) ->
|
|
ct:print("Running write_array test for 150 iterations..."),
|
|
Chunk = ?config(chunk, Config),
|
|
write_array(?config(db, Config), Chunk, [Chunk], 15).
|
|
|
|
write_array(_Db, _Block, _Bytes, 0) ->
|
|
ok;
|
|
write_array(Db, Block, Bytes, Iter) ->
|
|
bdberl:put(Db, Iter, Bytes),
|
|
write_array(Db, Block, [Block|Bytes], Iter - 1).
|
|
|
|
%%---------------------------------------------------------------------------
|
|
|
|
write_bytes_test(Config) ->
|
|
ct:print("Running write_bytes test for 2500 iterations..."),
|
|
write_bytes(?config(db, Config), ?config(chunk, Config), 25).
|
|
|
|
write_bytes(_Db, _Bytes, 0) ->
|
|
ok;
|
|
write_bytes(Db, Bytes, Iter) ->
|
|
bdberl:put(Db, Iter, Bytes),
|
|
write_bytes(Db, Bytes, Iter - 1).
|