186 lines
8.1 KiB
Java
186 lines
8.1 KiB
Java
|
package com.google.firebase.firestore.local;
|
||
|
|
||
|
import android.database.Cursor;
|
||
|
import android.util.SparseArray;
|
||
|
import com.google.firebase.firestore.core.ListenSequence;
|
||
|
import com.google.firebase.firestore.local.LruGarbageCollector;
|
||
|
import com.google.firebase.firestore.model.DocumentKey;
|
||
|
import com.google.firebase.firestore.util.Assert;
|
||
|
import com.google.firebase.firestore.util.Consumer;
|
||
|
import com.google.firebase.firestore.util.Function;
|
||
|
import java.util.ArrayList;
|
||
|
import java.util.List;
|
||
|
|
||
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
/* loaded from: classes2.dex */
|
||
|
public class SQLiteLruReferenceDelegate implements ReferenceDelegate, LruDelegate {
|
||
|
private long currentSequenceNumber = -1;
|
||
|
private final LruGarbageCollector garbageCollector;
|
||
|
private ReferenceSet inMemoryPins;
|
||
|
private ListenSequence listenSequence;
|
||
|
private final SQLitePersistence persistence;
|
||
|
|
||
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
public SQLiteLruReferenceDelegate(SQLitePersistence sQLitePersistence, LruGarbageCollector.Params params) {
|
||
|
this.persistence = sQLitePersistence;
|
||
|
this.garbageCollector = new LruGarbageCollector(this, params);
|
||
|
}
|
||
|
|
||
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
public void start(long j) {
|
||
|
this.listenSequence = new ListenSequence(j);
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void onTransactionStarted() {
|
||
|
Assert.hardAssert(this.currentSequenceNumber == -1, "Starting a transaction without committing the previous one", new Object[0]);
|
||
|
this.currentSequenceNumber = this.listenSequence.next();
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void onTransactionCommitted() {
|
||
|
Assert.hardAssert(this.currentSequenceNumber != -1, "Committing a transaction without having started one", new Object[0]);
|
||
|
this.currentSequenceNumber = -1L;
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public long getCurrentSequenceNumber() {
|
||
|
Assert.hardAssert(this.currentSequenceNumber != -1, "Attempting to get a sequence number outside of a transaction", new Object[0]);
|
||
|
return this.currentSequenceNumber;
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public long getSequenceNumberCount() {
|
||
|
return this.persistence.getTargetCache().getTargetCount() + ((Long) this.persistence.query("SELECT COUNT(*) FROM (SELECT sequence_number FROM target_documents GROUP BY path HAVING COUNT(*) = 1 AND target_id = 0)").firstValue(new Function() { // from class: com.google.firebase.firestore.local.SQLiteLruReferenceDelegate$$ExternalSyntheticLambda0
|
||
|
@Override // com.google.firebase.firestore.util.Function
|
||
|
public final Object apply(Object obj) {
|
||
|
Long valueOf;
|
||
|
valueOf = Long.valueOf(((Cursor) obj).getLong(0));
|
||
|
return valueOf;
|
||
|
}
|
||
|
})).longValue();
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public void forEachTarget(Consumer<TargetData> consumer) {
|
||
|
this.persistence.getTargetCache().forEachTarget(consumer);
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public void forEachOrphanedDocumentSequenceNumber(final Consumer<Long> consumer) {
|
||
|
this.persistence.query("select sequence_number from target_documents group by path having COUNT(*) = 1 AND target_id = 0").forEach(new Consumer(consumer) { // from class: com.google.firebase.firestore.local.SQLiteLruReferenceDelegate$$ExternalSyntheticLambda1
|
||
|
public final Consumer f$0;
|
||
|
|
||
|
@Override // com.google.firebase.firestore.util.Consumer
|
||
|
public final void accept(Object obj) {
|
||
|
this.f$0.accept(Long.valueOf(((Cursor) obj).getLong(0)));
|
||
|
}
|
||
|
|
||
|
{
|
||
|
this.f$0 = consumer;
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void addReference(DocumentKey documentKey) {
|
||
|
writeSentinel(documentKey);
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void removeReference(DocumentKey documentKey) {
|
||
|
writeSentinel(documentKey);
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public int removeTargets(long j, SparseArray<?> sparseArray) {
|
||
|
return this.persistence.getTargetCache().removeQueries(j, sparseArray);
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void removeMutationReference(DocumentKey documentKey) {
|
||
|
writeSentinel(documentKey);
|
||
|
}
|
||
|
|
||
|
private boolean mutationQueuesContainKey(DocumentKey documentKey) {
|
||
|
return !this.persistence.query("SELECT 1 FROM document_mutations WHERE path = ?").binding(EncodedPath.encode(documentKey.getPath())).isEmpty();
|
||
|
}
|
||
|
|
||
|
private boolean isPinned(DocumentKey documentKey) {
|
||
|
if (this.inMemoryPins.containsKey(documentKey)) {
|
||
|
return true;
|
||
|
}
|
||
|
return mutationQueuesContainKey(documentKey);
|
||
|
}
|
||
|
|
||
|
private void removeSentinel(DocumentKey documentKey) {
|
||
|
this.persistence.execute("DELETE FROM target_documents WHERE path = ? AND target_id = 0", EncodedPath.encode(documentKey.getPath()));
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public int removeOrphanedDocuments(long j) {
|
||
|
final int[] iArr = new int[1];
|
||
|
final ArrayList arrayList = new ArrayList();
|
||
|
do {
|
||
|
} while (this.persistence.query("select path from target_documents group by path having COUNT(*) = 1 AND target_id = 0 AND sequence_number <= ? LIMIT ?").binding(Long.valueOf(j), 100).forEach(new Consumer(this, iArr, arrayList) { // from class: com.google.firebase.firestore.local.SQLiteLruReferenceDelegate$$ExternalSyntheticLambda2
|
||
|
public final SQLiteLruReferenceDelegate f$0;
|
||
|
public final int[] f$1;
|
||
|
public final List f$2;
|
||
|
|
||
|
@Override // com.google.firebase.firestore.util.Consumer
|
||
|
public final void accept(Object obj) {
|
||
|
this.f$0.m207x5fd3c655(this.f$1, this.f$2, (Cursor) obj);
|
||
|
}
|
||
|
|
||
|
{
|
||
|
this.f$0 = this;
|
||
|
this.f$1 = iArr;
|
||
|
this.f$2 = arrayList;
|
||
|
}
|
||
|
}) == 100);
|
||
|
this.persistence.getRemoteDocumentCache().removeAll(arrayList);
|
||
|
return iArr[0];
|
||
|
}
|
||
|
|
||
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
/* renamed from: lambda$removeOrphanedDocuments$2$com-google-firebase-firestore-local-SQLiteLruReferenceDelegate, reason: not valid java name */
|
||
|
public /* synthetic */ void m207x5fd3c655(int[] iArr, List list, Cursor cursor) {
|
||
|
DocumentKey fromPath = DocumentKey.fromPath(EncodedPath.decodeResourcePath(cursor.getString(0)));
|
||
|
if (isPinned(fromPath)) {
|
||
|
return;
|
||
|
}
|
||
|
iArr[0] = iArr[0] + 1;
|
||
|
list.add(fromPath);
|
||
|
removeSentinel(fromPath);
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void removeTarget(TargetData targetData) {
|
||
|
this.persistence.getTargetCache().updateTargetData(targetData.withSequenceNumber(getCurrentSequenceNumber()));
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void updateLimboDocument(DocumentKey documentKey) {
|
||
|
writeSentinel(documentKey);
|
||
|
}
|
||
|
|
||
|
private void writeSentinel(DocumentKey documentKey) {
|
||
|
this.persistence.execute("INSERT OR REPLACE INTO target_documents (target_id, path, sequence_number) VALUES (0, ?, ?)", EncodedPath.encode(documentKey.getPath()), Long.valueOf(getCurrentSequenceNumber()));
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public long getByteSize() {
|
||
|
return this.persistence.getByteSize();
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.ReferenceDelegate
|
||
|
public void setInMemoryPins(ReferenceSet referenceSet) {
|
||
|
this.inMemoryPins = referenceSet;
|
||
|
}
|
||
|
|
||
|
@Override // com.google.firebase.firestore.local.LruDelegate
|
||
|
public LruGarbageCollector getGarbageCollector() {
|
||
|
return this.garbageCollector;
|
||
|
}
|
||
|
}
|