Optimize CountriesCell to add rows only when the cell loads

Author: staskus

WordPress/Classes/ViewRelated/Stats/Period Stats/Countries/CountriesCell.swift

@@ -21,6 +21,12 @@ class CountriesCell: StatsBaseCell, NibLoadable {
     private typealias Style = WPStyleGuide.Stats
     private var forDetails = false
 
+    override func awakeFromNib() {
+        super.awakeFromNib()
+
+        addDefaultTotalRows(toStackView: rowsStackView)
+    }
+
     // MARK: - Configure
 
     func configure(itemSubtitle: String,
@@ -38,21 +44,27 @@ class CountriesCell: StatsBaseCell, NibLoadable {
         bottomSeparatorLine.isHidden = forDetails
 
         if !forDetails {
-        addRows(dataRows,
-                toStackView: rowsStackView,
+            if rowsStackView.arrangedSubviews.isEmpty {
+                addDefaultTotalRows(toStackView: rowsStackView)
+            }
+            configureTotalRows(
+                dataRows,
+                inStackView: rowsStackView,
                 forType: .period,
-                limitRowsDisplayed: true,
-                viewMoreDelegate: self)
+                configuration: .init(
+                    limitRowsDisplayed: true,
+                    rowDelegate: nil,
+                    referrerDelegate: nil,
+                    viewMoreDelegate: self
+                )
+            )
+        } else {
+            removeRowsFromStackView(rowsStackView)
         }
 
         setSubtitleVisibility()
         applyStyles()
     }
-
-    override func prepareForReuse() {
-        super.prepareForReuse()
-        removeRowsFromStackView(rowsStackView)
-    }
 }
 
 private extension CountriesCell {