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;