lenses path

HTTPArchive · Jan 7, 2025 · 55adcf4 · 55adcf4
1 parent 5637e83
commit 55adcf4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/infra/bigquery-export/reports.js b/infra/bigquery-export/reports.js
@@ -6,6 +6,7 @@ export class ReportsExporter {
   constructor () {
     this.bigquery = new BigQueryExport()
     this.storage = new StorageExport()
+    this.storagePath = 'reports/dev/' // TODO change to prod
   }
 
   // Export timeseries reports
@@ -18,7 +19,7 @@ SELECT
 FROM reports.${metric}_timeseries
 `
     const rows = await this.bigquery.queryResults(query)
-    await this.storage.exportToJson(rows, metric)
+    await this.storage.exportToJson(rows, `${this.storagePath}${metric}.json`)
   }
 
   // Export monthly histogram report
@@ -41,6 +42,10 @@ WHERE date = '${date}'
       return
     }
 
+    if (exportData.lense && exportData.lense !== 'all') {
+      this.storagePath = this.storagePath + `${exportData.lense}/`
+    }
+
     if (exportData.type === 'histogram') {
       await this.exportHistogram(exportData)
     } else if (exportData.type === 'timeseries') {

diff --git a/infra/bigquery-export/storage.js b/infra/bigquery-export/storage.js
@@ -7,16 +7,13 @@ const storage = new Storage()
 export class StorageExport {
   constructor (bucket = 'httparchive') {
     this.bucket = bucket
-    this.storagePath = 'reports/dev/' // TODO change to prod
     this.stream = new Readable({
       objectMode: true,
       read () {}
     })
   }
 
   async exportToJson (data, fileName) {
-    fileName = this.storagePath + fileName + '.json'
-
     const bucket = storage.bucket(this.bucket)
     const file = bucket.file(fileName)