Skip to content

Commit c2ec180

Browse files
committed
JIT: add explicit logging
1 parent ac0ba1f commit c2ec180

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/server/endpoints/covidcast.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from sqlalchemy import text
1212
from pandas import read_csv, to_datetime
1313

14-
from .._common import is_compatibility_mode, db
14+
from .._common import is_compatibility_mode, app, db
1515
from .._config import MAX_SMOOTHER_WINDOW
1616
from .._exceptions import ValidationFailedException, DatabaseErrorException
1717
from .._params import (
@@ -197,7 +197,9 @@ def alias_row(row):
197197
transform_args = parse_transform_args()
198198
pad_length = get_pad_length(source_signal_pairs, transform_args.get("smoother_window_length"))
199199
time_pairs = pad_time_pairs(time_pairs, pad_length)
200+
app.logger.info(f"JIT compute enabled for route '/': {source_signal_pairs}")
200201
source_signal_pairs, row_transform_generator = get_basename_signal_and_jit_generator(source_signal_pairs, transform_args=transform_args)
202+
app.logger.info(f"JIT base signals: {source_signal_pairs}")
201203

202204
def gen_transform(rows):
203205
parsed_rows = (parse_row(row, fields_string, fields_int, fields_float) for row in rows)
@@ -284,7 +286,9 @@ def gen_trend(rows):
284286
use_server_side_compute = all((is_day, is_also_day)) and JIT_COMPUTE and not jit_bypass
285287
if use_server_side_compute:
286288
pad_length = get_pad_length(source_signal_pairs, transform_args.get("smoother_window_length"))
289+
app.logger.info(f"JIT compute enabled for route '/trend': {source_signal_pairs}")
287290
source_signal_pairs, row_transform_generator = get_basename_signal_and_jit_generator(source_signal_pairs)
291+
app.logger.info(f"JIT base signals: {source_signal_pairs}")
288292
time_window = pad_time_window(time_window, pad_length)
289293

290294
def gen_transform(rows):
@@ -362,7 +366,9 @@ def gen_trend(rows):
362366
use_server_side_compute = is_day and JIT_COMPUTE and not jit_bypass
363367
if use_server_side_compute:
364368
pad_length = get_pad_length(source_signal_pairs, transform_args.get("smoother_window_length"))
369+
app.logger.info(f"JIT compute enabled for route '/trendseries': {source_signal_pairs}")
365370
source_signal_pairs, row_transform_generator = get_basename_signal_and_jit_generator(source_signal_pairs)
371+
app.logger.info(f"JIT base signals: {source_signal_pairs}")
366372
time_window = pad_time_window(time_window, pad_length)
367373

368374
def gen_transform(rows):
@@ -507,7 +513,9 @@ def handle_export():
507513
use_server_side_compute = all([is_day, is_end_day]) and JIT_COMPUTE and not jit_bypass
508514
if use_server_side_compute:
509515
pad_length = get_pad_length(source_signal_pairs, transform_args.get("smoother_window_length"))
516+
app.logger.info(f"JIT compute enabled for route '/csv': {source_signal_pairs}")
510517
source_signal_pairs, row_transform_generator = get_basename_signal_and_jit_generator(source_signal_pairs)
518+
app.logger.info(f"JIT base signals: {source_signal_pairs}")
511519
time_window = pad_time_window(time_window, pad_length)
512520

513521
def gen_transform(rows):

0 commit comments

Comments
 (0)