feat(bigquery): parametrise location

ismith · Jan 16, 2019 · 53a1a32 · 53a1a32
1 parent a752065
commit 53a1a32
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/big_query.ml b/src/big_query.ml
@@ -437,7 +437,7 @@ module Jobs = struct
 
   let string (str : string) = Ok str
 
-  let query ?project_id ?(use_legacy_sql=false) ?(params = []) q : (query_response, [> Error.t ]) Lwt_result.t =
+  let query ?project_id ?(use_legacy_sql=false) ?(params = []) ?(location = "EU") q : (query_response, [> Error.t ]) Lwt_result.t =
     let parameter_mode =
       if use_legacy_sql || params = [] then
         None
@@ -449,7 +449,7 @@ module Jobs = struct
       { kind = "bigquery#queryRequest"
       ; query = q
       ; use_legacy_sql = use_legacy_sql
-      ; location = "EU"
+      ; location
       ; parameter_mode = parameter_mode
       ; query_parameters = params
       }

diff --git a/src/big_query.mli b/src/big_query.mli
@@ -95,7 +95,7 @@ module Jobs : sig
 
   val query_response_to_yojson : query_response -> Yojson.Safe.json
 
-  val query : ?project_id:string -> ?use_legacy_sql:bool -> ?params:Param.query_parameter list -> string -> (query_response, [> Error.t ]) Lwt_result.t
+  val query : ?project_id:string -> ?use_legacy_sql:bool -> ?params:Param.query_parameter list -> ?location:string -> string -> (query_response, [> Error.t ]) Lwt_result.t
   val get_query_results : ?page_token:string -> job_reference -> (query_response, [> Error.t ]) Lwt_result.t
 
   (** Type of query responses, when the job is complete and we definitely have some data. *)