feat[es]: es support

zfoo-project · Apr 20, 2024 · 7e912f9 · 7e912f9
1 parent 4c5fb4e
commit 7e912f9
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolNote.java b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolNote.java
@@ -93,6 +93,7 @@ private static String formatNote(CodeLanguage language, String fieldNote) {
             case Cpp:
             case Go:
             case JavaScript:
+            case ES:
             case TypeScript:
             case CSharp:
             case Protobuf:

diff --git a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolPath.java b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolPath.java
@@ -59,6 +59,8 @@ public static String protocolAbsolutePath(short protocolId, CodeLanguage languag
                 break;
             case JavaScript:
                 break;
+            case ES:
+                break;
             case TypeScript:
                 break;
             case CSharp:

diff --git a/protocol/src/main/java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java b/protocol/src/main/java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java
@@ -98,7 +98,7 @@ public static void createProtocolManager(List<IProtocolRegistration> protocolLis
             var protocolId = protocol.protocolId();
             var protocolName = protocol.protocolConstructor().getDeclaringClass().getSimpleName();
             var path = GenerateProtocolPath.protocolAbsolutePath(protocol.protocolId(), CodeLanguage.ES);
-            importBuilder.append(StringUtils.format("import {} from './{}';", protocolName, path)).append(LS);
+            importBuilder.append(StringUtils.format("import {} from './{}.mjs';", protocolName, path)).append(LS);
             initProtocolBuilder.append(StringUtils.format("protocols.set({}, {});", protocolId, protocolName)).append(LS);
         }
 

diff --git a/protocol/src/test/java/com/zfoo/protocol/generate/GenerateTesting.java b/protocol/src/test/java/com/zfoo/protocol/generate/GenerateTesting.java
@@ -38,6 +38,7 @@ public void generateAllProtocols() {
         generateLanguages.add(CodeLanguage.Cpp);
         generateLanguages.add(CodeLanguage.Go);
         generateLanguages.add(CodeLanguage.JavaScript);
+        generateLanguages.add(CodeLanguage.ES);
         generateLanguages.add(CodeLanguage.TypeScript);
         generateLanguages.add(CodeLanguage.Lua);
         generateLanguages.add(CodeLanguage.CSharp);