idea Datasource entity生成器 groovy文件(java，ssm springboot等适用)，xml生成器

最新推荐文章于 2024-04-04 18:00:19 发布
最新推荐文章于 2024-04-04 18:00:19 发布
阅读量259 收藏
点赞数 1
文章标签： java
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_42207808/article/details/113881917
版权 
import com.intellij.database.model.DasTable
import com.intellij.database.model.ObjectKind
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
import com.pty.bip.budPlan.BipImpPlan
import com.pty.bip.utils.MapperGenerator

import java.io.*
import java.text.SimpleDateFormat

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 */
packageName = ""
typeMapping = [
        (~/(?i)tinyint|smallint|mediumint/)      : "Integer",
        (~/(?i)int/)                             : "Long",
        (~/(?i)bool|bit/)                        : "Boolean",
        (~/(?i)float|double|decimal|real/)       : "Double",
        (~/(?i)datetime|timestamp|date|time/)    : "Timestamp",
        (~/(?i)blob|binary|bfile|clob|raw|image/): "InputStream",
        (~/(?i)/)                                : "String"
]


FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
    SELECTION.filter { it instanceof DasTable && it.getKind() == ObjectKind.TABLE }.each { generate(it, dir) }
}

def generate(table, dir) {
    //def className = javaClassName(table.getName(), true)
    def className = javaName(table.getName(), true)
    def fields = calcFields(table)
    packageName = getPackageName(dir)
    PrintWriter printWriter = new PrintWriter(new OutputStreamWriter(new FileOutputStream(new File(dir, className + ".java")), "UTF-8"))
    printWriter.withPrintWriter {out -> generate(out, className, fields,table)}

//    new File(dir, className + ".java").withPrintWriter { out -> generate(out, className, fields,table) }
}

// 获取包所在文件夹路径
def getPackageName(dir) {
    return dir.toString().replaceAll("\\\\", ".").replaceAll("/", ".").replaceAll("^.*src(\\.main\\.java\\.)?", "") + ";"
}

def generate(out, className, fields,table) {
    def tableName = table.getName()
    out.println "package $packageName"
    out.println ""
    out.println "import javax.persistence.*;"
    out.println "import java.io.Serializable;"
    out.println ""
    out.println "import com.pty.bip.utils.MapperGenerator;"
    out.println "import lombok.Data;"
    Set types = new HashSet()

    fields.each() {
        types.add(it.type)
    }

    if (types.contains("Date")) {
        out.println "import java.util.Date;"
    }

    if (types.contains("InputStream")) {
        out.println "import java.io.InputStream;"
    }
    out.println ""
    out.println "/**\n" +
            " * @Description  \n" +
            " * @Author  jsy \n" +
            " * @Date "+ new SimpleDateFormat("yyyy-MM-dd").format(new Date()) + " \n" +
            " */"
    out.println "@Entity"
    out.println "@Table ( name =\""+table.getName() +"\" )"
    out.println "@Data"
    out.println "public class $className  implements Serializable {"
    out.println genSerialID()


    // 判断自增
    if ((tableName + "_id").equalsIgnoreCase(fields[0].colum) || "id".equalsIgnoreCase(fields[0].colum)) {
        out.println "\t@Id"
    }


    fields.each() {
        out.println ""
        // 输出注释
        if (isNotEmpty(it.commoent)) {
            out.println "\t/**"
            out.println "\t * ${it.commoent.toString()}"
            out.println "\t */"
        }

        if (it.annos != "") out.println "   ${it.annos.replace("[@Id]", "")}"

        // 输出成员变量
        out.println "\tprivate ${it.type} ${it.name};"
    }
    out.println "    public static void main(String... args) {\n" +
            "        System.out.println(MapperGenerator.genMapper($className.class));\n" +
            "    }"
    out.println ""
    out.println "}"
}

def calcFields(table) {
    DasUtil.getColumns(table).reduce([]) { fields, col ->
        def spec = Case.LOWER.apply(col.getDataType().getSpecification())

        def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
        def comm =[
                colName : col.getName(),
                name :  javaName(col.getName(), false),
                type : typeStr,
                commoent: col.getComment(),
                annos: "\t@Column(name = \""+col.getName()+"\" )"]
        if("id".equals(Case.LOWER.apply(col.getName())))
            comm.annos +=["@Id"]
        fields += [comm]
    }
}

