java 将_分隔的字符串转为驼峰格式

目的

        将数据库中全大写_分隔的存储的字段名对应到程序中的驼峰命名。

实现

        

package org.springblade.test;

import java.util.StringTokenizer;

public class Convert2CamelCaseOrLowerCase {

    /**
     * 根据输入字符串的格式将其转换为驼峰命名法或全小写
     *
     * @param input 输入字符串
     * @return 转换后的字符串
     */
    public static String convertToCamelCaseOrLowerCase(String input) {
        if (input == null || input.isEmpty()) {
            return input;
        }
        boolean containsUnderscore = input.contains("_");
        if (containsUnderscore) {
            // 包含下划线，转换为驼峰式命名
            return underscoreToCamelCase(input);
        } else {
            // 不包含下划线，转换为全小写
            return input.toLowerCase();
        }
    }

    /**
     * 将下划线分隔的字符串转换为驼峰式命名法
     *
     * @param input 下划线分隔的字符串
     * @return 转换后的驼峰式命名字符串
     */
    private static String underscoreToCamelCase(String input) {
        StringBuilder result = new StringBuilder(input.length());
        StringTokenizer st = new StringTokenizer(input, "_");
        while (st.hasMoreTokens()) {
            String token = st.nextToken();
            if (result.length() == 0) {
                result.append(token.toLowerCase());
            } else {
                result.append(convertToCamelCase(token));
                // 如果仅需要转换首字母,用此方法
                // result.append(capitalizeFirstLetter(token));
            }
        }
        return result.toString();
    }

    /**
     * 将字符串的第一个字符转换为大写
     *
     * @param str 需要首字母大写的字符串
     * @return 首字母大写后的字符串
     */
    private String capitalizeFirstLetter(String str) {
        if (str == null || str.isEmpty()) {
            return str;
        }
        return Character.toUpperCase(str.charAt(0)) + str.substring(1);
    }

    /**
     * 将字符串的第一个字符转换为大写，其余部分转为小写
     *
     * @param input 需要首字母大写、其余小写的字符串
     * @return 转换后的字符串
     */
    public static String convertToCamelCase(String input) {
        if (input == null || input.isEmpty()) {
            return input;
        }
        char[] chars = input.toCharArray();
        chars[0] = Character.toUpperCase(chars[0]);
        for (int i = 1; i < chars.length; i++) {
            chars[i] = Character.toLowerCase(chars[i]);
        }

        return new String(chars);
    }

    @org.junit.jupiter.api.Test
    public void underscoreToCamelCaseTest() {
        String input = "XXX_XXX_XXX";
        System.out.println(convertToCamelCaseOrLowerCase(input));  
        // 输出：xxxXxxXxx
        String input2 = "XXX";
        System.out.println(convertToCamelCaseOrLowerCase(input2));  
        // 输出：xxx
    }

}