Function get_edge_overrides_data

api/features/features_service.py:76–114

Get the number of identity / segment overrides in a given environment for each feature in the project. Retrieve identity override data from DynamoDB. :param environment: the environment to get the overrides data for :return OverridesData: dictionary of {feature_id: EnvironmentF

(environment: "Environment")

def get_edge_overrides_data(environment: "Environment") -> OverridesData:
    """
    Get the number of identity / segment overrides in a given environment for each feature in the
    project.
    Retrieve identity override data from DynamoDB.

    :param environment: the environment to get the overrides data for
    :return OverridesData: dictionary of {feature_id: EnvironmentFeatureOverridesData}
    """

    with ThreadPoolExecutor() as executor:
        # Note: We intentionally let the get_environment_flags_list
        # call happen on the main thread to simplify testing.
        # The call to dynamo happens on a separate thread, which
        # still gives us the parallelism we need without needing
        # an extra thread. This does mean that the order of execution
        # is important here.
        get_overrides_data_future = executor.submit(
            get_edge_identity_override_keys,
            environment_id=environment.id,
        )
        flags_list = get_environment_flags_list(environment)
    all_overrides_data: OverridesData = {}

    for feature_state in flags_list:
        env_feature_overrides_data = all_overrides_data.setdefault(
            feature_state.feature_id, EnvironmentFeatureOverridesData()
        )
        if feature_state.feature_segment_id:
            env_feature_overrides_data.num_segment_overrides += 1
    for identity_override_key in get_overrides_data_future.result():
        feature_id = get_feature_id_from_identity_override_document_key(
            identity_override_key
        )
        # Only override features that exists in core
        if feature_id in all_overrides_data:
            all_overrides_data[feature_id].add_identity_override()  # type: ignore[no-untyped-call]

    return all_overrides_data