chore(GLAM): Support setting of slot reservation and enable it for Glean jobs (#8557)

Author: edugfilho

script/glam/generate_and_run_desktop_sql

@@ -11,7 +11,8 @@ IMPORT=true source script/glam/run_glam_sql
 function generate_and_run_sql {
     local probe_type=$1
     local windows_release_sample_percent=$2
-    local process=$3
+    local use_slots=$3
+    local process=$4
     local script_type
     local destination_table
     local write_disposition
@@ -60,12 +61,12 @@ function generate_and_run_sql {
     run_query "clients_daily_${script_type}_aggregates_v1" \
         true 0 99 "clients_daily_${probe_type}_aggregates_${table_name_suffix}v1" \
         "$write_disposition --clustering_fields=app_version,channel --time_partitioning_field=submission_date" \
-        "$(if [[ -n $windows_release_sample_percent ]]; then echo $windows_release_sample_percent; fi)"
-
+        "$(if [[ -n $windows_release_sample_percent ]]; then echo $windows_release_sample_percent; fi)" \
+        "$(if [[ -n $use_slots ]]; then echo $use_slots; fi)"
 }
 
 probe_types=${1:-"scalar" "keyed_boolean" "keyed_scalar" "histogram" "keyed_histogram"}
 
 for probe_type in ${probe_types[@]}; do
-    generate_and_run_sql $probe_type $2 $3
+    generate_and_run_sql $probe_type $2 $3 $4
 done

script/glam/run_glam_sql

@@ -52,6 +52,7 @@ function run_query {
     local query_location=${5:-$destination_table}
     local additional_arguments="${6:---replace}"
     local sample_size=${7:-10}
+    local use_slots=${8:-true}
     local query="sql/$PROJECT/$PROD_DATASET/$query_location/query.sql"
 
     # add an option to write to a time-partitioned table
@@ -74,6 +75,7 @@ function run_query {
         --parameter="max_sample_id:INT64:$max_sample_id" \
         --parameter="sample_size:INT64:$sample_size" \
         "$(if $time_partition; then echo --time_partitioning_type="DAY"; fi)" \
+        "$(if $use_slots; then echo --reservation_id=glam; fi)" \
         < "$tmp"
 }