Skip to content

Commit a3818a1

Browse files
mukundv-chromeinferno-chromium
authored andcommitted
BigQuery fix for Fuzz Strategy Selection Cron (#723)
* Only upload to bigquery if data is present. * Added a log message.
1 parent 823fe0e commit a3818a1

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/appengine/handlers/cron/fuzz_strategy_selection.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from google_cloud_utils import big_query
2828
from handlers import base_handler
2929
from libs import handler
30+
from metrics import logs
3031

3132
HIGH_TEMPERATURE_PARAMETER = .75
3233
MEDIUM_TEMPERATURE_PARAMETER = .5
@@ -176,9 +177,13 @@ def _store_probabilities_in_bigquery(data):
176177
}
177178
bigquery_data.append(big_query.Insert(row=bigquery_row, insert_id=None))
178179

179-
client = big_query.Client(
180-
dataset_id='main', table_id='fuzz_strategy_experiments')
181-
client.insert(bigquery_data)
180+
if bigquery_data:
181+
client = big_query.Client(
182+
dataset_id='main', table_id='fuzz_strategy_experiments')
183+
client.insert(bigquery_data)
184+
else:
185+
logs.log("No fuzz strategy distribution data was found to upload to "
186+
"BigQuery.")
182187

183188

184189
def _query_and_upload_strategy_probabilities():

0 commit comments

Comments
 (0)