Add timeout to network operations.

1 files changed, 12 insertions, 2 deletions

diff --git a/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java b/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
index 0a1e54f..f74bca7 100644
--- a/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
+++ b/same/src/main/java/com/orbekk/same/ConnectionManagerImpl.java
@@ -8,16 +8,26 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 public class ConnectionManagerImpl implements ConnectionManager {
-
+    private int connectionTimeout;
+    private int readTimeout;
+    
     private Logger logger = LoggerFactory.getLogger(getClass());
 
-    public ConnectionManagerImpl() {
+    /**
+     * @param connectionTimout Timeout for establishing connection in milliseconds.
+     * @param readTimeout Timeout for waiting for answer in milliseconds.
+     */
+    public ConnectionManagerImpl(int connectionTimout, int readTimeout) {
+        this.connectionTimeout = connectionTimout;
+        this.readTimeout = readTimeout;
     }
 
     private <T>T getClassProxy(String url, Class<T> clazz) {
         T service = null;
         try {
             JsonRpcHttpClient client = new JsonRpcHttpClient(new URL(url));
+            client.setConnectionTimeoutMillis(connectionTimeout);
+            client.setReadTimeoutMillis(readTimeout);
             service = ProxyUtil.createProxy(
                     this.getClass().getClassLoader(),
                     clazz,