MahsaSeifikar commented on code in PR #19742: URL: https://github.com/apache/kafka/pull/19742#discussion_r2127827598
########## core/src/main/scala/kafka/server/ClientQuotaManager.scala: ########## @@ -451,6 +450,35 @@ class ClientQuotaManager(private val config: ClientQuotaManagerConfig, } } + /** + * Helper method to update quotaTypesEnabled which is a bitwise OR combination of the enabled quota types. + * For example: + * - If UserQuotaEnabled = 2 and ClientIdQuotaEnabled = 1, then quotaTypesEnabled = 3 (2 | 1 = 3) + * - If UserClientIdQuotaEnabled = 4 and UserQuotaEnabled = 1, then quotaTypesEnabled = (4 | 1 = 5) + * - If UserClientIdQuotaEnabled = 4 and ClientIdQuotaEnabled = 2, then quotaTypesEnabled = 6 (4 | 2 = 6) + * - If all three are enabled (1 | 2 | 4), then quotaTypesEnabled = 7 + */ + private def updateQuotaTypes(): Unit = { + quotaTypesEnabled = if (clientQuotaCallbackPlugin.isDefined) { + QuotaTypes.CustomQuotas + } else { + QuotaTypes.NoQuotas + } + + activeQuotaEntities.forEach { + case KafkaQuotaEntity(Some(_), Some(_)) => + quotaTypesEnabled |= QuotaTypes.UserClientIdQuotaEnabled + case KafkaQuotaEntity(Some(_), None) => + quotaTypesEnabled |= QuotaTypes.UserQuotaEnabled + case KafkaQuotaEntity(None, Some(_)) => + quotaTypesEnabled |= QuotaTypes.ClientIdQuotaEnabled + case _ => // Unexpected entity type + } + + val activeEntities = if (activeQuotaEntities.isEmpty) "No active quota entities" else activeQuotaEntities.asScala.map(_.toString).mkString(", ") + info(s"Quota types enabled has been changed with active quota entities: [$activeEntities]") Review Comment: I updated this to show just quota types not all active entities. In case of an incident or issue, we need to know which quota types are active. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org