| % Licensed under the Apache License, Version 2.0 (the "License"); you may not |
| % use this file except in compliance with the License. You may obtain a copy of |
| % the License at |
| % |
| % http://www.apache.org/licenses/LICENSE-2.0 |
| % |
| % Unless required by applicable law or agreed to in writing, software |
| % distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| % WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| % License for the specific language governing permissions and limitations under |
| % the License. |
| |
| -module(couchdb_os_proc_pool). |
| |
| -include_lib("couch/include/couch_eunit.hrl"). |
| -include_lib("couch/include/couch_db.hrl"). |
| |
| -define(TIMEOUT, 3000). |
| |
| |
| start() -> |
| Ctx = test_util:start_couch(), |
| config:set("query_server_config", "os_process_limit", "3", false), |
| timer:sleep(100), %% we need to wait to let gen_server:cast finish |
| Ctx. |
| |
| |
| os_proc_pool_test_() -> |
| { |
| "OS processes pool tests", |
| { |
| setup, |
| fun start/0, fun test_util:stop_couch/1, |
| [ |
| should_block_new_proc_on_full_pool(), |
| should_free_slot_on_proc_unexpected_exit() |
| ] |
| } |
| }. |
| |
| |
| should_block_new_proc_on_full_pool() -> |
| ?_test(begin |
| Client1 = spawn_client(), |
| Client2 = spawn_client(), |
| Client3 = spawn_client(), |
| |
| ?assertEqual(ok, ping_client(Client1)), |
| ?assertEqual(ok, ping_client(Client2)), |
| ?assertEqual(ok, ping_client(Client3)), |
| |
| Proc1 = get_client_proc(Client1, "1"), |
| Proc2 = get_client_proc(Client2, "2"), |
| Proc3 = get_client_proc(Client3, "3"), |
| |
| ?assertNotEqual(Proc1, Proc2), |
| ?assertNotEqual(Proc2, Proc3), |
| ?assertNotEqual(Proc3, Proc1), |
| |
| Client4 = spawn_client(), |
| ?assertEqual(timeout, ping_client(Client4)), |
| |
| ?assertEqual(ok, stop_client(Client1)), |
| ?assertEqual(ok, ping_client(Client4)), |
| |
| Proc4 = get_client_proc(Client4, "4"), |
| |
| ?assertEqual(Proc1#proc.pid, Proc4#proc.pid), |
| ?assertNotEqual(Proc1#proc.client, Proc4#proc.client), |
| |
| lists:map(fun(C) -> |
| ?assertEqual(ok, stop_client(C)) |
| end, [Client2, Client3, Client4]) |
| end). |
| |
| |
| should_free_slot_on_proc_unexpected_exit() -> |
| ?_test(begin |
| Client1 = spawn_client(), |
| Client2 = spawn_client(), |
| Client3 = spawn_client(), |
| |
| ?assertEqual(ok, ping_client(Client1)), |
| ?assertEqual(ok, ping_client(Client2)), |
| ?assertEqual(ok, ping_client(Client3)), |
| |
| Proc1 = get_client_proc(Client1, "1"), |
| Proc2 = get_client_proc(Client2, "2"), |
| Proc3 = get_client_proc(Client3, "3"), |
| |
| ?assertNotEqual(Proc1#proc.pid, Proc2#proc.pid), |
| ?assertNotEqual(Proc1#proc.client, Proc2#proc.client), |
| ?assertNotEqual(Proc2#proc.pid, Proc3#proc.pid), |
| ?assertNotEqual(Proc2#proc.client, Proc3#proc.client), |
| ?assertNotEqual(Proc3#proc.pid, Proc1#proc.pid), |
| ?assertNotEqual(Proc3#proc.client, Proc1#proc.client), |
| |
| ?assertEqual(ok, kill_client(Client1)), |
| |
| Client4 = spawn_client(), |
| ?assertEqual(ok, ping_client(Client4)), |
| |
| Proc4 = get_client_proc(Client4, "4"), |
| |
| ?assertEqual(Proc4#proc.pid, Proc1#proc.pid), |
| ?assertNotEqual(Proc4#proc.client, Proc1#proc.client), |
| ?assertNotEqual(Proc2#proc.pid, Proc4#proc.pid), |
| ?assertNotEqual(Proc2#proc.client, Proc4#proc.client), |
| ?assertNotEqual(Proc3#proc.pid, Proc4#proc.pid), |
| ?assertNotEqual(Proc3#proc.client, Proc4#proc.client), |
| |
| lists:map(fun(C) -> |
| ?assertEqual(ok, stop_client(C)) |
| end, [Client2, Client3, Client4]) |
| end). |
| |
| |
| spawn_client() -> |
| Parent = self(), |
| Ref = make_ref(), |
| Pid = spawn(fun() -> |
| Proc = couch_query_servers:get_os_process(<<"javascript">>), |
| loop(Parent, Ref, Proc) |
| end), |
| {Pid, Ref}. |
| |
| ping_client({Pid, Ref}) -> |
| Pid ! ping, |
| receive |
| {pong, Ref} -> |
| ok |
| after ?TIMEOUT -> |
| timeout |
| end. |
| |
| get_client_proc({Pid, Ref}, ClientName) -> |
| Pid ! get_proc, |
| receive |
| {proc, Ref, Proc} -> Proc |
| after ?TIMEOUT -> |
| erlang:error({assertion_failed, |
| [{module, ?MODULE}, |
| {line, ?LINE}, |
| {reason, "Timeout getting client " |
| ++ ClientName ++ " proc"}]}) |
| end. |
| |
| stop_client({Pid, Ref}) -> |
| Pid ! stop, |
| receive |
| {stop, Ref} -> |
| ok |
| after ?TIMEOUT -> |
| timeout |
| end. |
| |
| kill_client({Pid, Ref}) -> |
| Pid ! die, |
| receive |
| {die, Ref} -> |
| ok |
| after ?TIMEOUT -> |
| timeout |
| end. |
| |
| loop(Parent, Ref, Proc) -> |
| receive |
| ping -> |
| Parent ! {pong, Ref}, |
| loop(Parent, Ref, Proc); |
| get_proc -> |
| Parent ! {proc, Ref, Proc}, |
| loop(Parent, Ref, Proc); |
| stop -> |
| couch_query_servers:ret_os_process(Proc), |
| Parent ! {stop, Ref}; |
| die -> |
| Parent ! {die, Ref}, |
| exit(some_error) |
| end. |