Replaced consumer with Handler<T>

Author: ryntric

README.md

@@ -68,17 +68,13 @@ Internal message queue:
 ## 🛠️ Usage Example
 
 ```java
-AffinityDispatcher<String> dispatcher = new AffinityDispatcher<>(
-    "testDispatcher",
-     value -> {},
-    DefaultHashCodeProvider.INSTANCE,
-    Config.builder().build()
-);
-
-dispatcher.start(); // ✅ Start workers
-
-dispatcher.dispatch("key1", "Hello World"); // 📤 Dispatch messages
-dispatcher.dispatch(42, "Another message");
-
-dispatcher.shutdown(); // 🛑 Stop workers
+Handler<Object> handler = (workerName, value) -> {
+    System.out.println("Handled by " + workerName + ": " + value);
+};
+AffinityDispatcher<Object> dispatcher = new AffinityDispatcher<>("test", handler, DefaultHashCodeProvider.INSTANCE, Config.builder().build());
+dispatcher.start();
+
+for (int i = 0; i < 10_000_000; i++) {
+    dispatcher.dispatch(UUID.randomUUID().toString(), i);
+}
 ```

src/main/java/io/github/ryntric/AffinityDispatcher.java

@@ -2,8 +2,8 @@
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.UUID;
 import java.util.concurrent.atomic.AtomicInteger;
-import java.util.function.Consumer;
 
 /**
  * AffinityDispatcher is a high-performance dispatcher that routes messages between workers based on a key's hash code.
@@ -35,7 +35,7 @@ public final class AffinityDispatcher<T> {
      * @param hashCodeProvider provider to compute hash codes from keys
      * @param config           dispatcher configuration (worker count, buffer size, etc.)
      */
-    public AffinityDispatcher(String name, Consumer<T> handler, HashCodeProvider hashCodeProvider, Config config) {
+    public AffinityDispatcher(String name, Handler<T> handler, HashCodeProvider hashCodeProvider, Config config) {
         this.name = name;
         this.workerCount = config.getWorkerCount();
         this.nodesPerWorker = config.getRoutingNodePerWorker();
@@ -47,7 +47,7 @@ public AffinityDispatcher(String name, Consumer<T> handler, HashCodeProvider has
         this.init(name, handler, config);
     }
 
-    private void init(String name, Consumer<T> handler, Config config) {
+    private void init(String name, Handler<T> handler, Config config) {
         WorkerFactory<T> workerFactory = new WorkerFactory<>(name, config.getWorkerThreadPriority(), config.getBatchSize(), handler);
 
         for (int i = 0; i < workerCount; i++) {
@@ -212,4 +212,17 @@ public Map<String, Long> getChannelSizes() {
         return result;
     }
 
+    public static void main(String[] args) {
+        Handler<Object> handler = (workerName, value) -> {
+            System.out.println("Handled by " + workerName + ": " + value);
+        };
+        AffinityDispatcher<Object> dispatcher = new AffinityDispatcher<>("test", handler, DefaultHashCodeProvider.INSTANCE, Config.builder().build());
+
+        dispatcher.start();
+
+        for (int i = 0; i < 10_000_000; i++) {
+            dispatcher.dispatch(UUID.randomUUID().toString(), i);
+        }
+    }
+
 }

src/main/java/io/github/ryntric/Handler.java

@@ -0,0 +1,5 @@
+package io.github.ryntric;
+
+public interface Handler<T> {
+    void handle(String workerName, T value);
+}

src/main/java/io/github/ryntric/Utils.java

@@ -1,6 +1,6 @@
 package io.github.ryntric;
 
-final class Utils {
+public final class Utils {
     private static final Runtime RUNTIME = Runtime.getRuntime();
     private static final Integer AVAILABLE_PROCESSORS = RUNTIME.availableProcessors();
 

src/main/java/io/github/ryntric/WorkerFactory.java

@@ -1,19 +1,17 @@
 package io.github.ryntric;
 
-import java.util.function.Consumer;
-
 final class WorkerFactory<T> {
     private static final String NAME_TEMPLATE = "%s-worker-th-%d";
 
     private final String name;
     private final int priority;
     private final ThreadGroup group;
     private final int batchsize;
-    private final Consumer<T> handler;
+    private final Handler<T> handler;
 
     private int nextId = 0;
 
-    public WorkerFactory(String name, int priority, int batchsize, Consumer<T> handler) {
+    public WorkerFactory(String name, int priority, int batchsize, Handler<T> handler) {
         this.name = name;
         this.priority = priority;
         this.group = new ThreadGroup(name);

src/main/java/io/github/ryntric/WorkerThread.java

@@ -4,16 +4,16 @@
 import java.util.function.Consumer;
 
 final class WorkerThread<T> extends Thread {
-    private final Consumer<T> handler;
     private final Channel<T> channel;
     private final int batchsize;
     private final AtomicBoolean isRunning;
+    private final Consumer<T> handler;
 
-    public WorkerThread(String name, ThreadGroup group, int batchsize, Channel<T> channel, Consumer<T> handler) {
+    public WorkerThread(String name, ThreadGroup group, int batchsize, Channel<T> channel, Handler<T> delegated) {
         super(group, name);
         this.batchsize = batchsize;
         this.channel = channel;
-        this.handler = handler;
+        this.handler = (value) -> delegated.handle(name, value);
         this.isRunning = new AtomicBoolean(false);
     }