From dce8d88ae2f8cc5140ac2006b8eedab7eaf13989 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kjetil=20=C3=98rbekk?= <kjetil.orbekk@gmail.com>
Date: Sun, 15 Jan 2012 18:28:59 +0100
Subject: Change ConnectionManager.

Support Client and Master connections.
---
 .../java/com/orbekk/same/ConnectionManagerImpl.java     | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

(limited to 'same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java')

diff --git a/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java b/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
index 841d5fa..0a1e54f 100644
--- a/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
+++ b/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
@@ -14,18 +14,27 @@ public class ConnectionManagerImpl implements ConnectionManager {
     public ConnectionManagerImpl() {
     }
 
-    @Override
-    public SameService getConnection(String url) {
-        SameService service = null;
+    private <T>T getClassProxy(String url, Class<T> clazz) {
+        T service = null;
         try {
             JsonRpcHttpClient client = new JsonRpcHttpClient(new URL(url));
             service = ProxyUtil.createProxy(
                     this.getClass().getClassLoader(),
-                    SameService.class,
+                    clazz,
                     client);
         } catch (MalformedURLException e) {
             logger.warn("Unable to create client for {}, {}", url, e);
         }
         return service;
     }
+
+    @Override
+    public ClientService getClient(String url) {
+        return getClassProxy(url, ClientService.class);
+    }
+
+    @Override
+    public MasterService getMaster(String url) {
+        return getClassProxy(url, MasterService.class);
+    }
 }
-- 
cgit v1.2.3