2020import androidx .annotation .VisibleForTesting ;
2121import androidx .datastore .preferences .core .*;
2222import androidx .datastore .preferences .core .Preferences ;
23- import com .google .firebase .datastore . DataStorage ;
24- import com .google .firebase .datastore . DataStoreKt ;
23+ import com .google .firebase .datastorage . JavaDataStorage ;
24+ import com .google .firebase .datastorage . JavaDataStorageKt ;
2525import java .text .SimpleDateFormat ;
2626import java .time .Instant ;
2727import java .time .LocalDateTime ;
@@ -58,17 +58,17 @@ class HeartBeatInfoStorage {
5858 // As soon as you hit the limit of heartbeats. The number of stored heartbeats is halved.
5959 private static final int HEART_BEAT_COUNT_LIMIT = 30 ;
6060
61- private final DataStorage firebaseDataStore ;
61+ private final JavaDataStorage firebaseDataStore ;
6262
6363 public HeartBeatInfoStorage (Context applicationContext , String persistenceKey ) {
6464 this .firebaseDataStore =
65- new DataStorage (applicationContext , HEARTBEAT_PREFERENCES_NAME + persistenceKey );
65+ new JavaDataStorage (applicationContext , HEARTBEAT_PREFERENCES_NAME + persistenceKey );
6666 }
6767
6868 @ VisibleForTesting
6969 @ RestrictTo (RestrictTo .Scope .TESTS )
70- HeartBeatInfoStorage (DataStorage dataStorage ) {
71- this .firebaseDataStore = dataStorage ;
70+ HeartBeatInfoStorage (JavaDataStorage javaDataStorage ) {
71+ this .firebaseDataStore = javaDataStorage ;
7272 }
7373
7474 @ VisibleForTesting
@@ -146,7 +146,7 @@ private synchronized void updateStoredUserAgent(
146146 MutablePreferences preferences , Preferences .Key <Set <String >> userAgent , String dateString ) {
147147 removeStoredDate (preferences , dateString );
148148 Set <String > userAgentDateSet =
149- new HashSet <>(DataStoreKt .getOrDefault (preferences , userAgent , new HashSet <>()));
149+ new HashSet <>(JavaDataStorageKt .getOrDefault (preferences , userAgent , new HashSet <>()));
150150 userAgentDateSet .add (dateString );
151151 preferences .set (userAgent , userAgentDateSet );
152152 }
@@ -158,7 +158,7 @@ private synchronized void removeStoredDate(MutablePreferences preferences, Strin
158158 return ;
159159 }
160160 Set <String > userAgentDateSet =
161- new HashSet <>(DataStoreKt .getOrDefault (preferences , userAgent , new HashSet <>()));
161+ new HashSet <>(JavaDataStorageKt .getOrDefault (preferences , userAgent , new HashSet <>()));
162162 userAgentDateSet .remove (dateString );
163163 if (userAgentDateSet .isEmpty ()) {
164164 preferences .remove (userAgent );
@@ -193,7 +193,7 @@ synchronized void storeHeartBeat(long millis, String userAgentString) {
193193 Preferences .Key <Set <String >> userAgent = PreferencesKeys .stringSetKey (userAgentString );
194194 firebaseDataStore .editSync (
195195 (pref ) -> {
196- String lastDateString = DataStoreKt .getOrDefault (pref , LAST_STORED_DATE , "" );
196+ String lastDateString = JavaDataStorageKt .getOrDefault (pref , LAST_STORED_DATE , "" );
197197 if (lastDateString .equals (dateString )) {
198198 Preferences .Key <Set <String >> storedUserAgent =
199199 getStoredUserAgentString (pref , dateString );
@@ -208,12 +208,12 @@ synchronized void storeHeartBeat(long millis, String userAgentString) {
208208 return null ;
209209 }
210210 }
211- long heartBeatCount = DataStoreKt .getOrDefault (pref , HEART_BEAT_COUNT_TAG , 0L );
211+ long heartBeatCount = JavaDataStorageKt .getOrDefault (pref , HEART_BEAT_COUNT_TAG , 0L );
212212 if (heartBeatCount + 1 == HEART_BEAT_COUNT_LIMIT ) {
213213 heartBeatCount = cleanUpStoredHeartBeats (pref );
214214 }
215215 Set <String > userAgentDateSet =
216- new HashSet <>(DataStoreKt .getOrDefault (pref , userAgent , new HashSet <>()));
216+ new HashSet <>(JavaDataStorageKt .getOrDefault (pref , userAgent , new HashSet <>()));
217217 userAgentDateSet .add (dateString );
218218 heartBeatCount += 1 ;
219219
@@ -226,7 +226,7 @@ synchronized void storeHeartBeat(long millis, String userAgentString) {
226226 }
227227
228228 private synchronized long cleanUpStoredHeartBeats (MutablePreferences preferences ) {
229- long heartBeatCount = DataStoreKt .getOrDefault (preferences , HEART_BEAT_COUNT_TAG , 0L );
229+ long heartBeatCount = JavaDataStorageKt .getOrDefault (preferences , HEART_BEAT_COUNT_TAG , 0L );
230230
231231 String lowestDate = null ;
232232 String userAgentString = "" ;
0 commit comments