ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-02-05 01:25:45 -05:00
Code Issues Packages Projects Releases Wiki Activity

Adding cpu count and ram size to server info (#1220)

Browse Source 
* adding network file validator contract

* Adding additional properties to serverInfo

* Renaming to serversysinfo

* fixing var name

* Changed to smo to get the properties

* Fixing casting error

* renaming to sysinfo

* Removing unused sql scripts

* Throwing an error and renaming classes

* Changing Physicalmemory prop name from kb to mb

* pushing prop name change

* fixing comment
This commit is contained in:
Aasim Khan
2021-07-13 12:57:53 -07:00
committed by GitHub
parent 1f7da97e6c
commit 539e47b40f
3 changed files with 50 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/Microsoft.SqlTools.ManagedBatchParser/ReliableConnection/ReliableConnectionHelper.cs
View File

@@ -13,6 +13,7 @@ using System.Globalization;
using System.Security;
using Microsoft.SqlTools.Utility;
using Microsoft.SqlServer.Management.Common;
using Microsoft.SqlServer.Management.Smo;
namespace Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection
{
@@ -649,12 +650,11 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection
// of SQL Server do not have their metadata upgraded to include the xml_index_type column in the sys.xml_indexes view. Because
// of this, we must detect the presence of the column to determine if we can query for Selective Xml Indexes
public bool IsSelectiveXmlIndexMetadataPresent;
public string OsVersion;
public string MachineName;
public string ServerName;
public int CpuCount;
public int PhysicalMemoryInMB;
public Dictionary<string, object> Options { get; set; }
}
@@ -676,6 +676,12 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection
public string ServicePackLevel;
}
public class ServerSystemInfo
{
public int CpuCount;
public int PhysicalMemoryInMB;
}
public static bool TryGetServerVersion(string connectionString, out ServerInfo serverInfo, string azureAccountToken)
{
serverInfo = null;
@@ -753,6 +759,29 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection
return hostInfo;
}
/// <summary>
/// Gets the server host cpu count and memory from sys.dm_os_sys_info view
/// </summary>
/// <param name="connection">The connection</param>
public static ServerSystemInfo GetServerSystemInfo(IDbConnection connection)
{
var sysInfo = new ServerSystemInfo();
try
{
SqlConnection sqlConnection = GetAsSqlConnection(connection);
var server = new Server(new ServerConnection(sqlConnection));
server.SetDefaultInitFields(server.GetType(), new String[] { nameof(server.Processors), nameof(server.PhysicalMemory) });
sysInfo.CpuCount = server.Processors;
sysInfo.PhysicalMemoryInMB = server.PhysicalMemory;
}
catch (Exception ex)
{
Logger.Write(TraceEventType.Error, ex.ToString());
throw ex;
}
return sysInfo;
}
/// <summary>
/// Returns the version of the server. This routine will throw if an exception is encountered.
/// </summary>
@@ -827,6 +856,11 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection
// otherwise - Windows Server 2019 Standard 10.0
serverInfo.OsVersion = hostInfo.Distribution != null ? string.Format("{0} {1}", hostInfo.Distribution, hostInfo.Release) : string.Format("{0} {1}", hostInfo.Platform, hostInfo.Release);
var sysInfo = GetServerSystemInfo(connection);
serverInfo.CpuCount = sysInfo.CpuCount;
serverInfo.PhysicalMemoryInMB = sysInfo.PhysicalMemoryInMB;
serverInfo.Options = new Dictionary<string, object>();
// Get BDC endpoints