Added service for model management for machine learning extension (#1138)

* Added service for model management for ml extension

src/Microsoft.SqlTools.ServiceLayer/ModelManagement/ModelManagementService.cs

@@ -0,0 +1,241 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using Microsoft.SqlTools.Hosting.Protocol;
+using Microsoft.SqlTools.ServiceLayer.Connection;
+using Microsoft.SqlTools.ServiceLayer.Hosting;
+using Microsoft.SqlTools.ServiceLayer.ModelManagement.Contracts;
+using Microsoft.SqlTools.Utility;
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Diagnostics;
+using System.Threading.Tasks;
+
+namespace Microsoft.SqlTools.ServiceLayer.ModelManagement
+{
+    public class ModelManagementService
+    {
+        private ModelOperations serviceOperations = new ModelOperations();
+        private ConnectionService connectionService = null;
+        private static readonly Lazy<ModelManagementService> instance = new Lazy<ModelManagementService>(() => new ModelManagementService());
+
+
+        /// <summary>
+        /// Gets the singleton instance object
+        /// </summary>
+        public static ModelManagementService Instance
+        {
+            get { return instance.Value; }
+        }
+
+        /// <summary>
+        /// Internal for testing purposes only
+        /// </summary>
+        internal ConnectionService ConnectionServiceInstance
+        {
+            get
+            {
+                if (connectionService == null)
+                {
+                    connectionService = ConnectionService.Instance;
+                }
+                return connectionService;
+            }
+
+            set
+            {
+                connectionService = value;
+            }
+        }
+
+        public ModelOperations ModelOperations
+        {
+            get
+            {
+                return serviceOperations;
+            }
+            set
+            {
+                serviceOperations = value;
+            }
+        }
+
+        public void InitializeService(ServiceHost serviceHost)
+        {
+            serviceHost.SetRequestHandler(ImportModelRequest.Type, this.HandleModelImportRequest);
+            serviceHost.SetRequestHandler(ConfigureModelTableRequest.Type, this.HandleConfigureModelTableRequest);
+            serviceHost.SetRequestHandler(DeleteModelRequest.Type, this.HandleDeleteModelRequest);
+            serviceHost.SetRequestHandler(DownloadModelRequest.Type, this.HandleDownloadModelRequest);
+            serviceHost.SetRequestHandler(GetModelsRequest.Type, this.HandleGetModelsRequest);
+            serviceHost.SetRequestHandler(UpdateModelRequest.Type, this.HandleUpdateModelRequest);
+            serviceHost.SetRequestHandler(VerifyModelTableRequest.Type, this.HandleVerifyModelTableRequest);
+        }
+
+        /// <summary>
+        /// Handles import model request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleModelImportRequest(ImportModelRequestParams parameters, RequestContext<ImportModelResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleModelImportRequest");
+                ImportModelResponseParams response = new ImportModelResponseParams
+                {
+                };
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                ModelOperations.ImportModel(dbConnection, parameters);
+                return response;
+            });
+        }
+
+        /// <summary>
+        /// Handles get models request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleGetModelsRequest(GetModelsRequestParams parameters, RequestContext<GetModelsResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleGetModelsRequest");
+            GetModelsResponseParams response = new GetModelsResponseParams
+            {
+            };
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                List<ModelMetadata> models = ModelOperations.GetModels(dbConnection, parameters);
+                response.Models = models;
+                return response;
+            });
+        }
+
+        /// <summary>
+        /// Handles update model request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleUpdateModelRequest(UpdateModelRequestParams parameters, RequestContext<UpdateModelResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleUpdateModelRequest");
+            UpdateModelResponseParams response = new UpdateModelResponseParams
+            {
+            };
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                ModelOperations.UpdateModel(dbConnection, parameters);
+                return response;
+            });
+        }
+
+        /// <summary>
+        /// Handles delete model request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleDeleteModelRequest(DeleteModelRequestParams parameters, RequestContext<DeleteModelResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleDeleteModelRequest");
+            DeleteModelResponseParams response = new DeleteModelResponseParams
+            {
+            };
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                ModelOperations.DeleteModel(dbConnection, parameters);
+                return response;
+            });
+        }
+
+        /// <summary>
+        /// Handles download model request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleDownloadModelRequest(DownloadModelRequestParams parameters, RequestContext<DownloadModelResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleDownloadModelRequest");
+            DownloadModelResponseParams response = new DownloadModelResponseParams
+            {
+            };
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                response.FilePath = ModelOperations.DownloadModel(dbConnection, parameters);
+                return response;
+            });
+        }
+
+        /// <summary>
+        /// Handles verify model table request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleVerifyModelTableRequest(VerifyModelTableRequestParams parameters, RequestContext<VerifyModelTableResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleVerifyModelTableRequest");
+            VerifyModelTableResponseParams response = new VerifyModelTableResponseParams
+            {
+            };
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                response.Verified = ModelOperations.VerifyImportTable(dbConnection, parameters);
+                return response;
+            });
+        }
+
+        /// <summary>
+        /// Handles configure model table request
+        /// </summary>
+        /// <param name="parameters">Request parameters</param>
+        /// <param name="requestContext">Request Context</param>
+        public async Task HandleConfigureModelTableRequest(ConfigureModelTableRequestParams parameters, RequestContext<ConfigureModelTableResponseParams> requestContext)
+        {
+            Logger.Write(TraceEventType.Verbose, "HandleConfigureModelTableRequest");
+            ConfigureModelTableResponseParams response = new ConfigureModelTableResponseParams();
+
+            await HandleRequest(parameters, response, requestContext, (dbConnection, parameters, response) =>
+            {
+                ModelOperations.ConfigureImportTable(dbConnection, parameters);
+                return response;
+            });
+        }
+
+        private async Task HandleRequest<T, TResponse>(
+            T parameters, 
+            TResponse response, 
+            RequestContext<TResponse> requestContext,
+            Func<IDbConnection, T, TResponse, TResponse> operation) where T : ModelRequestBase where TResponse : ModelResponseBase
+        {
+            try
+            {
+                ConnectionInfo connInfo;
+                ConnectionServiceInstance.TryFindConnection(
+                    parameters.OwnerUri,
+                    out connInfo);
+                if (connInfo == null)
+                {
+                    await requestContext.SendError(new Exception(SR.ConnectionServiceDbErrorDefaultNotConnected(parameters.OwnerUri)));
+                }
+                else
+                {
+                    using (IDbConnection dbConnection = ConnectionService.OpenSqlConnection(connInfo))
+                    {
+                        response = operation(dbConnection, parameters, response);
+                    }
+                    await requestContext.SendResult(response);
+                }
+            }
+            catch (Exception e)
+            {
+                // Exception related to run task will be captured here
+                await requestContext.SendError(e);
+            }
+        }
+    }
+}