From f9203738fb20ddcce03598a16b92b63358ecb861 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kjetil=20=C3=98rbekk?= <kjetil.orbekk@gmail.com>
Date: Tue, 17 Apr 2012 13:28:55 +0200
Subject: Add Services.Client to ConnectionManager.

---
 same/src/main/java/com/orbekk/same/ConnectionManager.java     |  1 +
 same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java | 10 ++++++++++
 same/src/main/java/com/orbekk/same/TestConnectionManager.java |  7 +++++++
 3 files changed, 18 insertions(+)

diff --git a/same/src/main/java/com/orbekk/same/ConnectionManager.java b/same/src/main/java/com/orbekk/same/ConnectionManager.java
index 4a24da5..f1ca65f 100644
--- a/same/src/main/java/com/orbekk/same/ConnectionManager.java
+++ b/same/src/main/java/com/orbekk/same/ConnectionManager.java
@@ -13,5 +13,6 @@ public interface ConnectionManager {
     PaxosService getPaxos(String url);
 
     Services.Master getMaster0(String location);
+    Services.Client getClient0(String location);
     Services.Directory getDirectory(String location);
 }
diff --git a/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java b/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
index 8c3b9df..34536a0 100644
--- a/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
+++ b/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
@@ -126,6 +126,16 @@ public class ConnectionManagerImpl implements ConnectionManager {
         }
     }
     
+    @Override
+    public Services.Client getClient0(String location) {
+        RpcChannel channel = getChannel(location);
+        if (channel != null) {
+            return Services.Client.newStub(channel);
+        } else {
+            return null;
+        }
+    }
+    
     @Override
     public Services.Directory getDirectory(String location) {
         RpcChannel channel = getChannel(location);
diff --git a/same/src/main/java/com/orbekk/same/TestConnectionManager.java b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
index 8319467..2305311 100644
--- a/same/src/main/java/com/orbekk/same/TestConnectionManager.java
+++ b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
@@ -21,6 +21,8 @@ public class TestConnectionManager implements ConnectionManager {
             new HashMap<String, Services.Directory>();
     public Map<String, Services.Master> masterMap0 =
             new HashMap<String, Services.Master>();
+    public Map<String, Services.Client> clientMap0 =
+            new HashMap<String, Services.Client>();
 
     public TestConnectionManager() {
     }
@@ -49,4 +51,9 @@ public class TestConnectionManager implements ConnectionManager {
     public Master getMaster0(String location) {
         return masterMap0.get(location);
     }
+
+    @Override
+    public Services.Client getClient0(String location) {
+        return clientMap0.get(location);
+    }
 }
-- 
cgit v1.2.3