adds support for non US dataset locations

Author: archiewood

connection.schema.json

@@ -22,6 +22,12 @@
         "GCloud CLI"
       ],
       "default": "SERVICE_ACCOUNT"
+    },
+    "location": {
+      "title": "Location",
+      "type": "string",
+      "minLength": 1,
+      "default": "us"
     }
   },
   "dependencies": {
@@ -41,7 +47,7 @@
             }
           },
           "required": [
-            "authenticator", "keyfile"
+            "authenticator", "keyfile", "location"
           ]
         },
         {
@@ -63,7 +69,7 @@
             }
           },
           "required": [
-            "authenticator", "token", "projectId"
+            "authenticator", "token", "projectId", "location"
           ]
         },
         {
@@ -80,7 +86,7 @@
             }
           },
           "required": [
-            "authenticator", "projectId"
+            "authenticator", "projectId", "location"
           ]
         }
       ]

src/ls/driver.ts

@@ -29,10 +29,13 @@ export default class BigQueryDriver extends AbstractDriver<DriverLib, DriverOpti
 
   public async open() {
     const getCredentials = () => {
+
+      
       const authentication_method = this.credentials.authenticator
       if (authentication_method === 'CLI') {
         return {
-          projectId: this.credentials.projectId
+          projectId: this.credentials.projectId,
+          location: this.credentials.location
         };
       } else if (authentication_method === 'OAUTH') {
         // currently causing error
@@ -43,12 +46,14 @@ export default class BigQueryDriver extends AbstractDriver<DriverLib, DriverOpti
         return {
           // is this a legit way to handle this typescript error
           authClient: oauth as JSONClient,
-          projectId: this.credentials.projectId
+          projectId: this.credentials.projectId,
+          location: this.credentials.location
         };
       } else {
         console.log('keyfile', this.credentials.keyfile)
         return {
-          keyFilename: this.credentials.keyfile
+          keyFilename: this.credentials.keyfile,
+          location: this.credentials.location
         }
       };
     }
@@ -57,6 +62,7 @@ export default class BigQueryDriver extends AbstractDriver<DriverLib, DriverOpti
 
     this.connection = new Promise((resolve, reject) => {
       try {
+        console.log('connOptions', connOptions)
         const bigquery = new BigQuery({ ...connOptions, maxRetries: 10 });
         resolve(bigquery);
       } catch (error) {
@@ -91,6 +97,7 @@ export default class BigQueryDriver extends AbstractDriver<DriverLib, DriverOpti
     const options = {
       // typescript complains if this is not an array
       query: [query],
+      location: this.credentials.location,
     };
     const resultsAgg: NSDatabase.IResult[] = [];
 

ui.schema.json

@@ -16,6 +16,9 @@
   },
   "token": {
     "ui:widget": "password"
+  },
+  "location": {
+    "ui:help": "Default `us`. Each connection can access one location. Full location list: https://cloud.google.com/bigquery/docs/locations"
   }
 
 }