kafka/s: Wire-up usage manager

graphcareful · Mar 16, 2023 · e0b6c7c · e0b6c7c
1 parent e22d9f0
commit e0b6c7c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/src/v/kafka/server/server.cc b/src/v/kafka/server/server.cc
@@ -43,6 +43,7 @@
 #include "kafka/server/quota_manager.h"
 #include "kafka/server/request_context.h"
 #include "kafka/server/response.h"
+#include "kafka/server/usage_manager.h"
 #include "net/connection.h"
 #include "security/errc.h"
 #include "security/exceptions.h"
@@ -83,6 +84,7 @@ server::server(
   ss::sharded<quota_manager>& quota,
   ss::sharded<snc_quota_manager>& snc_quota_mgr,
   ss::sharded<kafka::group_router>& router,
+  ss::sharded<kafka::usage_manager>& usage_manager,
   ss::sharded<cluster::shard_table>& tbl,
   ss::sharded<cluster::partition_manager>& pm,
   ss::sharded<fetch_session_cache>& session_cache,
@@ -104,6 +106,7 @@ server::server(
   , _quota_mgr(quota)
   , _snc_quota_mgr(snc_quota_mgr)
   , _group_router(router)
+  , _usage_manager(usage_manager)
   , _shard_table(tbl)
   , _partition_manager(pm)
   , _fetch_session_cache(session_cache)

diff --git a/src/v/kafka/server/server.h b/src/v/kafka/server/server.h
@@ -45,6 +45,7 @@ class server final : public net::server {
       ss::sharded<quota_manager>&,
       ss::sharded<snc_quota_manager>&,
       ss::sharded<kafka::group_router>&,
+      ss::sharded<kafka::usage_manager>&,
       ss::sharded<cluster::shard_table>&,
       ss::sharded<cluster::partition_manager>&,
       ss::sharded<fetch_session_cache>&,
@@ -102,6 +103,7 @@ class server final : public net::server {
         return _fetch_session_cache.local();
     }
     quota_manager& quota_mgr() { return _quota_mgr.local(); }
+    usage_manager& usage_mgr() { return _usage_manager.local(); }
     snc_quota_manager& snc_quota_mgr() { return _snc_quota_mgr.local(); }
     bool is_idempotence_enabled() const { return _is_idempotence_enabled; }
     bool are_transactions_enabled() const { return _are_transactions_enabled; }
@@ -153,6 +155,7 @@ class server final : public net::server {
     ss::sharded<quota_manager>& _quota_mgr;
     ss::sharded<snc_quota_manager>& _snc_quota_mgr;
     ss::sharded<kafka::group_router>& _group_router;
+    ss::sharded<kafka::usage_manager>& _usage_manager;
     ss::sharded<cluster::shard_table>& _shard_table;
     ss::sharded<cluster::partition_manager>& _partition_manager;
     ss::sharded<kafka::fetch_session_cache>& _fetch_session_cache;

diff --git a/src/v/redpanda/tests/fixture.h b/src/v/redpanda/tests/fixture.h
@@ -120,6 +120,7 @@ class redpanda_thread_fixture {
           app.quota_mgr,
           app.snc_quota_mgr,
           app.group_router,
+          app.usage_manager,
           app.shard_table,
           app.partition_manager,
           app.fetch_session_cache,