def METHOD_NAME(
    query, snuba_params, data_source=Dataset.Discover
) -> dict[str, list[str]]:
    select_column, value = _get_select_column(query)
    query = query + FILTER_HAS_A_REPLAY if data_source == Dataset.Discover else query
    if select_column == "replay_id":
        # just return a mapping of replay_id:replay_id instead of hitting discover
        # if we want to validate list of replay_ids existence
        return {v: [v] for v in value}
    builder = QueryBuilder(
        dataset=data_source,
        params={},
        snuba_params=snuba_params,
        selected_columns=["group_uniq_array(100,replayId)", select_column],
        query=query,
        limit=25,
        offset=0,
        config=QueryBuilderConfig(
            functions_acl=["group_uniq_array"],
        ),
    )
    discover_results = builder.run_query(
        referrer="api.organization-issue-replay-count", use_cache=True
    )
    replay_id_to_issue_map = defaultdict(list)
    for row in discover_results["data"]:
        for replay_id in row["group_uniq_array_100_replayId"]:
            replay_id_to_issue_map[replay_id].append(row[select_column])
    return replay_id_to_issue_map