@@ -282,6 +282,14 @@ async def add_latency(self, method, latency):
282282 self ._treatment_with_config [latency_bucket ] += 1
283283 elif method == MethodExceptionsAndLatencies .TREATMENTS_WITH_CONFIG :
284284 self ._treatments_with_config [latency_bucket ] += 1
285+ elif method == MethodExceptionsAndLatencies .TREATMENTS_BY_FLAG_SET :
286+ self ._treatments_by_flag_set [latency_bucket ] += 1
287+ elif method == MethodExceptionsAndLatencies .TREATMENTS_BY_FLAG_SETS :
288+ self ._treatments_by_flag_sets [latency_bucket ] += 1
289+ elif method == MethodExceptionsAndLatencies .TREATMENTS_WITH_CONFIG_BY_FLAG_SET :
290+ self ._treatments_with_config_by_flag_set [latency_bucket ] += 1
291+ elif method == MethodExceptionsAndLatencies .TREATMENTS_WITH_CONFIG_BY_FLAG_SETS :
292+ self ._treatments_with_config_by_flag_sets [latency_bucket ] += 1
285293 elif method == MethodExceptionsAndLatencies .TRACK :
286294 self ._track [latency_bucket ] += 1
287295 else :
@@ -573,6 +581,14 @@ async def add_exception(self, method):
573581 self ._treatment_with_config += 1
574582 elif method == MethodExceptionsAndLatencies .TREATMENTS_WITH_CONFIG :
575583 self ._treatments_with_config += 1
584+ elif method == MethodExceptionsAndLatencies .TREATMENTS_BY_FLAG_SET :
585+ self ._treatments_by_flag_set += 1
586+ elif method == MethodExceptionsAndLatencies .TREATMENTS_BY_FLAG_SETS :
587+ self ._treatments_by_flag_sets += 1
588+ elif method == MethodExceptionsAndLatencies .TREATMENTS_WITH_CONFIG_BY_FLAG_SET :
589+ self ._treatments_with_config_by_flag_set += 1
590+ elif method == MethodExceptionsAndLatencies .TREATMENTS_WITH_CONFIG_BY_FLAG_SETS :
591+ self ._treatments_with_config_by_flag_sets += 1
576592 elif method == MethodExceptionsAndLatencies .TRACK :
577593 self ._track += 1
578594 else :
0 commit comments