反射+注解或去数据库查询语句

反射+注解获取数据库查询语句


通常我们对数据最经常的操作无非就是增删改查了。那么本次讲解一个简单的小案例,就是用户只需要设置需要查询的对象中的一些基本属性,就可以自动拿到对应的语句,然后通过该语句就可以查询了

Department.java 部门表

@Table("Department")
public class Department {
    @Column("did")
    private int id;// 部门编号
    @Column("name")
    private String name;// 部门名字
    @Column("phone")
    private String phone;// 部门电话

    //下面的无参构造、有参构造、以及所有的set、get方法都已经省略,大家可以自行添加

Employee.java 员工表

@Table("Employee")
public class Employee {

    @Column("id")
    private int id;// 员工编号

    @Column("name")
    private String name;// 员工姓名

    @Column("sex")
    private String sex;// 员工性别

    @Column("age")
    private Integer age;// 员工年龄

    //下面的无参构造、有参构造、以及所有的set、get方法都已经省略,大家可以自行添加

Users.java 用户表

@Table("users")
public class Users {

    @Column("user_id")
    private int id;// 用户编号

    @Column("userName")
    private String name;// 用户名

    @Column("userAge")
    private Integer age;// 用户年龄

    @Column("userSex")
    private String sex;// 用户性别

    @Column("userEmail")
    private String email;// 用户邮箱

    //下面的无参构造、有参构造、以及所有的set、get方法都已经省略,大家可以自行添加

}

Table.java注解 表示每个类上所存在的注解

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface Table {
    String value();
}

Column.java注解 表示每个类中属性(列字段)上的注解

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Column {
    String value();
}

Fileter.java 过滤属性,拿到我们需要查询的语句的功能类

public class Filter {

