Examples with TargetChange com.google.firestore.v1.TargetChange used on opensource projects

Example 1 with TargetChange

use of com.google.firestore.v1.TargetChange in project firebase-android-sdk by firebase.

the class RemoteSerializer method decodeWatchChange.

// Watch changes
public WatchChange decodeWatchChange(ListenResponse protoChange) {
    WatchChange watchChange;
    switch(protoChange.getResponseTypeCase()) {
        case TARGET_CHANGE:
            com.google.firestore.v1.TargetChange targetChange = protoChange.getTargetChange();
            WatchTargetChangeType changeType;
            Status cause = null;
            switch(targetChange.getTargetChangeType()) {
                case NO_CHANGE:
                    changeType = WatchTargetChangeType.NoChange;
                    break;
                case ADD:
                    changeType = WatchTargetChangeType.Added;
                    break;
                case REMOVE:
                    changeType = WatchTargetChangeType.Removed;
                    cause = fromStatus(targetChange.getCause());
                    break;
                case CURRENT:
                    changeType = WatchTargetChangeType.Current;
                    break;
                case RESET:
                    changeType = WatchTargetChangeType.Reset;
                    break;
                case UNRECOGNIZED:
                default:
                    throw new IllegalArgumentException("Unknown target change type");
            }
            watchChange = new WatchTargetChange(changeType, targetChange.getTargetIdsList(), targetChange.getResumeToken(), cause);
            break;
        case DOCUMENT_CHANGE:
            DocumentChange docChange = protoChange.getDocumentChange();
            List<Integer> added = docChange.getTargetIdsList();
            List<Integer> removed = docChange.getRemovedTargetIdsList();
            DocumentKey key = decodeKey(docChange.getDocument().getName());
            SnapshotVersion version = decodeVersion(docChange.getDocument().getUpdateTime());
            hardAssert(!version.equals(SnapshotVersion.NONE), "Got a document change without an update time");
            ObjectValue data = ObjectValue.fromMap(docChange.getDocument().getFieldsMap());
            MutableDocument document = MutableDocument.newFoundDocument(key, version, data);
            watchChange = new WatchChange.DocumentChange(added, removed, document.getKey(), document);
            break;
        case DOCUMENT_DELETE:
            DocumentDelete docDelete = protoChange.getDocumentDelete();
            removed = docDelete.getRemovedTargetIdsList();
            key = decodeKey(docDelete.getDocument());
            // Note that version might be unset in which case we use SnapshotVersion.NONE
            version = decodeVersion(docDelete.getReadTime());
            MutableDocument doc = MutableDocument.newNoDocument(key, version);
            watchChange = new WatchChange.DocumentChange(Collections.emptyList(), removed, doc.getKey(), doc);
            break;
        case DOCUMENT_REMOVE:
            DocumentRemove docRemove = protoChange.getDocumentRemove();
            removed = docRemove.getRemovedTargetIdsList();
            key = decodeKey(docRemove.getDocument());
            watchChange = new WatchChange.DocumentChange(Collections.emptyList(), removed, key, null);
            break;
        case FILTER:
            com.google.firestore.v1.ExistenceFilter protoFilter = protoChange.getFilter();
            // TODO: implement existence filter parsing (see b/33076578)
            ExistenceFilter filter = new ExistenceFilter(protoFilter.getCount());
            int targetId = protoFilter.getTargetId();
            watchChange = new ExistenceFilterWatchChange(targetId, filter);
            break;
        case RESPONSETYPE_NOT_SET:
        default:
            throw new IllegalArgumentException("Unknown change type set");
    }
    return watchChange;
}

Example 2 with TargetChange

use of com.google.firestore.v1.TargetChange in project java-firestore by googleapis.

the class Watch method onResponse.

@Override
public synchronized void onResponse(ListenResponse listenResponse) {
    switch(listenResponse.getResponseTypeCase()) {
        case TARGET_CHANGE:
            TargetChange change = listenResponse.getTargetChange();
            boolean noTargetIds = change.getTargetIdsCount() == 0;
            switch(change.getTargetChangeType()) {
                case NO_CHANGE:
                    if (noTargetIds && change.hasReadTime() && current) {
                        // This means everything is up-to-date, so emit the current set of docs as a snapshot,
                        // if there were changes.
                        pushSnapshot(Timestamp.fromProto(change.getReadTime()), change.getResumeToken());
                    }
                    break;
                case ADD:
                    if (WATCH_TARGET_ID != change.getTargetIds(0)) {
                        closeStream(FirestoreException.forInvalidArgument("Target ID must be 0x01"));
                    }
                    break;
                case REMOVE:
                    Status status = change.hasCause() ? Status.fromCodeValue(change.getCause().getCode()) : Status.CANCELLED;
                    closeStream(FirestoreException.forServerRejection(status, "Backend ended Listen stream: " + change.getCause().getMessage()));
                    break;
                case CURRENT:
                    current = true;
                    break;
                case RESET:
                    resetDocs();
                    break;
                default:
                    closeStream(FirestoreException.forInvalidArgument("Encountered invalid target change type: " + change.getTargetChangeType()));
            }
            if (change.getResumeToken() != null && affectsTarget(change.getTargetIdsList(), WATCH_TARGET_ID)) {
                nextAttempt = backoff.createFirstAttempt();
            }
            break;
        case DOCUMENT_CHANGE:
            // No other targetIds can show up here, but we still need to see if the targetId was in the
            // added list or removed list.
            List<Integer> targetIds = listenResponse.getDocumentChange().getTargetIdsList();
            List<Integer> removedTargetIds = listenResponse.getDocumentChange().getRemovedTargetIdsList();
            boolean changed = targetIds.contains(WATCH_TARGET_ID);
            boolean removed = removedTargetIds.contains(WATCH_TARGET_ID);
            Document document = listenResponse.getDocumentChange().getDocument();
            ResourcePath name = ResourcePath.create(document.getName());
            if (changed) {
                changeMap.put(name, document);
            } else if (removed) {
                changeMap.put(name, null);
            }
            break;
        case DOCUMENT_DELETE:
            changeMap.put(ResourcePath.create(listenResponse.getDocumentDelete().getDocument()), null);
            break;
        case DOCUMENT_REMOVE:
            changeMap.put(ResourcePath.create(listenResponse.getDocumentRemove().getDocument()), null);
            break;
        case FILTER:
            if (listenResponse.getFilter().getCount() != currentSize()) {
                // We need to remove all the current results.
                resetDocs();
                // The filter didn't match, so re-issue the query.
                resetStream();
            }
            break;
        default:
            closeStream(FirestoreException.forInvalidArgument("Encountered invalid listen response type"));
            break;
    }
}