tweaks

beemobot · Dec 17, 2023 · b7954a3 · b7954a3
1 parent 03323c1
commit b7954a3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/latte/src/main/java/gg/beemo/latte/broker/BrokerClient.kt b/latte/src/main/java/gg/beemo/latte/broker/BrokerClient.kt
@@ -12,7 +12,6 @@ private class TopicMetadata(
 
 private class KeyMetadata(
     val topic: TopicMetadata,
-    val key: String,
     val producers: MutableSet<ProducerSubclient<*>>,
     val consumers: MutableSet<ConsumerSubclient<*>>,
 )
@@ -76,7 +75,7 @@ abstract class BrokerClient(
         topic: String,
         key: String,
         options: BrokerClientOptions = BrokerClientOptions(),
-        noinline callback: suspend CoroutineScope.(BaseRpcRequestMessage<RequestT, ResponseT>) -> Pair<RpcStatus, ResponseT>,
+        noinline callback: suspend CoroutineScope.(BaseRpcRequestMessage<RequestT, ResponseT>) -> RpcResponse<ResponseT>,
     ): RpcClient<RequestT, ResponseT> {
         return RpcClient(
             this,
@@ -146,7 +145,7 @@ abstract class BrokerClient(
             TopicMetadata(topic, Collections.synchronizedMap(HashMap()))
         }
         val keyData = topicData.keys.computeIfAbsent(key) {
-            KeyMetadata(topicData, key, Collections.synchronizedSet(HashSet()), Collections.synchronizedSet(HashSet()))
+            KeyMetadata(topicData, Collections.synchronizedSet(HashSet()), Collections.synchronizedSet(HashSet()))
         }
         return keyData
     }

diff --git a/latte/src/main/java/gg/beemo/latte/broker/BrokerSubclients.kt b/latte/src/main/java/gg/beemo/latte/broker/BrokerSubclients.kt
@@ -169,6 +169,8 @@ class ConsumerSubclient<T>(
 
 }
 
+typealias RpcResponse<ResponseT> = Pair<RpcStatus, ResponseT>
+
 class RpcClient<RequestT, ResponseT>(
     client: BrokerClient,
     topic: String,
@@ -178,7 +180,7 @@ class RpcClient<RequestT, ResponseT>(
     requestIsNullable: Boolean,
     private val responseType: Class<ResponseT>,
     private val responseIsNullable: Boolean,
-    private val callback: suspend CoroutineScope.(BaseRpcRequestMessage<RequestT, ResponseT>) -> Pair<RpcStatus, ResponseT>,
+    private val callback: suspend CoroutineScope.(BaseRpcRequestMessage<RequestT, ResponseT>) -> RpcResponse<ResponseT>,
 ) : BaseSubclient(
     client.connection,
     client,

diff --git a/vanilla/src/main/java/gg/beemo/vanilla/KafkaRatelimitClient.kt b/vanilla/src/main/java/gg/beemo/vanilla/KafkaRatelimitClient.kt
@@ -3,6 +3,7 @@ package gg.beemo.vanilla
 import gg.beemo.latte.broker.BrokerClient
 import gg.beemo.latte.broker.BrokerConnection
 import gg.beemo.latte.broker.IgnoreRpcRequest
+import gg.beemo.latte.broker.RpcStatus
 import gg.beemo.latte.logging.Log
 import gg.beemo.latte.ratelimit.SharedRatelimitData
 import gg.beemo.latte.util.SuspendingRatelimit
@@ -46,6 +47,8 @@ class RatelimitClient(connection: BrokerConnection) : BrokerClient(connection) {
             provider.getClientRatelimit(clientId).requestQuota()
 
             log.debug("Granted {} quota request for service {}", type, service)
+
+            return@rpc RpcStatus.OK to Unit
         }
     }