From 553b9af36e7a216e197ffe8bc67cfec6b9e07f0d Mon Sep 17 00:00:00 2001 From: Aditya Bist Date: Tue, 22 Mar 2022 21:43:43 -0700 Subject: [PATCH] graph type logic for node sub type (#1440) --- .../ObjectExplorer/SmoModel/SmoTableCustomNode.cs | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/src/Microsoft.SqlTools.ServiceLayer/ObjectExplorer/SmoModel/SmoTableCustomNode.cs b/src/Microsoft.SqlTools.ServiceLayer/ObjectExplorer/SmoModel/SmoTableCustomNode.cs index 448f9035..cb286a6a 100644 --- a/src/Microsoft.SqlTools.ServiceLayer/ObjectExplorer/SmoModel/SmoTableCustomNode.cs +++ b/src/Microsoft.SqlTools.ServiceLayer/ObjectExplorer/SmoModel/SmoTableCustomNode.cs @@ -43,9 +43,20 @@ namespace Microsoft.SqlTools.ServiceLayer.ObjectExplorer.SmoModel try { Table table = smoObject as Table; - if (table != null && IsPropertySupported("TemporalType", smoContext, table, CachedSmoProperties) && table.TemporalType != TableTemporalType.None) + if (table != null) { - return "Temporal"; + if (IsPropertySupported("TemporalType", smoContext, table, CachedSmoProperties) && table.TemporalType != TableTemporalType.None) + { + return "Temporal"; + } + else if (IsPropertySupported("IsEdge", smoContext, table, CachedSmoProperties) && table.IsEdge) + { + return "GraphEdge"; + } + else if (IsPropertySupported("IsNode", smoContext, table, CachedSmoProperties) && table.IsNode) + { + return "GraphNode"; + } } // TODO carbon issue 3125 enable "External" subtype once icon is ready. Otherwise will get missing icon here. // else if (table != null && IsPropertySupported("IsExternal", smoContext, table, CachedSmoProperties) && table.IsExternal)