init

MarketDataLib/CNNProcessing/CNNClient.cs

@@ -0,0 +1,98 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Net.Http;
+using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MarketData.CNNProcessing
+{
+  public class CNNClient
+	{
+	  public enum Model{resnet50,inception,vgg16,lenet5,ping};
+		private static readonly string Alive="Alive";
+    private readonly HttpClient client = new HttpClient();
+		private string baseUrl;
+    private static int REQUEST_TIMEOUT=15000;
+
+		public CNNClient(String baseUrl)
+		{
+      client.Timeout=new TimeSpan(0,0,0,0,REQUEST_TIMEOUT);
+      if(!baseUrl.StartsWith("http://") && !baseUrl.StartsWith("HTTP://"))baseUrl="http://"+baseUrl;
+		  this.baseUrl=baseUrl;
+		}
+
+		private String GetModelUrl(CNNClient.Model model)
+		{
+		  return baseUrl+"/predict_"+model.ToString();
+		}
+
+    public String Predict(CNNClient.Model model,Stream stream)
+    {
+		  try
+			{
+        string responsePayload = Upload(GetModelUrl(model),stream).GetAwaiter().GetResult();
+				return responsePayload;
+			}
+			catch(Exception exception)
+			{
+			  return exception.ToString();
+			}
+    }
+
+    public bool Ping()
+    {
+		  try
+			{
+        string responsePayload = Upload(baseUrl+"/ping").GetAwaiter().GetResult();
+        if(null==responsePayload)return false;
+				return responsePayload.Equals(Alive);
+			}
+			catch(Exception exception)
+			{
+        MDTrace.WriteLine(LogLevel.DEBUG,String.Format("Exception encountered: {0}",exception.ToString()));
+			  return false;
+			}
+    }
+    private async Task<String> Upload(String url,Stream stream)
+    {
+      try
+      {
+        using (HttpRequestMessage request = new HttpRequestMessage(HttpMethod.Post, url))
+        {
+				  int streamEnd = Convert.ToInt32(stream.Length);
+				  byte[] byteArray = new byte[streamEnd];
+				  stream.Read(byteArray, 0, streamEnd);
+          request.Content=new ByteArrayContent(byteArray);
+				  using (HttpResponseMessage response = await client.SendAsync(request))
+          {
+            return await response.Content.ReadAsStringAsync();
+          }
+        }
+      }
+      catch(Exception exception)
+      {
+        MDTrace.WriteLine(LogLevel.DEBUG,String.Format("Exception encountered: {0}",exception.ToString()));
+        return null;
+      }
+    }
+    private async Task<String> Upload(String url)
+    {
+      try
+      {
+        using (HttpRequestMessage request = new HttpRequestMessage(HttpMethod.Get, url))
+        {
+				  HttpResponseMessage response = await client.SendAsync(request);
+          return await response.Content.ReadAsStringAsync();
+        }
+      }
+      catch(Exception exception)
+      {
+        MDTrace.WriteLine(LogLevel.DEBUG,String.Format("Exception encountered: {0}",exception.ToString()));
+        return null;
+      }
+    }
+	}
+}