// 处理类名（这里是因为我的表都是以t_命名的，所以需要处理去掉生成类名时的开头的T，
// 如果你不需要那么请查找用到了 javaClassName这个方法的地方修改为 javaName 即可）
def javaClassName(str, capitalize) {
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    // 去除开头的T  http://developer.51cto.com/art/200906/129168.htm
    s = s[1..s.size() - 1]
    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def javaName(str, capitalize) {
//    def s = str.split(/(?<=[^\p{IsLetter}])/).collect { Case.LOWER.apply(it).capitalize() }
//            .join("").replaceAll(/[^\p{javaJavaIdentifierPart}]/, "_")
//    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
    def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
            .collect { Case.LOWER.apply(it).capitalize() }
            .join("")
            .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
    capitalize || s.length() == 1? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def isNotEmpty(content) {
    return content != null && content.toString().trim().length() > 0
}

static String changeStyle(String str, boolean toCamel){
    if(!str || str.size() <= 1)
        return str

    if(toCamel){
        String r = str.toLowerCase().split('_').collect{cc -> Case.LOWER.apply(cc).capitalize()}.join('')
        return r[0].toLowerCase() + r[1..-1]
    }else{
        str = str[0].toLowerCase() + str[1..-1]
        return str.collect{cc -> ((char)cc).isUpperCase() ? '_' + cc.toLowerCase() : cc}.join('')
    }
}

static String genSerialID()
{
    return "\tprivate static final long serialVersionUID =  "+Math.abs(new Random().nextLong())+"L;"
}

xml生成器main入口
 

    public static void main(String... args) {
        System.out.println(MapperGenerator.genMapper(Bean.class));
    }
MapperGenerator 类



import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import javax.persistence.Column;
import javax.persistence.Id;
import javax.persistence.Table;

public class MapperGenerator {
    private static final String lineSeparator = System.lineSeparator();
    private static String INSERT_SELECTIVE_COLUMN_SQL = "insertSelectiveColumn";
    private static String INSERT_SELECTIVE_VALUE_SQL = "insertSelectiveValue";
    private static String UPDATE_SELECTIVE_COLUMN_SQL = "updateSelectiveColumn";
    private static String UPDATE_SELECTIVE_VALUE_SQL = "updateSelectiveValue";

    public MapperGenerator() {
    }

    public static <T> String genMapper(Class<T> c) {
        return genMapper(c, false);
    }

    public static <T> String genMapper(Class<T> c, boolean isGenPrimaryKey) {
        StringBuilder builder = new StringBuilder();
        builder.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>").append(lineSeparator);
        builder.append("<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">").append(lineSeparator);
        builder.append("<mapper namespace=\"\">").append(lineSeparator);
        builder.append(genAllColumn(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genAllColumnAlias(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genAllColumnValue(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genAllColumnSet(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genInsertSelectiveColumn(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genInsertSelectiveValue(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genUpdateSelectiveColumn(c)).append(lineSeparator).append(lineSeparator);
        builder.append(genUpdateSelectiveValue(c)).append(lineSeparator).append(lineSeparator);
        if (isGenPrimaryKey) {
            builder.append(genPrimaryKeyColumnCond(c)).append(lineSeparator).append(lineSeparator);
        }

        builder.append(genAllColumnCond(c)).append(lineSeparator).append(lineSeparator);
        builder.append(insert(c)).append(lineSeparator).append(lineSeparator);
        builder.append(insertSelective(c)).append(lineSeparator).append(lineSeparator);
        builder.append(delById(c)).append(lineSeparator).append(lineSeparator);
        builder.append(delByIds(c)).append(lineSeparator).append(lineSeparator);
        builder.append(del(c)).append(lineSeparator).append(lineSeparator);
        if (isGenPrimaryKey) {
            builder.append(delByPrimaryKey(c)).append(lineSeparator).append(lineSeparator);
        }

        builder.append(updateById(c)).append(lineSeparator).append(lineSeparator);
        if (isGenPrimaryKey) {
            builder.append(updateByPrimaryKey(c)).append(lineSeparator).append(lineSeparator);
        }

        builder.append(selectById(c)).append(lineSeparator).append(lineSeparator);
        builder.append(selectByIds(c)).append(lineSeparator).append(lineSeparator);
        builder.append(select(c)).append(lineSeparator).append(lineSeparator);
        builder.append(count(c)).append(lineSeparator);
        builder.append("</mapper>");
        return builder.toString();
    }

    public static <T> String genAllColumn(Class<T> c) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"allColumn\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            if (i % 4 == 1) {
                fb.append("    ");
            }

            if (i == fields.size()) {
                fb.append(String.format("%-50s", ((Column)f.getAnnotation(Column.class)).name()));
            } else {
                fb.append(String.format("%-50s", ((Column)f.getAnnotation(Column.class)).name() + ","));
            }

            if (i % 4 == 0) {
                fb.append(lineSeparator);
            }
        }

        builder.append(fb.toString()).append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String genAllColumnAlias(Class<T> c) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"allColumnAlias\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            if (i % 4 == 1) {
                fb.append("    ");
            }

            if (i == fields.size()) {
                fb.append(String.format("%-50s", ((Column)f.getAnnotation(Column.class)).name() + " as " + f.getName()));
            } else {
                fb.append(String.format("%-50s", ((Column)f.getAnnotation(Column.class)).name() + " as " + f.getName() + ","));
            }

            if (i % 4 == 0) {
                fb.append(lineSeparator);
            }
        }

        builder.append(fb.toString()).append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String genAllColumnValue(Class<T> c) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"allColumnValue\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            if (i % 4 == 1) {
                fb.append("    ");
            }

            String param = String.format("#{%s,jdbcType=%s}", f.getName(), getJdbcType(f));
            if (i == fields.size()) {
                fb.append(String.format("%-40s", param));
            } else {
                fb.append(String.format("%-40s", param + ","));
            }

            if (i % 4 == 0) {
                fb.append(lineSeparator);
            }
        }

        builder.append(fb.toString()).append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String genAllColumnSet(Class<T> c) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"allColumnSet\">").append(lineSeparator);
        builder.append("    ");
        builder.append("<trim suffixOverrides=\",\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        Field idField = (Field)fields.stream().filter((fx) -> {
            return fx.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            if (f != idField) {
                fb.append("        ");
                if (f.getType() != String.class) {
                    fb.append(String.format("<if test=\"%s != null\">", f.getName())).append(lineSeparator);
                } else {
                    fb.append(String.format("<if test=\"%s != null and %s != ''\">", f.getName(), f.getName())).append(lineSeparator);
                }

                String param = String.format("            %s=#{%s,jdbcType=%s}", ((Column)f.getAnnotation(Column.class)).name(), f.getName(), getJdbcType(f));
                fb.append(String.format("%-50s", param + ",")).append(lineSeparator);
                fb.append("        </if>");
                fb.append(lineSeparator);
            }
        }

        builder.append(fb.toString());
        builder.append("    ");
        builder.append("</trim>");
        builder.append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String genInsertSelectiveColumn(Class<T> c) {
        return genColumnSelective(c, Boolean.FALSE, INSERT_SELECTIVE_COLUMN_SQL);
    }

    public static <T> String genInsertSelectiveValue(Class<T> c) {
        return genColumnSelectiveValue(c, Boolean.FALSE, INSERT_SELECTIVE_VALUE_SQL);
    }

    public static <T> String genUpdateSelectiveColumn(Class<T> c) {
        return genColumnSelective(c, Boolean.TRUE, UPDATE_SELECTIVE_COLUMN_SQL);
    }

    public static <T> String genUpdateSelectiveValue(Class<T> c) {
        return genColumnSelectiveValue(c, Boolean.TRUE, UPDATE_SELECTIVE_VALUE_SQL);
    }

    private static <T> String genColumnSelective(Class<T> c, Boolean isUpdate, String sqlId) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"" + sqlId + "\">").append(lineSeparator);
        builder.append("    ");
        builder.append("<trim suffixOverrides=\",\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        Field idField = (Field)fields.stream().filter((fx) -> {
            return fx.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            if (!isUpdate || f != idField) {
                fb.append("        ");
                if (f.getType() != String.class) {
                    fb.append(String.format("<if test=\"%s != null\">", f.getName())).append(lineSeparator);
                } else {
                    fb.append(String.format("<if test=\"%s != null and %s != ''\">", f.getName(), f.getName())).append(lineSeparator);
                }

                String param = String.format("            %s", ((Column)f.getAnnotation(Column.class)).name());
                fb.append(String.format("%-50s", param + ",")).append(lineSeparator);
                fb.append("        </if>");
                fb.append(lineSeparator);
            }
        }

        builder.append(fb.toString());
        builder.append("    ");
        builder.append("</trim>");
        builder.append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    private static <T> String genColumnSelectiveValue(Class<T> c, Boolean isUpdate, String sqlId) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"" + sqlId + "\">").append(lineSeparator);
        builder.append("    ");
        builder.append("<trim suffixOverrides=\",\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        Field idField = (Field)fields.stream().filter((fx) -> {
            return fx.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            if (!isUpdate || f != idField) {
                fb.append("        ");
                if (f.getType() != String.class) {
                    fb.append(String.format("<if test=\"%s != null\">", f.getName())).append(lineSeparator);
                } else {
                    fb.append(String.format("<if test=\"%s != null and %s != ''\">", f.getName(), f.getName())).append(lineSeparator);
                }

                String param = String.format("            #{%s,jdbcType=%s}", f.getName(), getJdbcType(f));
                fb.append(String.format("%-50s", param + ",")).append(lineSeparator);
                fb.append(String.format("        </if>")).append(lineSeparator);
            }
        }

        builder.append(fb.toString());
        builder.append("    ");
        builder.append("</trim>");
        builder.append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String genPrimaryKeyColumnCond(Class<T> c) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"primaryKeyColumnCond\">").append(lineSeparator);
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String genAllColumnCond(Class<T> c) {
        StringBuilder builder = new StringBuilder();
        builder.append("<sql id=\"allColumnCond\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        StringBuilder fb = new StringBuilder();
        Field f = null;

        for(int i = 1; i <= fields.size(); ++i) {
            f = (Field)fields.get(i - 1);
            fb.append("    ");
            if (f.getType() != String.class) {
                fb.append(String.format("<if test=\"%s != null\">", f.getName())).append(lineSeparator);
            } else {
                fb.append(String.format("<if test=\"%s != null and %s != ''\">", f.getName(), f.getName())).append(lineSeparator);
            }

            String param = String.format("        AND %s=#{%s,jdbcType=%s} ", ((Column)f.getAnnotation(Column.class)).name(), f.getName(), getJdbcType(f));
            fb.append(String.format("%-50s", param)).append(lineSeparator);
            fb.append(String.format("    </if>"));
            fb.append(lineSeparator);
        }

        builder.append(fb.toString());
        builder.append("</sql>");
        return builder.toString();
    }

    public static <T> String insert(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        StringBuilder builder = new StringBuilder();
        builder.append("<insert id=\"insert\" parameterType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append(String.format("    INSERT INTO %s (", tableName)).append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumn")).append(lineSeparator);
        builder.append("    ) VALUES (").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnValue")).append(lineSeparator);
        builder.append("    )").append(lineSeparator);
        builder.append("</insert>");
        return builder.toString();
    }

    public static <T> String insertSelective(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        StringBuilder builder = new StringBuilder();
        builder.append("<insert id=\"insertSelective\" parameterType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append(String.format("    INSERT INTO %s (", tableName)).append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", INSERT_SELECTIVE_COLUMN_SQL)).append(lineSeparator);
        builder.append("    ) VALUES (").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", INSERT_SELECTIVE_VALUE_SQL)).append(lineSeparator);
        builder.append("    )").append(lineSeparator);
        builder.append("</insert>");
        return builder.toString();
    }

    public static <T> String updateById(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<update id=\"updateById\" parameterType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append(String.format("    UPDATE %s SET ", tableName)).append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnSet")).append(lineSeparator);
        builder.append(String.format("    WHERE %s=%s", ((Column)idField.getAnnotation(Column.class)).name(), "#{" + idField.getName() + "}")).append(lineSeparator);
        builder.append("</update>");
        return builder.toString();
    }

    public static <T> String updateByPrimaryKey(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<update id=\"updateByPrimaryKey\" parameterType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append(String.format("    UPDATE %s SET ", tableName)).append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnSet")).append(lineSeparator);
        builder.append("    WHERE 1=1 ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "primaryKeyColumnCond")).append(lineSeparator);
        builder.append("</update>");
        return builder.toString();
    }

    public static <T> String selectById(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<select id=\"selectById\" parameterType=\"string\" resultType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append("    SELECT ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnAlias")).append(lineSeparator);
        builder.append("    FROM " + tableName + " ").append(lineSeparator);
        builder.append(String.format("    WHERE %s=%s", ((Column)idField.getAnnotation(Column.class)).name(), "#{value}")).append(lineSeparator);
        builder.append("</select>");
        return builder.toString();
    }

    public static <T> String selectByIds(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<select id=\"selectByIds\" parameterType=\"java.util.List\" resultType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append("    SELECT ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnAlias")).append(lineSeparator);
        builder.append("    FROM " + tableName + " ").append(lineSeparator);
        builder.append(String.format("    WHERE %s IN ", ((Column)idField.getAnnotation(Column.class)).name())).append(lineSeparator);
        builder.append("    <foreach collection=\"list\" index=\"index\" item=\"id\" open=\"(\" separator=\",\" close=\")\">").append(lineSeparator);
        builder.append("        #{id}").append(lineSeparator);
        builder.append("    </foreach>").append(lineSeparator);
        builder.append("</select>");
        return builder.toString();
    }

    public static <T> String select(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        StringBuilder builder = new StringBuilder();
        builder.append("<select id=\"select\" parameterType=\"" + c.getName() + "\" resultType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append("    SELECT ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnAlias")).append(lineSeparator);
        builder.append("    FROM " + tableName + " ").append(lineSeparator);
        builder.append("    WHERE 1=1 ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnCond")).append(lineSeparator);
        builder.append("</select>");
        return builder.toString();
    }

    public static <T> String count(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        StringBuilder builder = new StringBuilder();
        builder.append("<select id=\"count\" parameterType=\"" + c.getName() + "\" resultType=\"java.lang.Integer\">").append(lineSeparator);
        builder.append("    SELECT COUNT(1) ").append(lineSeparator);
        builder.append("    FROM " + tableName + " ").append(lineSeparator);
        builder.append("    WHERE 1=1 ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnCond")).append(lineSeparator);
        builder.append("</select>");
        return builder.toString();
    }

    public static <T> String delById(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<delete id=\"delById\" parameterType=\"string\">").append(lineSeparator);
        builder.append("    DELETE FROM " + tableName).append(lineSeparator);
        builder.append("    WHERE ").append(lineSeparator);
        builder.append(String.format("        %s=#{value}", ((Column)idField.getAnnotation(Column.class)).name())).append(lineSeparator);
        builder.append("</delete>");
        return builder.toString();
    }

    public static <T> String delByIds(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<delete id=\"delByIds\" parameterType=\"java.util.List\">").append(lineSeparator);
        builder.append("    DELETE FROM " + tableName).append(lineSeparator);
        builder.append(String.format("    WHERE %s IN ", ((Column)idField.getAnnotation(Column.class)).name())).append(lineSeparator);
        builder.append("    <foreach collection=\"list\" index=\"index\" item=\"id\" open=\"(\" separator=\",\" close=\")\">").append(lineSeparator);
        builder.append("        #{id}").append(lineSeparator);
        builder.append("    </foreach>").append(lineSeparator);
        builder.append("</delete>");
        return builder.toString();
    }

    public static <T> String del(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<delete id=\"del\" parameterType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append("    DELETE FROM " + tableName).append(lineSeparator);
        builder.append("    WHERE 1=1 ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "allColumnCond")).append(lineSeparator);
        builder.append("</delete>");
        return builder.toString();
    }

    public static <T> String delByPrimaryKey(Class<T> c) {
        String tableName = ((Table)c.getAnnotation(Table.class)).name();
        Field idField = (Field)getAllColumnField(c).stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst().get();
        StringBuilder builder = new StringBuilder();
        builder.append("<delete id=\"delByPrimaryKey\" parameterType=\"" + c.getName() + "\">").append(lineSeparator);
        builder.append("    DELETE FROM " + tableName).append(lineSeparator);
        builder.append("    WHERE 1=1 ").append(lineSeparator);
        builder.append(String.format("        <include refid=\"%s\" />", "primaryKeyColumnCond")).append(lineSeparator);
        builder.append("</delete>");
        return builder.toString();
    }

    public static <T> String selectWithPage(Class<T> c) {
        return "";
    }

    public static <T> String genResultMap(Class<T> c) {
        StringBuilder b = new StringBuilder();
        b.append("<resultMap id=\"" + c.getSimpleName() + "Map\" type=\"" + c.getName() + "\">").append(lineSeparator);
        List<Field> fields = getAllColumnField(c);
        Optional<Field> idField = fields.stream().filter((f) -> {
            return f.getAnnotation(Id.class) != null;
        }).findFirst();
        if (idField.isPresent()) {
            b.append(String.format("    <id %-30s %-30s />", "column=\"" + ((Column)((Field)idField.get()).getAnnotation(Column.class)).name() + "\"", "property=\"" + ((Field)idField.get()).getName() + "\""));
            b.append(lineSeparator);
        }

        fields.stream().map((f) -> {
            String jdbcType = getJdbcType(f);
            return String.format("    <result %-30s %-30s %-20s />", "column=\"" + ((Column)f.getAnnotation(Column.class)).name() + "\"", "property=\"" + f.getName() + "\"", "jdbcType=\"" + jdbcType + "\"");
        }).forEach((s) -> {
            b.append(s);
            b.append(lineSeparator);
        });
        b.append("</resultMap>").append(lineSeparator);
        return b.toString();
    }

    private static List<Field> getAllColumnField(Class c) {
        List<Field> fields = (List) ClassUtils.getAllField(c).stream().filter((f) -> {
            return f.isAnnotationPresent(Column.class);
        }).collect(Collectors.toList());
        return fields;
    }

    private static String getJdbcType(Field f) {
        Class type = f.getType();
        if (type.equals(String.class)) {
            return "VARCHAR";
        } else if (type.equals(BigDecimal.class)) {
            return "DECIMAL";
        } else if (type.equals(Integer.class)) {
            return "INTEGER";
        } else if (type.equals(Date.class)) {
            return "DATE";
        } else if (type.equals(Long.class)) {
            return "BIGINT";
        } else {
            return !type.equals(Double.class) && !type.equals(Float.class) ? "VARCHAR" : "DOUBLE";
        }
    }
}

