Add transaction support

README.md

@@ -111,6 +111,27 @@ Types:
   Params = list()
 ```
 
+##### Transaction
+
+```erlang
+pgpool:transaction(DatabaseName, Fun) -> (see epgsql for reply/error types)
+
+Types:
+  DatabaseName = atom()
+  Fun = fun()
+```
+
+For example:
+
+```erlang
+pgpool:transaction(db1_name,
+                   fun(C) ->
+                           pgpool:squery(C, "DELETE FROM users WHERE foo = true"),
+                           pgpool:squery(C, "UPDATE users SET bar = 100"),
+                           pgpool:squery(C, "SELECT * FROM users")
+                   end).
+```
+
 ## Contributing
 So you want to contribute? That's great! Please follow the guidelines below. It will make it easier to get merged in.
 

src/pgpool_worker.erl

@@ -4,7 +4,7 @@
 
 %% API
 -export([start_link/1]).
--export([squery/2, equery/3]).
+-export([squery/2, equery/3, transaction/2]).
 
 %% gen_server callbacks
 -export([init/1, handle_call/3, handle_cast/2, handle_info/2, terminate/2, code_change/3]).
@@ -43,6 +43,12 @@ equery(DatabaseName, Statement, Params) ->
         gen_server:call(Worker, {equery, Statement, Params})
     end).
 
+-spec transaction(DatabaseName :: atom(), Fun :: fun()) -> any() | {error, no_connection}.
+transaction(DatabaseName, Fun) ->
+    poolboy:transaction(DatabaseName, fun(Worker) ->
+        gen_server:call(Worker, {transaction, Fun})
+    end).
+
 %% ===================================================================
 %% Callbacks
 %% ===================================================================
@@ -102,6 +108,9 @@ handle_call({squery, Sql}, _From, #state{conn = Conn} = State) ->
 handle_call({equery, Statement, Params}, _From, #state{conn = Conn} = State) ->
     {reply, epgsql:equery(Conn, Statement, Params), State};
 
+handle_call({transaction, Fun}, _From, #state{conn = Conn} = State) ->
+    {reply, epgsql:with_transaction(Conn, Fun), State};
+
 handle_call(Request, From, State) ->
     error_logger:warning_msg("Received from ~p an unknown call message: ~p", [Request, From]),
     {reply, undefined, State}.