From 09f1c9fd70a230826bd74c3af5e3b1cbb4bee9a4 Mon Sep 17 00:00:00 2001
From: jaysunxiao <jaysunxiao@gmail.com>
Date: Sat, 5 Jul 2025 10:33:46 +0800
Subject: [PATCH] perf[set]: add unboxed contains method to HashSetInt

---
 .../main/java/com/zfoo/protocol/collection/HashSetInt.java    | 4 ++++
 .../main/java/com/zfoo/protocol/collection/HashSetShort.java  | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/protocol/src/main/java/com/zfoo/protocol/collection/HashSetInt.java b/protocol/src/main/java/com/zfoo/protocol/collection/HashSetInt.java
index 0e50ecea..573763ba 100644
--- a/protocol/src/main/java/com/zfoo/protocol/collection/HashSetInt.java
+++ b/protocol/src/main/java/com/zfoo/protocol/collection/HashSetInt.java
@@ -53,6 +53,10 @@ public class HashSetInt extends AbstractSet<Integer> {
         return map.containsKey(o);
     }
 
+    public boolean contains(int key) {
+        return map.containsKey(key);
+    }
+
     @Override
     public boolean add(Integer e) {
         return map.put(e, Boolean.TRUE) == null;
diff --git a/protocol/src/main/java/com/zfoo/protocol/collection/HashSetShort.java b/protocol/src/main/java/com/zfoo/protocol/collection/HashSetShort.java
index afdcbbe4..fb3af7bd 100644
--- a/protocol/src/main/java/com/zfoo/protocol/collection/HashSetShort.java
+++ b/protocol/src/main/java/com/zfoo/protocol/collection/HashSetShort.java
@@ -53,6 +53,10 @@ public class HashSetShort extends AbstractSet<Short> {
         return map.containsKey(o);
     }
 
+    public boolean contains(short key) {
+        return map.containsKey(key);
+    }
+
     @Override
     public boolean add(Short e) {
         return map.put(e, Boolean.TRUE) == null;