Adapt data collection cmdlets to new registry

Author: markcowl

src/ServiceManagement/Profile/Commands.Profile/DataCollection/EnableAzureDataCollection.cs

@@ -12,6 +12,8 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using Microsoft.Azure.Commands.Common.Authentication;
+using Microsoft.WindowsAzure.Commands.Common;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
 using System.Management.Automation;
 using System.Security.Permissions;
@@ -29,9 +31,10 @@ public override void ExecuteCmdlet()
 
         protected void SetDataCollectionProfile(bool enable)
         {
-            var profile = GetDataCollectionProfile();
+            var profile = _dataCollectionProfile;
             profile.EnableAzureDataCollection = enable;
-            SaveDataCollectionProfile();
+            DataCollectionController.WritePSDataCollectionProfile(AzureSession.Instance, profile);
+            AzureSession.Instance.RegisterComponent(DataCollectionController.RegistryKey, () => DataCollectionController.Create(profile), true);
         }
     }
 }