org.simantics.r.scl/src/org/simantics/r/scl/RSessionManager.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2014, 2016 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.r.scl;\r
  13 \r
  14 import java.util.UUID;\r
  15 import java.util.concurrent.ConcurrentHashMap;\r
  16 \r
  17 import org.rosuda.REngine.Rserve.RConnection;\r
  18 import org.rosuda.REngine.Rserve.RserveException;\r
  19 import org.simantics.db.layer0.variable.NodeSupport;\r
  20 import org.simantics.scl.runtime.function.Function;\r
  21 \r
  22 public class RSessionManager {\r
  23     static ConcurrentHashMap<String, RSession> CONNECTIONS =\r
  24             new ConcurrentHashMap<String, RSession>(); \r
  25 \r
  26     static ConcurrentHashMap<String, NodeSupport<String>> SUPPORTS =\r
  27             new ConcurrentHashMap<String, NodeSupport<String>>(); \r
  28     \r
  29     public static RSession getSession(String id) {\r
  30         // CONNECTIONS is ConcurrentHashMap so no synchronization is needed here\r
  31         return CONNECTIONS.get(id);\r
  32     }\r
  33     \r
  34     public static RSession createSession(RSessionConfiguration configuration) throws RserveException {\r
  35         synchronized(CONNECTIONS) {\r
  36             String id = UUID.randomUUID().toString();\r
  37             return createSession(configuration, id);\r
  38         }\r
  39     }\r
  40     \r
  41     public static Object withConfiguration(RSessionConfiguration configuration, @SuppressWarnings("rawtypes") Function f) throws RserveException {\r
  42         RSession session = createSession(configuration);\r
  43         try {\r
  44             return session.syncExec(f);\r
  45         } catch (InterruptedException e) {\r
  46             throw new RuntimeException(e);\r
  47         } finally {\r
  48             session.close();\r
  49         }\r
  50     }\r
  51     \r
  52     public static RSession getOrCreateSession(RSessionConfiguration configuration, String id) throws RserveException {\r
  53         synchronized(CONNECTIONS) {\r
  54             RSession session = getSession(id);\r
  55             if(session == null)\r
  56                 return createSession(configuration, id);\r
  57             else\r
  58                 return session;\r
  59         }\r
  60     }\r
  61     \r
  62     private static RSession createSession(RSessionConfiguration configuration, String id) throws RserveException {\r
  63         synchronized(CONNECTIONS) {\r
  64             RConnection connection = new RConnection(configuration.host, configuration.port);\r
  65             if(configuration.username != null && !configuration.username.isEmpty())\r
  66                 connection.login(configuration.username, configuration.password);\r
  67             RSession managedConnection = new RSession(connection, id);\r
  68             CONNECTIONS.put(id, managedConnection);\r
  69             return managedConnection;\r
  70         }\r
  71     }\r
  72     \r
  73     public static NodeSupport<String> getOrCreateNodeSupport(String id) {\r
  74         synchronized(SUPPORTS) {\r
  75             NodeSupport<String> result = SUPPORTS.get(id);\r
  76             if(result == null) {\r
  77                 RSession session = getSession(id);\r
  78                 result = new NodeSupport<String>(session.getNodeManager());\r
  79                 SUPPORTS.put(id, result);\r
  80             }\r
  81             return result;\r
  82         }\r
  83     }\r
  84 }\r