From 63d57f401e9e9a8ed376934581f42793d00e32cc Mon Sep 17 00:00:00 2001
From: Anders Svensson <anders@erlang.org>
Date: Fri, 8 Feb 2013 01:38:25 +0100
Subject: Tweak service interface towards diameter_peer

---
 lib/diameter/src/base/diameter_peer.erl    |  9 +--------
 lib/diameter/src/base/diameter_service.erl | 11 ++++++++++-
 2 files changed, 11 insertions(+), 9 deletions(-)

(limited to 'lib/diameter/src/base')

diff --git a/lib/diameter/src/base/diameter_peer.erl b/lib/diameter/src/base/diameter_peer.erl
index 25c9eab4cb..20b22c7949 100644
--- a/lib/diameter/src/base/diameter_peer.erl
+++ b/lib/diameter/src/base/diameter_peer.erl
@@ -272,7 +272,7 @@ handle_cast(Msg, State) ->
 
 %% Remote service is distributing a message.
 handle_info({notify, SvcName, T}, S) ->
-    bang(diameter_service:whois(SvcName), T),
+    diameter_service:notify(SvcName, T),
     {noreply, S};
 
 handle_info(Info, State) ->
@@ -304,13 +304,6 @@ code_change(_OldVsn, State, _Extra) ->
 ifc_send(Pid, T) ->
     Pid ! {diameter, T}.
 
-%% bang/2
-
-bang(undefined = No, _) ->
-    No;
-bang(Pid, T) ->
-    Pid ! T.
-
 %% call/1
 
 call(Request) ->
diff --git a/lib/diameter/src/base/diameter_service.erl b/lib/diameter/src/base/diameter_service.erl
index a4b788af4e..f1342df16c 100644
--- a/lib/diameter/src/base/diameter_service.erl
+++ b/lib/diameter/src/base/diameter_service.erl
@@ -40,7 +40,7 @@
          stop_transport/2]).
 
 %% towards diameter_peer
--export([whois/1]).
+-export([notify/2]).
 
 %% towards diameter_traffic
 -export([find_incoming_app/4,
@@ -51,6 +51,7 @@
          subscriptions/1,
          subscriptions/0,
          call_module/3,
+         whois/1,
          state/1,
          uptime/1]).
 
@@ -324,6 +325,14 @@ find_incoming_app(PeerT, TPid, Id, Apps) ->
             false
     end.
 
+%% ---------------------------------------------------------------------------
+%% # notify/2
+%% ---------------------------------------------------------------------------
+
+notify(SvcName, Msg) ->
+    Pid = whois(SvcName),
+    is_pid(Pid) andalso (Pid ! Msg).
+
 %% ===========================================================================
 %% ===========================================================================
 
-- 
cgit v1.2.3