Update bigquery.gs (#292)

Co-authored-by: Priyankarp24 <[email protected]>

Author: RajeshGogo

advanced/bigquery.gs

@@ -20,59 +20,60 @@
 function runQuery() {
   // Replace this value with the project ID listed in the Google
   // Cloud Platform project.
-  var projectId = 'XXXXXXXX';
+  const projectId = 'XXXXXXXX';
 
-  var request = {
+  const request = {
+    // TODO (developer) - Replace query with yours
     query: 'SELECT TOP(word, 300) AS word, COUNT(*) AS word_count ' +
       'FROM `publicdata.samples.shakespeare` WHERE LENGTH(word) > 10;',
     useLegacySql: false
   };
-  var queryResults = BigQuery.Jobs.query(request, projectId);
-  var jobId = queryResults.jobReference.jobId;
+  let queryResults = BigQuery.Jobs.query(request, projectId);
+  const jobId = queryResults.jobReference.jobId;
 
   // Check on status of the Query Job.
-  var sleepTimeMs = 500;
+  let sleepTimeMs = 500;
   while (!queryResults.jobComplete) {
     Utilities.sleep(sleepTimeMs);
     sleepTimeMs *= 2;
     queryResults = BigQuery.Jobs.getQueryResults(projectId, jobId);
   }
 
   // Get all the rows of results.
-  var rows = queryResults.rows;
+  let rows = queryResults.rows;
   while (queryResults.pageToken) {
     queryResults = BigQuery.Jobs.getQueryResults(projectId, jobId, {
       pageToken: queryResults.pageToken
     });
     rows = rows.concat(queryResults.rows);
   }
 
-  if (rows) {
-    var spreadsheet = SpreadsheetApp.create('BiqQuery Results');
-    var sheet = spreadsheet.getActiveSheet();
+  if (!rows) {
+    Logger.log('No rows returned.');
+    return;
+  }
+  const spreadsheet = SpreadsheetApp.create('BiqQuery Results');
+  const sheet = spreadsheet.getActiveSheet();
 
-    // Append the headers.
-    var headers = queryResults.schema.fields.map(function(field) {
-      return field.name;
-    });
-    sheet.appendRow(headers);
+  // Append the headers.
+  const headers = queryResults.schema.fields.map(function(field) {
+    return field.name;
+  });
+  sheet.appendRow(headers);
 
-    // Append the results.
-    var data = new Array(rows.length);
-    for (var i = 0; i < rows.length; i++) {
-      var cols = rows[i].f;
-      data[i] = new Array(cols.length);
-      for (var j = 0; j < cols.length; j++) {
-        data[i][j] = cols[j].v;
-      }
+  // Append the results.
+  var data = new Array(rows.length);
+  for (let i = 0; i < rows.length; i++) {
+    const cols = rows[i].f;
+    data[i] = new Array(cols.length);
+    for (let j = 0; j < cols.length; j++) {
+      data[i][j] = cols[j].v;
     }
-    sheet.getRange(2, 1, rows.length, headers.length).setValues(data);
-
-    Logger.log('Results spreadsheet created: %s',
-        spreadsheet.getUrl());
-  } else {
-    Logger.log('No rows returned.');
   }
+  sheet.getRange(2, 1, rows.length, headers.length).setValues(data);
+
+  Logger.log('Results spreadsheet created: %s',
+      spreadsheet.getUrl());
 }
 // [END apps_script_bigquery_run_query]
 
@@ -83,17 +84,17 @@ function runQuery() {
 function loadCsv() {
   // Replace this value with the project ID listed in the Google
   // Cloud Platform project.
-  var projectId = 'XXXXXXXX';
+  const projectId = 'XXXXXXXX';
   // Create a dataset in the BigQuery UI (https://bigquery.cloud.google.com)
   // and enter its ID below.
-  var datasetId = 'YYYYYYYY';
+  const datasetId = 'YYYYYYYY';
   // Sample CSV file of Google Trends data conforming to the schema below.
   // https://docs.google.com/file/d/0BwzA1Orbvy5WMXFLaTR1Z1p2UDg/edit
-  var csvFileId = '0BwzA1Orbvy5WMXFLaTR1Z1p2UDg';
+  const csvFileId = '0BwzA1Orbvy5WMXFLaTR1Z1p2UDg';
 
   // Create the table.
-  var tableId = 'pets_' + new Date().getTime();
-  var table = {
+  const tableId = 'pets_' + new Date().getTime();
+  let table = {
     tableReference: {
       projectId: projectId,
       datasetId: datasetId,
@@ -108,15 +109,18 @@ function loadCsv() {
       ]
     }
   };
-  table = BigQuery.Tables.insert(table, projectId, datasetId);
-  Logger.log('Table created: %s', table.id);
-
+  try {
+    table = BigQuery.Tables.insert(table, projectId, datasetId);
+    Logger.log('Table created: %s', table.id);
+  } catch (err) {
+    Logger.log('unable to create table');
+  }
   // Load CSV data from Drive and convert to the correct format for upload.
-  var file = DriveApp.getFileById(csvFileId);
-  var data = file.getBlob().setContentType('application/octet-stream');
+  const file = DriveApp.getFileById(csvFileId);
+  const data = file.getBlob().setContentType('application/octet-stream');
 
   // Create the data upload job.
-  var job = {
+  const job = {
     configuration: {
       load: {
         destinationTable: {
@@ -128,8 +132,12 @@ function loadCsv() {
       }
     }
   };
-  job = BigQuery.Jobs.insert(job, projectId, data);
-  Logger.log('Load job started. Check on the status of it here: ' +
+  try {
+    BigQuery.Jobs.insert(job, projectId, data);
+    Logger.log('Load job started. Check on the status of it here: ' +
       'https://bigquery.cloud.google.com/jobs/%s', projectId);
+  } catch (err) {
+    Logger.log('unable to insert job');
+  }
 }
 // [END apps_script_bigquery_load_csv]