-
-
Save MichaelSolati/874543594145fa29691b883835a73460 to your computer and use it in GitHub Desktop.
| const admin = require("firebase-admin"); | |
| const { GeoFirestore, GeoCollectionReference } = require("geofirestore"); | |
| /** | |
| * Generate new private key by selecting a project from below: | |
| * https://console.firebase.google.com/u/0/project/_/settings/serviceaccounts/adminsdk | |
| */ | |
| const SERVICE_ACCOUNT = require("./serviceAccountKey.json"); | |
| /** | |
| * Set the collection to update here. | |
| */ | |
| const COLLECTION_NAME = "restaurants"; | |
| /** | |
| * If you use a custom key, define it here, otherwise leave this. | |
| */ | |
| const CUSTOM_KEY = "coordinates"; | |
| admin.initializeApp({ | |
| credential: admin.credential.cert(SERVICE_ACCOUNT), | |
| databaseURL: "https://" + SERVICE_ACCOUNT["project_id"] + ".firebaseio.com", | |
| }); | |
| const firestore = admin.firestore(); | |
| const geofirestore = new GeoFirestore(firestore); | |
| const collection = firestore.collection(COLLECTION_NAME); | |
| const geocollection = new GeoCollectionReference(collection); | |
| let totalUpdated = 0; | |
| function updateCollection(query) { | |
| return new Promise((resolve, reject) => | |
| deleteQueryBatch(query.limit(500), resolve, reject) | |
| ); | |
| } | |
| function deleteQueryBatch(query, resolve, reject) { | |
| query | |
| .get() | |
| .then((snapshot) => { | |
| // When there are no documents left, we are done | |
| if (snapshot.size === 0) { | |
| console.log(`DONE, UPDATED ${totalUpdated}`); | |
| return 0; | |
| } | |
| console.log( | |
| `UPDATING ${snapshot.size} GEODOCUMENTS, ALREADY UPDATED ${totalUpdated}` | |
| ); | |
| // Update documents in a batch | |
| const batch = geofirestore.batch(); | |
| snapshot.docs.forEach((doc) => { | |
| const geodoc = geocollection.doc(doc.id); | |
| batch.set(geodoc, doc.data().d, { customKey: CUSTOM_KEY }); | |
| }); | |
| return batch.commit().then(() => snapshot.size); | |
| }) | |
| .catch((e) => { | |
| console.warn( | |
| "This script was unable to convert your collection for GeoFirestore v4.", | |
| e.details | |
| ); | |
| return 0; | |
| }) | |
| .then((numUpdated) => { | |
| totalUpdated += numUpdated; | |
| if (numUpdated === 0) { | |
| resolve(); | |
| return; | |
| } | |
| process.nextTick(() => deleteQueryBatch(query, resolve, reject)); | |
| }) | |
| .catch((err) => reject(err)); | |
| } | |
| (async () => { | |
| await updateCollection(collection.orderBy("g").orderBy("l").orderBy("d")); | |
| })(); |
This script was designed to migrate collections created be geofirestore 3/2/1 to be compatible for 4. So if you used v3 before you'd run the script then use v4 of geofirestore.
Hi, I am using 3.4.1. currently. Are there any major improvements in 4.0.
Also I am using this in google cloud functions in firebase project. Can you tell me how to run this script?
Regards,
Shyam.
There are no major improvements in performance, just restructured data. As far as running the script I'd advise you go through the code above and make sure it makes sense for you. I don't want to be responsible for accidentally corrupting your data, so please read it and make sure you want to execute it. This is not code you'd run on a cloud function, it's something you'd run locally.
@MichaelSolati Thank you!
Also, I am getting empty results when I use where condition as below
const value = await query.where('notifications', '==', true).get();
I thought "notifications" field needs to be in all firestore documents of a collection. so updated all documents with "notifications" field but still the query returns empty set of documents.
Can you please let me know if it is possible to filter the query results using where clause?
Regards,
Shyam.
I think you can delete the line #20
You'd run the script once per collection in order to update it. After that you should use the 4+ library to add/set/anything rather than 3/2/1.