反射+注解实现数据库Map转换成实体Map工具类

àì夳堔傛蜴生んèń 2022-04-13 04:10 320阅读 0赞

### **实现场景** ###

因为原始的需求是从数据库中查询出来的一个实体字段

比如一个User user = new User（）的实体类，有userName，userId两个字段。

然后我返回一个list给前端，List<User> list

现在需求变化，我需要添加一个额外的字段，比如userCompany，就没有办法使用刚才那个返回对象的方法

现在直接从数据库中多表查询一个Map 如下

("user\_name","curry"),("user\_id","00030"),("user\_company","worriors")

我现在返回给前端的list就变成 List<Map> list

又因为数据库的字段命名方式为下划线分割，而实体类的是驼峰命名

这样的Map传给前端，因为前端取的是原先实体类的key，现在key发生了变化,会导致前端报错

而这添加一个字段，又是一个比较常用的需求，所以想写一个工具来进行适配。

### **实现** ###

**实现前提**

1.项目中的实体类应严格按照javabean的标准来生成；

2.属性命名使用驼峰命名法；

3.get方法使用注解

![20181127110314226.png][]

**实现开始**

static <T> Map convertMapToBean(Map map, Class<T> entity) {
            try {
                //1.获取类
                Class cEntity = entity;
                //2.通过反射得到所有方法
                Method[] methods = cEntity.getMethods();
                //3.新建一个来存放key转变的entry
                Map<String,String> listMap = new LinkedHashMap<>();
                //4.遍历传进来的map
                Iterator<Map.Entry<String, String>> iterator = map.entrySet().iterator();
                while (iterator.hasNext()){
                    Map.Entry<String,String> entry = iterator.next();
                    String key = entry.getKey();
                    String value = entry.getValue();
                    //遍历方法
                    for (Method method : methods){
                        //判断次方法上的注解是否存在
                        boolean isMExist = method.isAnnotationPresent(Column.class);
                        if(isMExist){
                            //得到此方法的注解
                            Column column = (Column)method.getAnnotation(Column.class);
                            //得到方法名
                            String methodName = method.getName();
                            //根据get方法和字段的关系，第一步把get去掉
                            String MethodName1 = methodName.substring(3);
                            //第二步把属性的首字母变小写，得到属性字段名
                            String propertyName = new StringBuffer().append(Character.toLowerCase(MethodName1.charAt(0))).
                                    append(MethodName1.substring(1)).toString();
                            //判断注解的名字和数据库的字段是否相等，相等则进行替换
                            if(column.name().equals(key)){
                                iterator.remove();
                                listMap.put(propertyName,value);
                            }
                        }
                    }
                }
    
                for (Map.Entry<String, String> entry : listMap.entrySet()) {
                    map.put(entry.getKey(),entry.getValue());
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
            return map;
        }
    }

**注意的点**

**1.注意获取类的几种方法，我这里直接把类传进来**

**2.注意自己实体类get方法和属性名的关联，根据实际情况来**

**3.注意遍历Map时，用remove方法时，一定要使用我示例的这种遍历方式，不然会出现错误**

**4.get方法上的注解的值，一定要和数据库里的字段值一致**

**5.传参的具体类型可以根据自己的实际需求来设定**

[20181127110314226.png]: /images/20220413/18036d9815b24b5c82f54341aaef9176.png