ClassUtils类

 



import java.beans.PropertyDescriptor;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import javax.persistence.Column;
import org.apache.commons.beanutils.PropertyUtils;

public class ClassUtils {
    public ClassUtils() {
    }

    public static Field getField(Class clazz, String fName) {
        try {
            return clazz.getDeclaredField(fName);
        } catch (NoSuchFieldException var3) {
            return clazz.getSuperclass() != Object.class ? getField(clazz.getSuperclass(), fName) : null;
        }
    }

    public static List<Field> getAllField(Class clazz) {
        List<Field> result = new ArrayList();
        PropertyDescriptor[] ps = PropertyUtils.getPropertyDescriptors(clazz);
        Field f = null;
        PropertyDescriptor[] var4 = ps;
        int var5 = ps.length;

        for(int var6 = 0; var6 < var5; ++var6) {
            PropertyDescriptor prop = var4[var6];
            f = getField(clazz, prop.getName());
            if (f != null) {
                result.add(f);
            }
        }

        return result;
    }

    public static String getProperty(Class clazz, String column) {
        Optional<Field> optional = getAllField(clazz).stream().filter((field) -> {
            Column c = (Column)field.getAnnotation(Column.class);
            if (c == null) {
                return false;
            } else {
                return c.name().equalsIgnoreCase(column);
            }
        }).findFirst();
        return optional.isPresent() ? ((Field)optional.get()).getName() : "";
    }

