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注解:
在这里插入图片描述
输出结果的图片
在这里插入图片描述

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值