Skip to content

Commit

Permalink
Use instrument, instead of salIndex, from nextVisit messages
Browse files Browse the repository at this point in the history
  • Loading branch information
hsinfang committed Jun 25, 2024
1 parent 38c0266 commit 041c87f
Showing 1 changed file with 59 additions and 59 deletions.
118 changes: 59 additions & 59 deletions src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -320,12 +320,8 @@ async def main() -> None:
],
)

# Still use salIndex to know the instrument and its active
# detectors, despite that instrument is also known from the
# next_visit message. HSC has extra active detector
# configurations just for the upload.py test.
match next_visit_message_updated.salIndex:
case 2: # LATISS
match next_visit_message_updated.instrument:
case "LATISS":
latiss_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
Expand All @@ -335,7 +331,7 @@ async def main() -> None:
)
knative_serving_url = latiss_knative_serving_url
in_process_requests_gauge = latiss_in_process_requests_gauge
case 3: # LSSTComCamSim
case "LSSTComCamSim":
lsstcomcamsim_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
Expand All @@ -346,7 +342,7 @@ async def main() -> None:
)
knative_serving_url = lsstcomcamsim_knative_serving_url
in_process_requests_gauge = lsstcomcamsim_in_process_requests_gauge
case 1: # LSSTCam
case "LSSTCam":
lsstcam_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
Expand All @@ -358,59 +354,63 @@ async def main() -> None:
in_process_requests_gauge = (
lsstcam_in_process_requests_gauge
)
case 999: # HSC
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59134: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59134,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59142: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59142,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59150: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59150,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59160: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59160,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case "HSC":
# HSC has extra active detector configurations just for the
# upload.py test.
match next_visit_message_updated.salIndex:
case 999: # HSC datasets from using upload_from_repo.py
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59134: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59134,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59142: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59142,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59150: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59150,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case 59160: # HSC upload.py test dataset
hsc_gauge.inc()
fan_out_message_list = (
next_visit_message_updated.add_detectors(
dataclasses.asdict(next_visit_message_updated),
hsc_active_detectors_59160,
)
)
knative_serving_url = hsc_knative_serving_url
in_process_requests_gauge = hsc_in_process_requests_gauge
case _:
raise Exception(
f"no matching case for salIndex {next_visit_message_updated.salIndex} to know the instrument"
f"no matching case for instrument {next_visit_message_updated.instrument}."
)

try:
Expand Down

0 comments on commit 041c87f

Please sign in to comment.