    /**
     * T implements Serializable 泛型必须序列化
     * @param srcList
     * @param <T>
     * @return
     */
    public static <T> List<T> deepCopy(List<T> srcList) {
        ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
        try {
            ObjectOutputStream out = new ObjectOutputStream(byteOut);
            out.writeObject(srcList);
            ByteArrayInputStream byteIn = new ByteArrayInputStream(byteOut.toByteArray());
            ObjectInputStream inStream = new ObjectInputStream(byteIn);
            List<T> destList = (List<T>) inStream.readObject();
            return destList;
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
        return null;
    }
}

 

向着开发进攻
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
详解Java使用sqlite 数据库如何生成db文件
08-30
Java 使用 SQLite 数据库生成 DB 文件 Java 是一种流行的编程语言，广泛应用于 Android 开发、Web 开发、企业软件开发等领域。 SQLite 是一种轻量级的关系数据库管理系统，广泛应用于移动设备、嵌入式系统等领域。...
jpa代码自动生成工具-springboot
07-06
在给定的"codetool"压缩包中，很可能包含了一个这样的工具，它可以帮助我们自动生成domain（实体类）、dao（数据访问对象）、service（业务逻辑层）以及controller（控制器）等代码。这些代码通常遵循MVC（Model-...
在Idea中连接数据库并生成实体类
热门推荐
become a programmer
09-10 6万+
1.连接数据库
IDEA 配置datasource，提升编码效率，让你在 Mapper.xml 中编写sql可以飞起来~
小月施主的博客
10-16 1万+
IDEA 2018 创建springboot工程后，如果你打开一个.sql文件，或者一个mybatis的mapper.xml文件，会提示： No data source are configured to run this SQL and provide advanced code assistance. 不要觉得这个很烦，我个人觉得这个工具提供了这么强大的能提高你编码效率的功能给你，不用...
IDEA - 配置datasource，提升编码效率
sunrise52java的专栏
02-14 3960
新建了一个sql文件，总是提示我配置datasource。 配置后对sql文件，mapper.xml文件的书写都会有很大的帮助。 配置步骤： 1、根据以下步骤操作： 2、在打开的Data Sources and Drivers窗口里填写下面红框框里的关于mysql的信息，点击OK，就配置成功了。 ...
IDEA连接数据库
Jerry&Crj的博客
01-07 85
IDEA连接数据库 在IDEA开发环境中连接数据库 １、软件开始界面点击 ２、点击对象想要连接的数据库 ３、第一次运行的话需要下载驱动文件 ４、如图
idea datasource 全局配置
weixin_30546189的博客
07-16 3777
具体配置如下：打开Database界面，点击Data Source Properties, 选中要设置为全局的数据源配置，选择Make Global。这样就把其设置为全局数据源。再点一下，设置为项目数据源。具体操作见下图。 blog.csdn.net/wo541075754… 转载于:https://juejin.im/post/5d2d7d44f265da1bc5529698...
springboot+mybatis-plus 代码生成器 CRUD 分页插件
12-21
首先，你需要在你的`pom.xml`文件中添加以下依赖，确保引入了MyBatis-Plus的核心启动器和代码生成器： ```xml <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter <version>3.1.1 ...
Mybatis代码生成器
11-27
每个生成器都包含一个或多个模块（context），每个模块可以生成特定类型的文件，如Mapper接口、Mapper XML文件、实体类（Entity）以及DAO接口等。你可以根据实际需求调整这些配置，例如是否生成Example类（用于复杂...
基于SSM的电器商城系统源码.zip
10-03
如果项目中使用SpringBoot，那么它可能会自动配置许多Spring组件，如DataSource、JPA等，并提供内置的Tomcat服务器，简化项目的部署和运行。 **系统功能模块分析：** - **用户模块**：包括用户注册、登录、个人信息...
IDEA DataSource远程连接Mysql及其完整使用流程
tanxinji的博客
03-24 5579
一、配置连接  先后点击IDEA右侧的DataBase -> + > Data Source -> MySql  输入完连接名，用户名，密码。可以选择默认的URL,点击Test Connection  出现如下提示后，点击Set time zone  他会自动跳转设置时区  在次点击Test Connection，出现如下界面，点击Apple ,再点击OK  点击OK后，会自动跳转到如下界面: 二、使用连接 1.执行sql语句  在console窗口可以输入sql语句，选中需要
EntityDataSource不易察觉的错误
weixin_33904756的博客
06-24 144
今天在使用EntityDataSource显示数据时，返回的数据总是空的，确认过数据库内数据后，尝试去掉Where及参数，移除各种事件均告无效。 后来新建了一个EntityDataSource，简单配置一下数据源，测试通过。 挠墙半晌后看到Include属性中包含了一个已经在数据库及EDM中删除的导航属性，去掉该属性后测试即通过。 虽然是个人的小失误，但可恨的是对于这个简单的错误EntityDat...
IDEA java dataSource连接数据库 以及Mysql root密码修改
fllow_wind的博客
03-04 3612
IDEA java dataSource连接数据库 弄了一整天，终于成功了，效果如下 1.先得下一个Mysql 下载和安装步骤参考：https://blog.csdn.net/weixin_42869365/article/details/83472466 注意参考一下评论区 2.Mysql密码问题，改密码，没问题跳到下一步 我的mysql root账号密码初始化过(很复杂，最后找到居然是thH_7a%4WWrB)，当时不记得了，然后找了各种解决方案都行不通。 最后找到了一位博主的解决方法：https:/
idea中连接数据库并展示数据库表的信息
qq_45790384的博客
02-04 1万+
今天有小伙伴在idea中想要显示数据库表的数据信息时，连接失败或不知道如何操作导致手足无措，所以我今天就写一篇关于在idea中显示数据库表之间的信息。 首先我们在idea右侧点击DataSource 点击之后显示如图，点击加号按钮 选择Data Source然后右移到右边列表的MYSQL上点击 这里就需要填写一些信息。User是数据库的用户名，Password是数据库的登陆密码，Database是你要查的数据库名称 以上信息填好后记得点击Test Connection按钮测试，如没问题就会显示下面的
IntelliJ IDEA for Mac如何配置数据源(Data Source)和用户驱动(User Driver)及数据库控制台_数据源配置（Console）详解
liaowenxiong的博客
12-20 6908
文章目录直接添加数据源（Data Source）添加驱动和数据源（Driver and Data Source）用户驱动 直接添加数据源（Data Source） 直接添加数据源，IDE 会默认指定数据库驱动，如下图所示： 添加驱动和数据源（Driver and Data Source） 这样方式添加数据源后，必须添加用户驱动，这样数据源才能和用户驱动关联。 数据源配置完成后，就要配置用户驱动（User Driver）。 注意：添加驱动文件时，随便选择哪种都可以。 用户驱动 注意：创建的数据源
小型Springboot项目生成器（使用IDEA groovy模板）
qq_42059717的博客
12-20 574
1.背景介绍 因为最近因为一些个人的原因需要使用到代码生成器，但是本地又不想配置各种复杂的东西，参考了一下IDEA中的database插件里边的groovy，自己写了一个生成后端项目的cv模板哈哈 1. 介绍 首先，需要点击IDEA右侧的dataBase ,连接上数据库 连接上数据库之后，右键一张表，选中模板位置 将下列内容复制到里边 import com.intellij.database.model.DasTable import com.intellij.database.util.Cas
idea中MySQL数据库的配置
最新发布
m0_64021225的博客
04-04 2744
idea中数据库的配置
java 转 groovy_Groovy转换 – 来自Java
weixin_42310231的博客
02-27 506
我有一个java类的以下小部分：public class GBitMaskTest {@Testpublic void checkFirstBitTest() {BitMask bm = new BitMask(0x8000000000000000L);assertEquals(true, bm.isBitSet(63));}...}哪个应该可以在Groovy中编译,因为它是有效的Java代码.但...
IDEA连接数据库生成实体类方法与定制Groovy生成脚本
ayyu1314的博客
08-25 525
导语: 如今已2019年, 有许许多多的插件或者编辑器都支持根据数据表自动生成数据实体类了, 比如IDEA, 各种MyBatis的代码生成工具, 等等. 本篇介绍一下如何使用IDEA的groovy脚本文件生成带JPA注解的实体类 参考链接: Intellij IDEA 通过数据库表生成带注解的实体类详细步骤-悲凉的秋风 一, 使用IDEA连接数据库 需注意: 正式版IDEA才...
java读取sqlserver图片image数据生成图片文件，用springboot
05-30
在 pom.xml 文件中添加如下依赖： ```xml <groupId>org.springframework.boot <artifactId>spring-boot-starter-jdbc <groupId>com.microsoft.sqlserver</groupId> <artifactId>mssql-jdbc <version>9.4.0....

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值