From eaed62620b4352190815724de1bd26fae94e629b Mon Sep 17 00:00:00 2001
From: godotg <jaysunxiao@gmail.com>
Date: Tue, 16 Jan 2024 17:44:39 +0800
Subject: [PATCH] fix[enhance]: location null pointer error

---
 .../java/com/zfoo/protocol/registration/ProtocolAnalysis.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java b/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java
index cbb64e46..a8c84717 100644
--- a/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java
+++ b/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java
@@ -292,8 +292,8 @@ public class ProtocolAnalysis {
                     var protocolId = getProtocolIdAndCheckClass(protocolClass);
                     initProtocolClass(protocolId, protocolClass);
                     protocolModuleSet.add(protocolClass);
-                    classModuleDefinitionMap.put(clazz, moduleId);
-                    protocolXmlEnhanceMap.compute(clazz, (key, value) -> Boolean.TRUE.equals(value) || enhance);
+                    classModuleDefinitionMap.put(protocolClass, moduleId);
+                    protocolXmlEnhanceMap.compute(protocolClass, (key, value) -> Boolean.TRUE.equals(value) || enhance);
                 }
             }
         }