From b156aab26195dcd73a253c6cc8c824fe3598f0b2 Mon Sep 17 00:00:00 2001
From: wodiwudi <40229449+wodiwudi@users.noreply.github.com>
Date: Wed, 4 Sep 2024 17:16:33 +0800
Subject: [PATCH] Refactor codegen unit test (#884)

* update

* refactor codegen

* change pom

* change infra pom

* change template

* refactor: fix template
---
 .../run/mone/ai/codegen/util/StrUtil.java     | 43 +++++++++++++++++++
 .../mone/ai/codegen/util/TemplateUtils.java   |  2 +-
 .../src/main/resources/tlp/testSpring.java    |  4 +-
 .../xiaomi/youpin/codegen/test/StrTest.java   | 43 +++++++++++++++++++
 4 files changed, 89 insertions(+), 3 deletions(-)
 create mode 100644 jcommon/codegen/src/main/java/run/mone/ai/codegen/util/StrUtil.java
 create mode 100644 jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/StrTest.java

diff --git a/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/StrUtil.java b/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/StrUtil.java
new file mode 100644
index 000000000..df4fbfde3
--- /dev/null
+++ b/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/StrUtil.java
@@ -0,0 +1,43 @@
+package run.mone.ai.codegen.util;
+
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * @author zhangxiaowei6
+ * @Date 2024/9/4 16:31
+ */
+
+@Slf4j
+public class StrUtil {
+
+    public static String toCamelCase(String input) {
+        if (input == null || input.isEmpty()) {
+            return input;
+        }
+
+        StringBuilder result = new StringBuilder();
+        boolean capitalizeNext = false;
+
+        for (int i = 0; i < input.length(); i++) {
+            char currentChar = input.charAt(i);
+
+            if (currentChar == ' ') {
+                capitalizeNext = true;
+            } else {
+                if (capitalizeNext) {
+                    result.append(Character.toUpperCase(currentChar));
+                    capitalizeNext = false;
+                } else {
+                    result.append(Character.toLowerCase(currentChar));
+                }
+            }
+        }
+
+        // 确保第一个字符是小写的
+        if (result.length() > 0) {
+            result.setCharAt(0, Character.toLowerCase(result.charAt(0)));
+        }
+
+        return result.toString();
+    }
+}
diff --git a/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/TemplateUtils.java b/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/TemplateUtils.java
index 1c3aaac36..80d26fcd9 100644
--- a/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/TemplateUtils.java
+++ b/jcommon/codegen/src/main/java/run/mone/ai/codegen/util/TemplateUtils.java
@@ -38,7 +38,7 @@ public static String renderTemplate(String template, Map<String, ? extends Objec
             }
             StringTemplateResourceLoader resourceLoader = new StringTemplateResourceLoader();
             Configuration cfg = Configuration.defaultConfiguration();
-            GroupTemplate gt = new GroupTemplate(resourceLoader, cfg);
+            GroupTemplate gt = new GroupTemplate(resourceLoader, cfg, cfg.getClass().getClassLoader());
             functionList.forEach(it -> gt.registerFunction(it.getKey(), it.getValue()));
             Template t = gt.getTemplate(template);
             m.forEach((k, v) -> t.binding(k, v));
diff --git a/jcommon/codegen/src/main/resources/tlp/testSpring.java b/jcommon/codegen/src/main/resources/tlp/testSpring.java
index 3e639ed7d..e7e36c682 100644
--- a/jcommon/codegen/src/main/resources/tlp/testSpring.java
+++ b/jcommon/codegen/src/main/resources/tlp/testSpring.java
@@ -22,8 +22,8 @@
 public class ${testName} {
 
 
-@Resource
-private ${serviceName} ${strutil.toLowerCase(serviceName)}Service;
+    @Resource
+    private ${serviceName} ${className}Service;
 
 
 }
\ No newline at end of file
diff --git a/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/StrTest.java b/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/StrTest.java
new file mode 100644
index 000000000..8c7710708
--- /dev/null
+++ b/jcommon/codegen/src/test/java/com/xiaomi/youpin/codegen/test/StrTest.java
@@ -0,0 +1,43 @@
+package com.xiaomi.youpin.codegen.test;
+
+import org.junit.Test;
+import run.mone.ai.codegen.util.StrUtil;
+
+import static org.junit.Assert.assertEquals;
+
+
+public class StrTest {
+
+
+
+
+    @Test
+    public void testToCamelCase() {
+        String input = "hello world";
+        String expected = "helloWorld";
+        String actual = StrUtil.toCamelCase(input);
+        assertEquals(expected, actual);
+
+        input = "java programming language";
+        expected = "javaProgrammingLanguage";
+        actual = StrUtil.toCamelCase(input);
+        assertEquals(expected, actual);
+
+        input = "  leading and trailing spaces  ";
+        expected = "leadingAndTrailingSpaces";
+        actual = StrUtil.toCamelCase(input);
+        assertEquals(expected, actual);
+
+        input = "";
+        expected = "";
+        actual = StrUtil.toCamelCase(input);
+        assertEquals(expected, actual);
+
+        input = null;
+        expected = null;
+        actual = StrUtil.toCamelCase(input);
+        assertEquals(expected, actual);
+    }
+
+
+}