Use EntryCounters during GC.

Bug: 14425059
Change-Id: I61eb798686dc753fb6c0fe99a0719c1732198f30

1 files changed, 14 insertions, 0 deletions

diff --git a/native/jni/src/suggest/policyimpl/dictionary/utils/entry_counters.h b/native/jni/src/suggest/policyimpl/dictionary/utils/entry_counters.h
index b8fa5aa9e..73dc42a18 100644
--- a/native/jni/src/suggest/policyimpl/dictionary/utils/entry_counters.h
+++ b/native/jni/src/suggest/policyimpl/dictionary/utils/entry_counters.h
@@ -46,6 +46,13 @@ class EntryCounts final {
         return mEntryCounts[2];
     }
 
+    int getNgramCount(const size_t n) const {
+        if (n < 1 || n > mEntryCounts.size()) {
+            return 0;
+        }
+        return mEntryCounts[n - 1];
+    }
+
  private:
     DISALLOW_ASSIGNMENT_OPERATOR(EntryCounts);
 
@@ -110,6 +117,13 @@ class MutableEntryCounters final {
         --mEntryCounters[n - 1];
     }
 
+    void setNgramCount(const size_t n, const int count) {
+        if (n < 1 || n > mEntryCounters.size()) {
+            return;
+        }
+        mEntryCounters[n - 1] = count;
+    }
+
  private:
     DISALLOW_COPY_AND_ASSIGN(MutableEntryCounters);