feat(zonerouter): add client type (#2037)

Signed-off-by: Robin Han <[email protected]>

core/src/main/java/kafka/automq/zonerouter/ClientIdKey.java

@@ -13,4 +13,5 @@
 
 public class ClientIdKey {
     public static final String AVAILABILITY_ZONE = "automq_az";
+    public static final String CLIENT_TYPE = "automq_type";
 }

core/src/main/java/kafka/automq/zonerouter/ClientIdMetadata.java

@@ -43,6 +43,14 @@ public String rack() {
         return list.get(0);
     }
 
+    public ClientType clientType() {
+        List<String> list = metadata.get(ClientIdKey.CLIENT_TYPE);
+        if (list == null || list.isEmpty()) {
+            return null;
+        }
+        return ClientType.parse(list.get(0));
+    }
+
     public String clientId() {
         return clientId;
     }

core/src/main/java/kafka/automq/zonerouter/ClientType.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2024, AutoMQ HK Limited.
+ *
+ * The use of this file is governed by the Business Source License,
+ * as detailed in the file "/LICENSE.S3Stream" included in this repository.
+ *
+ * As of the Change Date specified in that file, in accordance with
+ * the Business Source License, use of this software will be governed
+ * by the Apache License, Version 2.0
+ */
+
+package kafka.automq.zonerouter;
+
+public enum ClientType {
+    PRODUCER, CONSUMER;
+
+    public static ClientType parse(String str) {
+        switch (str) {
+            case "producer":
+                return PRODUCER;
+            case "consumer":
+                return CONSUMER;
+            default:
+                return null;
+        }
+    }
+
+}