NotificationHandler is abstract method now

[StratumLibrary.git] / StratumLibrary / Stratum.cs

diff --git a/StratumLibrary/Stratum.cs b/StratumLibrary/Stratum.cs
index 722abda..9ce8935 100644 (file)
--- a/StratumLibrary/Stratum.cs
+++ b/StratumLibrary/Stratum.cs
@@ -2,6 +2,7 @@
 using System.Net;\r
 using System.Threading;\r
 using System.Net.Sockets;\r
+using Newtonsoft.Json;\r
 using Newtonsoft.Json.Linq;\r
 \r
 using System.Text;\r
@@ -9,9 +10,11 @@ using System.Collections.Generic;
 \r
 namespace Stratum\r
 {\r
-    public class Stratum\r
+    public abstract class Stratum\r
     {\r
         private Socket client;\r
+\r
+        object responsesLock = new object();\r
         private Dictionary<string, string> responses = new Dictionary<string, string>();\r
         ManualResetEvent gotResponse = new ManualResetEvent(false);\r
 \r
@@ -94,6 +97,22 @@ namespace Stratum
         /// </summary>\r
         /// <typeparam name="T">Return type</typeparam>\r
         /// <param name="method">Method name</param>\r
+        /// <returns>StratumResponse object</returns>\r
+        public StratumResponse<T> Invoke<T>(string method)\r
+        {\r
+            var req = new StratumRequest()\r
+            {\r
+                Method = method,\r
+                Params = new object[] { }\r
+            };\r
+            return Invoke<T>(req);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Invoke remote method\r
+        /// </summary>\r
+        /// <typeparam name="T">Return type</typeparam>\r
+        /// <param name="method">Method name</param>\r
         /// <param name="arg">Argument</param>\r
         /// <returns>StratumResponse object</returns>\r
         public StratumResponse<T> Invoke<T>(string method, object arg)\r
@@ -126,8 +145,7 @@ namespace Stratum
         private StratumResponse<T> Invoke<T>(StratumRequest stratumReq)\r
         {\r
             // Serialize stratumReq into JSON string\r
-            var reqJSON = Newtonsoft.Json.JsonConvert.SerializeObject(stratumReq) + '\n';\r
-            var reqId = (string) stratumReq.Id;\r
+            var reqJSON = JsonConvert.SerializeObject(stratumReq) + '\n';\r
 \r
             // Send JSON data to the remote device.\r
             Send(client, reqJSON);\r
@@ -135,10 +153,16 @@ namespace Stratum
             // Wait for response\r
             gotResponse.WaitOne();\r
 \r
-            // Deserialize the response\r
-            string strResponse = responses[reqId];\r
-            StratumResponse<T> responseObj = Newtonsoft.Json.JsonConvert.DeserializeObject<StratumResponse<T>>(strResponse);\r
-            responses.Remove(reqId);\r
+            var strResponse = string.Empty;\r
+            lock (responsesLock)\r
+            {\r
+                // Deserialize the response\r
+                strResponse = responses[stratumReq.Id];\r
+                responses.Remove(stratumReq.Id);\r
+            }\r
+\r
+            // Deserialize response into new instance of StratumResponse<T> \r
+            StratumResponse<T> responseObj = JsonConvert.DeserializeObject<StratumResponse<T>>(strResponse);\r
 \r
             // Reset the state\r
             gotResponse.Reset();\r
@@ -147,10 +171,10 @@ namespace Stratum
             {\r
                 try\r
                 {\r
-                    JObject jo = Newtonsoft.Json.JsonConvert.DeserializeObject(strResponse) as JObject;\r
-                    throw new Exception(jo["Error"].ToString());\r
+                    JObject jResponseObj = JsonConvert.DeserializeObject(strResponse) as JObject;\r
+                    throw new Exception(jResponseObj["Error"].ToString());\r
                 }\r
-                catch (Newtonsoft.Json.JsonSerializationException)\r
+                catch (JsonSerializationException)\r
                 {\r
                     throw new Exception("Inconsistent or empty response");\r
                 }\r
@@ -185,27 +209,32 @@ namespace Stratum
 \r
                     if (arStatus.buffer[bytesRead - 1] == '\n')\r
                     {\r
-                        string strMessage = arStatus.sb.ToString();\r
+                        var strMessage = arStatus.sb.ToString();\r
                         arStatus.sb.Clear();\r
 \r
                         try\r
                         {\r
-                            JObject jo = Newtonsoft.Json.JsonConvert.DeserializeObject(strMessage) as JObject;\r
-                            string requestId = (string)jo["id"];\r
+                            JObject jResponse = JsonConvert.DeserializeObject(strMessage) as JObject;\r
+                            var reqId = (string)jResponse["id"];\r
 \r
-                            if (!String.IsNullOrEmpty(requestId))\r
+                            if (!String.IsNullOrEmpty(reqId))\r
                             {\r
-                                responses.Add(requestId, strMessage);\r
+                                lock (responsesLock)\r
+                                {\r
+                                    responses.Add(reqId, strMessage);\r
+                                }\r
 \r
                                 gotResponse.Set();\r
                             }\r
                             else\r
                             {\r
-                                // TODO: notifications handling\r
-                                Console.WriteLine("Notification: {0}", strMessage);\r
+                                StratumNotification jNotification = JsonConvert.DeserializeObject<StratumNotification>(strMessage);\r
+\r
+                                var NotifyProcessThread = new Thread(() => NotificationHandler(jNotification.Method, jNotification.Params));\r
+                                NotifyProcessThread.Start();\r
                             }\r
                         }\r
-                        catch (Newtonsoft.Json.JsonSerializationException e)\r
+                        catch (JsonSerializationException e)\r
                         {\r
                             // TODO: handle parse error\r
                         }\r
@@ -217,6 +246,13 @@ namespace Stratum
 \r
             client.BeginReceive(state.buffer, 0, StratumReadState.BufferSize, SocketFlags.None, new AsyncCallback(ReceiveCallback), state);\r
         }\r
+\r
+        /// <summary>\r
+        /// Notifications stub which is run in a separate thread. You need to override this method in the derived class.\r
+        /// </summary>\r
+        /// <param name="NotificationMethod">Method name</param>\r
+        /// <param name="NotificationData">Array of values</param>\r
+               public abstract void NotificationHandler (string NotificationMethod, JArray NotificationData);\r
     }\r
 \r
 }
\ No newline at end of file