java通过反射注解生成对应的sql语句

去年上课老师要求写一个java项目作为作业 今天闲来无事收集起来 保留一下以前写的代码而已

首先直接上代码吧

public class generateSQL {

    public static void main(String[] args) {
        Teacher teacher = new Teacher();
        generateSQL generateSQL= new generateSQL();
        teacher.setId("1");
        teacher.setSalary("111");
        teacher.setProfessional("111");
        teacher.setBirth("111");
        teacher.setSchool("111");
        teacher.setName("111");
        String str1 = generateSQL.selectSQL(teacher.getClass());
        System.out.println("输出的SELECT的sql：" + str1);
        String str2 = generateSQL.deleteSQL(teacher.getClass(), teacher);
        System.out.println("输出的DELETE的sq2：" + str2);
        String str3 = generateSQL.insertSQL(teacher.getClass(), teacher);
        System.out.println("输出的INSERT的sq3：" + str3);
        String str4 = generateSQL.updateSQL(teacher.getClass(), teacher);
        System.out.println("输出的UPDATE的sq4：" + str4);

    }


    //生成select的sql语句，
    public  String selectSQL(Class<?> clazz) {
        StringBuilder str = new StringBuilder();
//        //获取一个Table的名字
//        boolean exists = clazz.isAnnotationPresent(Entity.class);
//        if (!exists){
//            return null;
//        }
//        Entity entity = clazz.getAnnotation(Entity.class);
        String tableName = getTableName(clazz);
        str.append("select * from ").append(tableName);
        String resultStr = str.toString();
        return resultStr;
    }

    //生成delete的sql语句
    public String deleteSQL(Class<?> clazz, Object object) {
        StringBuilder str = new StringBuilder();
        String tableName = getTableName(clazz);
        str.append("delete from ").append(tableName).append(" where ");
        Field[] fields = clazz.getDeclaredFields();
        String pk = "";
        for (int i = 0; i < fields.length; i++) {
            Column column = fields[i].getAnnotation(Column.class);
            if (column.isPK()) {
                pk = column.label();
                break;
            }
        }
        try {
            str.append(pk).append(" = ").append(getValue(pk, object));
        } catch (Exception e) {
            e.printStackTrace();
        }
        String resultStr = str.toString();
        return resultStr;
    }


    //生成insert的sql语句
    public String insertSQL(Class<?> clazz, Object object) {
        StringBuilder str = new StringBuilder();
        String tableName = getTableName(clazz);
        str.append("insert into ").append(tableName).append(" (");
        Field[] fields = clazz.getDeclaredFields();
        for (int i = 0; i < fields.length - 1; i++) {
            Column column = fields[i].getAnnotation(Column.class);
            str.append(column.label()).append(" , ");
        }
        Column columns = fields[fields.length - 1].getAnnotation(Column.class);
        str.append(columns.label()).append(") values(");
        try {
            for (int i = 0; i < fields.length - 1; i++) {
                Column column = fields[i].getAnnotation(Column.class);
                str.append(getValue(column.label(), object)).append(" , ");
            }
            str.append(getValue(columns.label(),object)).append(")");
        } catch (Exception e) {
            e.printStackTrace();
        }


        String resultStr = str.toString();
        return resultStr;
    }

    //生成update的sql语句
    public String updateSQL(Class<?> clazz, Object object) {
        StringBuilder str = new StringBuilder();
        String tableName = getTableName(clazz);
        str.append("update ").append(tableName).append(" set ");
        Field[] fields = clazz.getDeclaredFields();
        String pk = "";
        try {
            for (int i = 0; i < fields.length - 1; i++) {
                Column column = fields[i].getAnnotation(Column.class);
                if (column.isPK()) {
                    pk = column.label();
                } else {

                    str.append(column.label()).append(" = ").append(getValue(column.label(), object)).append(",");

                }
            }
            Column columns = fields[fields.length - 1].getAnnotation(Column.class);
            str.append(columns.label()).append(" = ").append(getValue(columns.label(), object)).append(" ");
            str.append(" where ").append(pk).append(" = ").append(getValue(pk, object));
        } catch (Exception e) {
            e.printStackTrace();
        }
        String resultStr = str.toString();
        return resultStr;
    }


    //获取表的名称
    private String getTableName(Class<?> clazz) {
        //判断是否为Table注释类型是方法返回true，否则返回false
        if (clazz.isAnnotationPresent(Entity.class)) {
            //获取注解信息
            Entity table = clazz.getAnnotation(Entity.class);
            if (!"".equals(table.tableName())) {
                return table.tableName();
            }
        }
        return null;
    }

    //通过对象的值，来获取实际的值
    private <T> String getValue(String columnName, Object object) throws Exception {
        //获取相应字段的getXXX方法
        String getMethod = columnName.substring(0, 1).toUpperCase() + columnName.substring(1);
        Method method = object.getClass().getMethod("get" + getMethod);
        return method.invoke(object).toString();
    }
}

Teache类：

entity注解：

Column注解：

输出结果的图片