Skip to content

Commit 08f6577

Browse files
committed
moved the ListNodeConfigurationsByConfigurations to a private method
1 parent dce5f6f commit 08f6577

File tree

1 file changed

+32
-1
lines changed

1 file changed

+32
-1
lines changed

src/ResourceManager/Automation/Commands.Automation/Common/AutomationClientDSC.cs

Lines changed: 32 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,7 @@ public Model.DscNode GetDscNodeById(
473473
IEnumerable<Model.DscNode> listOfNodes = Enumerable.Empty<Model.DscNode>();
474474

475475
// first get the list of node configurations for the given configuration
476-
IEnumerable < Model.NodeConfiguration > listOfNodeConfigurations = this.ListNodeConfigurationsByConfigurationName(
476+
IEnumerable<Model.NodeConfiguration> listOfNodeConfigurations = this.EnumerateNodeConfigurationsByConfigurationName(
477477
resourceGroupName,
478478
automationAccountName,
479479
configurationName);
@@ -895,6 +895,37 @@ public Model.NodeConfiguration GetNodeConfiguration(string resourceGroupName, st
895895
}
896896
}
897897

898+
/// <summary>
899+
/// Enumerate the list of NodeConfigurations for given configuration - without any rollup status
900+
/// </summary>
901+
/// <param name="resourceGroupName">Resource group name</param>
902+
/// <param name="automationAccountName">Automation account</param>
903+
/// <param name="configurationName">Name of configuration</param>
904+
/// <returns>List of NodeConfigurations</returns>
905+
private IEnumerable<Model.NodeConfiguration> EnumerateNodeConfigurationsByConfigurationName(string resourceGroupName, string automationAccountName, string configurationName)
906+
{
907+
using (var request = new RequestSettings(this.automationManagementClient))
908+
{
909+
IEnumerable<AutomationManagement.Models.DscNodeConfiguration> nodeConfigModels;
910+
911+
nodeConfigModels = AutomationManagementClient.ContinuationTokenHandler(
912+
skipToken =>
913+
{
914+
var response = this.automationManagementClient.NodeConfigurations.List(
915+
resourceGroupName,
916+
automationAccountName,
917+
new AutomationManagement.Models.DscNodeConfigurationListParameters
918+
{
919+
ConfigurationName = configurationName
920+
});
921+
return new ResponseWithSkipToken<AutomationManagement.Models.DscNodeConfiguration>(response, response.DscNodeConfigurations);
922+
});
923+
924+
925+
return nodeConfigModels.Select(nodeConfigModel => new Commands.Automation.Model.NodeConfiguration(automationAccountName, nodeConfigModel));
926+
}
927+
}
928+
898929
#endregion
899930

900931
#region privatemethods

0 commit comments

Comments
 (0)