mysql_client

%%
%% Copyright (C) 2010-2014 by krasnop@bellsouth.net (Alexei Krasnopolski)
%%
%% 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. 
%%

%% @hidden
%% @since 2011-05-23
%% @copyright 2010-2014 Alexei Krasnopolski
%% @author Alexei Krasnopolski <krasnop@bellsouth.net> [http://crasnopolski.com/]
%% @version {@version}
%% @doc This module is running unit tests for concurrent connections.

-module(concurrent_connection_eunit).

%%
%% Include files
%%
-include_lib("eunit/include/eunit.hrl").
-include("client_records.hrl").
-include("mysql_types.hrl").
-include("test.hrl").

%%
%% Import modules
%%

%%
%% Exported Functions
%%

%%
%% API Functions
%%

run_test_() ->
	[ 
    { foreachx, 
      fun testing:do_setup/1, 
      fun testing:do_cleanup/2,
      [ 
		    {{plain, run_consumer}, fun run_consumer/2}
      ]
    }
	].

%%
%% Local Functions
%%

%% @spec run_consumer(X, C) -> any()
%% @doc The function establishes connection to database and recursively process messages
%% from database.
%%
run_consumer(X, C) -> {timeout, 100, fun() ->
	my:new_datasource( #datasource{
		name = serializable_ds,
		host = ?TEST_SERVER_HOST_NAME, 
		port = ?TEST_SERVER_PORT, 
		database = "eunitdb", 
		user = ?TEST_USER, 
		password = ?TEST_PASSWORD,
		flags = #client_options{trans_isolation_level = serializable}
	}),
  Conn = datasource:get_connection(serializable_ds),
	connection:execute_query(Conn, "UPDATE all_in_one SET int_col = 0 WHERE persist_id < 11"), 
	?debug_Fmt("~n >>> initial setup ~p ~p", [X, C]),
	List_ids = [{1,1},{2,2},{3,1},{4,1},{5,1},{6,3},{7,1},{8,4},{9,5},{10,1},
		{11,7},{12,1},{13,8},{14,9},{15,10},{16,4},{17,4},{18,9},{19,9},{20,10}],

	Self = self(),
	lists:map(fun({Worker_id, Id}) -> spawn(fun() -> process_entry(Id, Worker_id), Self ! done end) end, List_ids),
	
	wait_all(20),
	
	{_,[[Val]|_]} = connection:execute_query(Conn, "SELECT int_col FROM all_in_one WHERE persist_id=1"),
  datasource:return_connection(serializable_ds, Conn),
  datasource:close(serializable_ds),
	?assertEqual(7, Val),
	?debug_Fmt("Val=~p",[Val]),
  ?PASSED
end}.

wait_all(0) -> ok;
wait_all(N) ->
	receive
		done -> wait_all(N - 1)
	end.

%% @spec process_entry(Connection::#connection{}, Id::integer()) -> any()
%% @doc Marks message with Id as processed.
%%
process_entry(Id, Worker_id) ->
	?debug_Fmt(" >>> Process message [id=~p] worker #~p", [Id, Worker_id]),
	Conn = datasource:get_connection(serializable_ds),
%	connection:execute_query(Conn,"SET SESSION TRANSACTION ISOLATION LEVEL SERIALIZABLE"),
	H1 = connection:get_prepared_statement_handle(Conn,
			"SELECT int_col FROM all_in_one WHERE persist_id=?"),
	H2 = connection:get_prepared_statement_handle(Conn,
			"UPDATE all_in_one SET int_col=? WHERE persist_id=?"),
	run_transaction(Conn, H1, H2, Id, Worker_id),
	connection:close_statement(Conn, H1),
	connection:close_statement(Conn, H2),
	datasource:return_connection(serializable_ds, Conn).

run_transaction(Conn, H1, H2, Id, Worker_id) ->
	case connection:transaction(Conn, fun(C) ->
			{_,[[Val] | _]} = connection:execute_statement(C, H1, [?LONGLONG], [Id]),
			RR = connection:execute_statement(C, H2, [?LONG, ?LONGLONG], [Val + 1, Id]),
      timer:sleep(150),
      RR
		end) 
	of
		#mysql_error{type = transaction, errno = 1213, sqlstate = _, source = _, message = _M} ->
				?debug_Fmt("*** restarting transaction *** [id=~p] worker #~p; ~p", [Id, Worker_id, _M]),	
				timer:sleep(300),
				run_transaction(Conn, H1, H2, Id, Worker_id);
		{#metadata{}, [#ok_packet{}]} -> ?debug_Fmt(" <<< End process message [id=~p] worker #~p;", [Id, Worker_id]) 
	end.