From 3a8d23bce4f3dcf681bd68165f18f7e4f334c5e2 Mon Sep 17 00:00:00 2001 From: ReaJason Date: Sun, 11 May 2025 13:07:31 +0800 Subject: [PATCH] refactor: simplify code --- .../javaweb/memshell/generator/command/CommandGenerator.java | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/generator/command/CommandGenerator.java b/generator/src/main/java/com/reajason/javaweb/memshell/generator/command/CommandGenerator.java index 82582d68..a6b205ad 100644 --- a/generator/src/main/java/com/reajason/javaweb/memshell/generator/command/CommandGenerator.java +++ b/generator/src/main/java/com/reajason/javaweb/memshell/generator/command/CommandGenerator.java @@ -41,6 +41,7 @@ public class CommandGenerator { DynamicType.Builder builder = new ByteBuddy() .redefine(commandConfig.getShellClass()) .name(commandConfig.getShellClassName()) + .field(named("paramName")).value(commandConfig.getParamName()) .visit(new TargetJreVersionVisitorWrapper(shellConfig.getTargetJreVersion())); if (shellConfig.isJakarta()) { @@ -59,9 +60,6 @@ public class CommandGenerator { put("paramName", commandConfig.getParamName()); }}) ); - } else if (!ShellType.WEBSOCKET.equals(shellType)) { - builder = builder.field(named("paramName")) - .value(commandConfig.getParamName()); } if (CommandConfig.Encryptor.DOUBLE_BASE64.equals(commandConfig.getEncryptor())) {