fix issues

Author: khanak0509

packages/devtools_app/lib/src/screens/memory/panes/profile/profile_pane_controller.dart

@@ -29,6 +29,7 @@ class ProfilePaneController extends DisposableController
       _pinnedClassFullNames.addAll(pinnedClassFullNames);
       _pinnedClassFullNamesListenable.value = pinnedClassFullNames;
     }
+    _pinnedClassFullNamesListenable.value = Set.of(_pinnedClassFullNames);
     // [profile] should only be non-null when loading offline data.
     if (profile != null) {
       _currentAllocationProfile.value = AdaptedProfile.withPinnedClasses(

packages/devtools_app/lib/src/screens/memory/panes/profile/profile_view.dart

@@ -580,6 +580,18 @@ class _AllocationProfileTableState extends State<_AllocationProfileTable> {
   @override
   void initState() {
     super.initState();
+    _initColumns();
+  }
+
+  @override
+  void didUpdateWidget(covariant _AllocationProfileTable oldWidget) {
+    super.didUpdateWidget(oldWidget);
+    if (oldWidget.controller != widget.controller) {
+      _initColumns();
+    }
+  }
+
+  void _initColumns() {
     _pinColumn = _PinColumn(controller: widget.controller);
     _columns = <ColumnData<ProfileRecord>>[
       _pinColumn,
@@ -607,33 +619,26 @@ class _AllocationProfileTableState extends State<_AllocationProfileTable> {
         if (profile == null) {
           return const CenteredCircularProgressIndicator();
         }
-        return ValueListenableBuilder<Set<String>>(
-          valueListenable: widget.controller.pinnedClassFullNames,
-          builder: (context, pinnedClassFullNames, _) {
-            return ValueListenableBuilder<bool>(
-              valueListenable: preferences.advancedDeveloperModeEnabled,
-              builder: (context, advancedDeveloperModeEnabled, _) {
-                return FlatTable<ProfileRecord>(
-                  keyFactory: (element) => Key(element.heapClass.fullName),
-                  data: profile.records,
-                  dataKey:
-                      'allocation-profile-${pinnedClassFullNames.length}-'
-                      '${pinnedClassFullNames.toList()..sort()}',
-                  columnGroups: advancedDeveloperModeEnabled
-                      ? _AllocationProfileTableState._vmModeColumnGroups
-                      : null,
-                  columns: [
-                    ..._columns,
-                    if (advancedDeveloperModeEnabled)
-                      ..._advancedDeveloperModeColumns,
-                  ],
-                  defaultSortColumn: _AllocationProfileTableState._fieldSizeColumn,
-                  defaultSortDirection: SortDirection.descending,
-                  pinBehavior: FlatTablePinBehavior.pinOriginalToTop,
-                  includeColumnGroupHeaders: false,
-                  selectionNotifier: widget.controller.selection,
-                );
-              },
+        return ValueListenableBuilder<bool>(
+          valueListenable: preferences.advancedDeveloperModeEnabled,
+          builder: (context, advancedDeveloperModeEnabled, _) {
+            return FlatTable<ProfileRecord>(
+              keyFactory: (element) => Key(element.heapClass.fullName),
+              data: profile.records,
+              dataKey: 'allocation-profile',
+              columnGroups: advancedDeveloperModeEnabled
+                  ? _AllocationProfileTableState._vmModeColumnGroups
+                  : null,
+              columns: [
+                ..._columns,
+                if (advancedDeveloperModeEnabled)
+                  ..._advancedDeveloperModeColumns,
+              ],
+              defaultSortColumn: _AllocationProfileTableState._fieldSizeColumn,
+              defaultSortDirection: SortDirection.descending,
+              pinBehavior: FlatTablePinBehavior.pinOriginalToTop,
+              includeColumnGroupHeaders: false,
+              selectionNotifier: widget.controller.selection,
             );
           },
         );