Keep alive Riak connections by default

processone · Mar 22, 2016 · 968eb25 · 968eb25
1 parent 35f66b2
commit 968eb25
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/ejabberd_riak.erl b/src/ejabberd_riak.erl
@@ -73,7 +73,13 @@ start_link(Num, Server, Port, _StartInterval, Options) ->
 
 %% @private
 is_connected() ->
-    catch riakc_pb_socket:is_connected(get_random_pid()).
+    lists:all(
+      fun({_Id, Pid, _Type, _Modules}) when is_pid(Pid) ->
+	      case catch riakc_pb_socket:is_connected(get_random_pid()) of
+		  true -> true;
+		  _ -> false
+	      end
+      end, supervisor:which_children(ejabberd_riak_sup)).
 
 %% @private
 get_proc(I) ->

diff --git a/src/ejabberd_riak_sup.erl b/src/ejabberd_riak_sup.erl
@@ -109,6 +109,7 @@ init([]) ->
     Options = lists:filter(
 		fun(X) -> X /= nil end,
 		[auto_reconnect,
+		 {keepalive, true},
 		 if CACertFile /= nil -> {cacertfile ,CACertFile};
 		    true -> nil
 		 end,