    public static String query(Object users) throws Exception {
        // 用来保存需要返回的SQL语句
        StringBuilder sb = new StringBuilder();

        // 获取类类型
        Class<? extends Object> cls1 = users.getClass();

        // 判断指定类型的注解是否在此元素上
        boolean isExist = cls1.isAnnotationPresent(Table.class);

        // 如果不存在,直接返回一个空
        if (!isExist) {
            return null;
        }

        // 如果存在该元素的指定类型的注解,则返回这些注解
        Table table = cls1.getAnnotation(Table.class);

        // 通过对应注解获取该值,也就是说我们获得了表名
        String tableName = table.value();

        // 开始添加sql语句,这里加一个where 1 = 1是为了防止如果用户没有设置对应的查询信息时
        //where语句后面没有对应的查询语句,产生语法错误。
        sb.append("select * from ").append(tableName).append(" where 1=1 ");

        // 获取该类型中所有的字段
        Field[] fields = cls1.getDeclaredFields();

        // 通过遍历该字段来完成列名的拼接
        for (Field f : fields) {

            // 判断指定类型的注解是否在此字段上
            boolean isFExist = f.isAnnotationPresent(Column.class);

            // 如果不在就不进行获取该字段
            if (!isFExist) {
                continue;
            }

            // 存在该元素指定的注解,就获取这个注解
            Column column = f.getAnnotation(Column.class);

            // 通过该注解拿到对应的列名字段
            String columnName = column.value();

            // 获取属性的字段名,当然,我们这里设置的列字段名和属性字段名是一致,之所以还需要用到列名字
            //段,是为了程序的可维护性
            String fieldName = f.getName();

            // 先拿到对应字段的方法
            String getMethodName = "get" + fieldName.substring(0, 1).toUpperCase() + fieldName.substring(1);

            // 通过反射拿到该类中对应已声明的方法
            Method method = cls1.getDeclaredMethod(getMethodName);

            // 通过拿到对应的方法,然后调用该方法,获取指定的值
            Object fieldValue = method.invoke(users);

            // 如果拿到的字符串值是空,年龄是0,我们就不需要做查询
            if (fieldValue == null || (fieldValue instanceof Integer && (Integer) fieldValue == 0)){
                continue;
            }

             // 否则的话,我们就可以对其
            sb.append(" and ").append(columnName).append("=");进行拼接

            // 如果是字符串,我们要讲其添加对应的单引号
            if (fieldValue instanceof String) {
                // 如果是邮箱,我们可以查询其中任意的一个邮箱,通过in来获取邮箱组中的任意存在的一个
                if ("userEmail".equals(fieldName)) {
                    String[] split = ((String) fieldValue).split(",");
                    sb.append("in(");
                    for (String s : split) {
                        sb.append("'").append(s).append("',");
                    }
                    //删除最后一个","
                    sb.deleteCharAt(sb.length() - 1);
                    sb.append(")");
                } else {
                    sb.append("'").append(fieldValue).append("'");
                }
                    // 如果是整形,就拼接整形数据
            } else if (fieldValue instanceof Integer) {
                sb.append(fieldValue);
            }
        }
        // 返回拼接好的SQL语句
        return sb.toString();
    }
}

Caller.java测试类

public class Caller {
    public static void main(String[] args) {
        //我们要获取users表中对应ID为10   邮箱为12234@qq.com的查询语句
        Users users = new Users();
        users.setId(10);
        users.setEmail("12234@qq.com");

        //获取部门表中ID为20   电话为:15823453158的查询语句
        Department department = new Department();
        department.setId(20);
        department.setPhone("15823453158");


        //获取员工表中ID 为50  年龄为30  名字为张三的查询语句  
        Employee emp = new Employee();
        emp.setId(50);
        emp.setAge(30);
        emp.setName("张三");
        try {
            String sql = Filter.query(users);
            System.out.println("用户表对应的查询语句:"+sql);
            sql = Filter.query(department);
            System.out.println("部门表对应的查询语句" + sql);
            sql = Filter.query(emp);
            System.out.println("员工表对应的查询语句" + sql);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

运行结果如下:
这里写图片描述

对于以上功能是不是很简单呢,只要掌握了反射和注解,就可以很好的运用他们给我们带来的方便了,而不需要我自己去书写语句了,如果有兴趣,可以自己尝试着去实现一套完整的对sql语句进行增删改查的功能。

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Java通过反射注解生成增删改的SQL语句主要涉及以下几个方法: 1. 获取实体类对应的数据库表名 2. 获取实体类中的属性名和对应的列名 3. 生成查询语句 4. 生成插入语句 5. 生成更新语句 6. 生成删除语句 下面分别对每个方法进行详细介绍: 1. 获取实体类对应的数据库表名 这个方法比较简单,只需要在实体类上使用一个注解来标识表名即可,例如: ``` @Table(name = "user") public class User { // ... } ``` 然后在获取表名的方法中,使用反射获取类上的注解信息即可: ``` public static String getTableName(Class<?> clazz) { Table table = clazz.getAnnotation(Table.class); if (table != null) { return table.name(); } return null; } ``` 2. 获取实体类中的属性名和对应的列名 同样地,在实体类中使用注解来标识属性对应的列名,例如: ``` @Column(name = "username") private String username; ``` 然后在获取列名的方法中,使用反射获取类中所有的属性,遍历每个属性,获取注解信息即可: ``` public static Map<String, String> getColumnMap(Class<?> clazz) { Map<String, String> map = new HashMap<>(); Field[] fields = clazz.getDeclaredFields(); for (Field field : fields) { Column column = field.getAnnotation(Column.class); if (column != null) { map.put(field.getName(), column.name()); } } return map; } ``` 3. 生成查询语句 查询语句比较简单,只需要拼接出 SELECT 和 FROM 子句即可。我们可以在方法中传入一个 Map 类型的参数,用来指定询条件: ``` public static String generateSelectSql(Class<?> clazz, Map<String, Object> condition) { String tableName = getTableName(clazz); Map<String, String> columnMap = getColumnMap(clazz); StringBuilder sb = new StringBuilder(); sb.append("SELECT "); for (String columnName : columnMap.values()) { sb.append(columnName).append(", "); } sb.delete(sb.length() - 2, sb.length()); sb.append(" FROM ").append(tableName); if (condition != null && !condition.isEmpty()) { sb.append(" WHERE "); for (Map.Entry<String, Object> entry : condition.entrySet()) { String fieldName = entry.getKey(); String columnName = columnMap.get(fieldName); Object value = entry.getValue(); sb.append(columnName).append(" = "); if (value instanceof String) { sb.append("'").append(value).append("'"); } else { sb.append(value); } sb.append(" AND "); } sb.delete(sb.length() - 5, sb.length()); } return sb.toString(); } ``` 4. 生成插入语句 插入语句需要获取实体类中的属性值,然后将其拼接成一条 SQL 语句。我们可以在方法中传入一个实体类对象,用来指定要插入的数据: ``` public static String generateInsertSql(Object entity) { Class<?> clazz = entity.getClass(); String tableName = getTableName(clazz); Map<String, String> columnMap = getColumnMap(clazz); StringBuilder sb = new StringBuilder(); sb.append("INSERT INTO ").append(tableName).append(" ("); for (String columnName : columnMap.values()) { sb.append(columnName).append(", "); } sb.delete(sb.length() - 2, sb.length()); sb.append(") VALUES ("); Field[] fields = clazz.getDeclaredFields(); for (Field field : fields) { Column column = field.getAnnotation(Column.class); if (column != null) { field.setAccessible(true); Object value; try { value = field.get(entity); } catch (IllegalAccessException e) { throw new RuntimeException(e); } if (value instanceof String) { sb.append("'").append(value).append("'"); } else { sb.append(value); } sb.append(", "); } } sb.delete(sb.length() - 2, sb.length()); sb.append(")"); return sb.toString(); } ``` 5. 生成更新语句 更新语句与插入语句类似,只需要将属性和属性值拼接成 SET 子句即可。我们可以在方法中传入一个实体类对象和一个 Map 类型的参数,用来指定更新条件: ``` public static String generateUpdateSql(Object entity, Map<String, Object> condition) { Class<?> clazz = entity.getClass(); String tableName = getTableName(clazz); Map<String, String> columnMap = getColumnMap(clazz); StringBuilder sb = new StringBuilder(); sb.append("UPDATE ").append(tableName).append(" SET "); Field[] fields = clazz.getDeclaredFields(); for (Field field : fields) { Column column = field.getAnnotation(Column.class); if (column != null) { String fieldName = field.getName(); String columnName = columnMap.get(fieldName); field.setAccessible(true); Object value; try { value = field.get(entity); } catch (IllegalAccessException e) { throw new RuntimeException(e); } if (value instanceof String) { sb.append(columnName).append(" = '").append(value).append("', "); } else { sb.append(columnName).append(" = ").append(value).append(", "); } } } sb.delete(sb.length() - 2, sb.length()); if (condition != null && !condition.isEmpty()) { sb.append(" WHERE "); for (Map.Entry<String, Object> entry : condition.entrySet()) { String fieldName = entry.getKey(); String columnName = columnMap.get(fieldName); Object value = entry.getValue(); sb.append(columnName).append(" = "); if (value instanceof String) { sb.append("'").append(value).append("'"); } else { sb.append(value); } sb.append(" AND "); } sb.delete(sb.length() - 5, sb.length()); } return sb.toString(); } ``` 6. 生成删除语句 删除语句与查询语句类似,只需要拼接出 DELETE 和 FROM 子句即可。我们可以在方法中传入一个 Map 类型的参数,用来指定删除条件: ``` public static String generateDeleteSql(Class<?> clazz, Map<String, Object> condition) { String tableName = getTableName(clazz); Map<String, String> columnMap = getColumnMap(clazz); StringBuilder sb = new StringBuilder(); sb.append("DELETE FROM ").append(tableName); if (condition != null && !condition.isEmpty()) { sb.append(" WHERE "); for (Map.Entry<String, Object> entry : condition.entrySet()) { String fieldName = entry.getKey(); String columnName = columnMap.get(fieldName); Object value = entry.getValue(); sb.append(columnName).append(" = "); if (value instanceof String) { sb.append("'").append(value).append("'"); } else { sb.append(value); } sb.append(" AND "); } sb.delete(sb.length() - 5, sb.length()); } return sb.toString(); } ``` 以上就是通过反射注解生成增删改的 SQL 语句的所有方法。需要注意的是,这里只是简单地介绍了如何使用反射注解生成 SQL 语句,实际上在实际开发中还需要考虑很多其他的因素,例如 SQL 注入、事务处理等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值