mirror of
https://github.com/ToolJet/ToolJet
synced 2026-05-21 16:08:35 +00:00
3ca60c5b2d
* Add multiple fields for order and orderType in firestore query * Add support for querying firestore with where, order and limit * Minor refactoring updates for firestore query operations file * Minor refactoring for firestore query operations * Update package-lock.json with npm 7.20.0
91 lines
2.3 KiB
TypeScript
91 lines
2.3 KiB
TypeScript
import { Firestore, Query } from "@google-cloud/firestore";
|
|
|
|
export async function queryCollection(
|
|
db: Firestore,
|
|
collection: string,
|
|
limit: number,
|
|
where_operation: any,
|
|
where_field: string,
|
|
where_value: any,
|
|
order_field: string,
|
|
order_type: any,
|
|
): Promise<object> {
|
|
const limitProvided = isNaN(limit) !== true;
|
|
const whereConditionProvided =
|
|
([undefined, ''].includes(where_field) === false) &&
|
|
([undefined, ''].includes(where_operation) === false) &&
|
|
(where_value != undefined);
|
|
const orderProvided = ([undefined, ''].includes(order_field) === false);
|
|
|
|
const collectionRef = db.collection(collection);
|
|
let query: Query = collectionRef;
|
|
|
|
if (whereConditionProvided)
|
|
query = query.where(where_field, where_operation, where_value);
|
|
if (limitProvided)
|
|
query = query.limit(limit);
|
|
if (orderProvided)
|
|
query = query.orderBy(order_field, order_type);
|
|
|
|
const snapshot = await query.get();
|
|
|
|
const data = [];
|
|
snapshot.forEach((doc) => {
|
|
data.push({
|
|
document_id: doc.id,
|
|
data: doc.data(),
|
|
});
|
|
});
|
|
|
|
return data;
|
|
}
|
|
|
|
export async function getDocument(db, path: string): Promise<object> {
|
|
const docRef = db.doc(path);
|
|
const doc = await docRef.get();
|
|
// if (!doc.exists) {
|
|
|
|
return doc.data();
|
|
}
|
|
|
|
export async function setDocument(db, path: string, body: string): Promise<object> {
|
|
const docRef = db.doc(path);
|
|
const result = await docRef.set(JSON.parse(body));
|
|
|
|
return result;
|
|
}
|
|
|
|
export async function addDocument(db, path: string, body: string): Promise<object> {
|
|
const docRef = db.doc(path);
|
|
const result = await docRef.set(JSON.parse(body));
|
|
|
|
return result;
|
|
}
|
|
|
|
export async function updateDocument(db, path: string, body: string): Promise<object> {
|
|
const docRef = db.doc(path);
|
|
const result = await docRef.update(JSON.parse(body));
|
|
|
|
return result;
|
|
}
|
|
|
|
export async function deleteDocument(db, path: string): Promise<object> {
|
|
const docRef = db.doc(path);
|
|
const result = await docRef.delete();
|
|
|
|
return result;
|
|
}
|
|
|
|
export async function bulkUpdate(
|
|
db,
|
|
collection: string,
|
|
records: Array<object>,
|
|
documentIdKey: string
|
|
): Promise<object> {
|
|
for (const record of records) {
|
|
const path = `${collection}/${record[documentIdKey]}`;
|
|
updateDocument(db, path, JSON.stringify(record));
|
|
}
|
|
|
|
return {};
|
|
}
|