java 分库关联查询工具类

问题:

  因为公司业务扩大,各个子系统陆续迁移和部署在不一样的数据源上,这样方便扩容,可是所以引出了一些问题。java

  举个例子:在查询"订单"(位于订单子系统)列表时,同时须要查询出所关联的"用户"(位于帐户子系统)的姓名,而这时因为数据存储在不一样的数据源上,没有办法经过一条连表的sql获取到所有的数据,而是必须进行两次数据库查询,从不一样的数据源分别获取数据,而且在web服务器中进行关联映射。在观察了一段时间后,发现进行关联映射的代码大部分都是模板化的,所以产生一个想法,想要把这些模板代码抽象出来,简化开发,也加强代码的可读性。同时,即便在同一个数据源上,若是能将多表联查的需求转化为单表屡次查询,也可以减小代码的耦合,同时提升数据库效率。git

设计主要思路:

在关系型数据库中:github

  一对一的关系通常表示为:一方的数据表结构中存在一个业务上的外键关联另外一张表的主键(订单和用户是一对一的关系,则订单表中存在外键对应于用户表的主键)。web

  一对多的关系通常表示为:多方的数据中存在一个业务上的外键关联一方的主键(门店和订单是一对多的关系,则订单表中存在外键对应于门店的主键)。sql

而在非关系型数据库中:数据库

  一对一的关系通常表示为:一方中存在一个属性,值为关联的另外一方的数据对象(订单和用户是一对一的关系,则订单对象中存在一个用户属性)。bash

  一对多的关系通常表示为:一方中存在一个属性,值为关联的另外一方的数据对象列表(门店和所属订单是一对多的关系,则门店对象表存在一个订单列表(List)属性)。服务器

  能够看出java的对象机制,自然就支持非关系型的数据模型,所以大概的思路就是,将查询出来的两个列表进行符合要求的映射便可。函数

  pojo类:   工具

public class OrderForm {
    /**
     * 主键id
     * */
    private String id;
    /**
     * 所属门店id
     * */
    private String shopID;
    /**
     * 关联的顾客id
     * */
    private String customerID;
    /**
     * 关联的顾客model
     * */
    private Customer customer;
}

public class Customer {
    /**
     * 主键id
     * */
    private String id;
    /**
     * 姓名
     * */
    private String userName;
}

public class Shop {
    /**
     * 主键id
     * */
    private String id;
    /**
     * 门店名
     * */
    private String shopName;
    /**
     * 订单列表 (一个门店关联N个订单 一对多)
     * */
    private List<OrderForm> orderFormList;
}
复制代码

  辅助工具函数:   

/***
     * 将经过keyName得到对应的bean对象的get方法名称的字符串
     * @param keyName 属性名
     * @return  返回get方法名称的字符串
     */
    private static String makeGetMethodName(String keyName){
        //:::将第一个字母转为大写
        String newKeyName = transFirstCharUpperCase(keyName);

        return "get" + newKeyName;
    }

    /***
     * 将经过keyName得到对应的bean对象的set方法名称的字符串
     * @param keyName 属性名
     * @return  返回set方法名称的字符串
     */
    private static String makeSetMethodName(String keyName){
        //:::将第一个字母转为大写
        String newKeyName = transFirstCharUpperCase(keyName);

        return "set" + newKeyName;
    }

    /**
     * 将字符串的第一个字母转为大写
     * @param str 须要被转变的字符串
     * @return 返回转变以后的字符串
     */
    private static String transFirstCharUpperCase(String str){
        return str.replaceFirst(str.substring(0, 1), str.substring(0, 1).toUpperCase());
    }

    /**
     * 判断当前的数据是否须要被转换
     *
     * 两个列表存在一个为空,则不须要转换
     * @return 不须要转换返回 false,须要返回 true
     * */
    private static boolean needTrans(List beanList,List dataList){
        if(listIsEmpty(beanList) || listIsEmpty(dataList)){
            return false;
        }else{
            return true;
        }
    }

    /**
     * 列表是否为空
     * */
    private static boolean listIsEmpty(List list){
        if(list == null || list.isEmpty()){
            return true;
        }else{
            return false;
        }
    }

/**
     * 将javaBean组成的list去重 转为map, key为bean中指定的一个属性
     *
     * @param beanList list 自己
     * @param keyName 生成的map中的key
     * @return
     * @throws Exception
     */
    public static Map<String,Object> beanListToMap(List beanList,String keyName) throws Exception{
        //:::建立一个map
        Map<String,Object> map = new HashMap<>();

        //:::由keyName得到对应的get方法字符串
        String getMethodName = makeGetMethodName(keyName);

        //:::遍历beanList
        for(Object obj : beanList){
            //:::若是当前数据是hashMap类型
            if(obj.getClass() == HashMap.class){
                Map currentMap = (Map)obj;

                //:::使用keyName从map中得到对应的key
                String result = (String)currentMap.get(keyName);

                //:::放入map中(若是key同样,则会被覆盖去重)
                map.put(result,currentMap);
            }else{
                //:::不然默认是pojo对象
                //:::得到get方法
                Method getMethod = obj.getClass().getMethod(getMethodName);

                //:::经过get方法从bean对象中获得数据key
                String result = (String)getMethod.invoke(obj);

                //:::放入map中(若是key同样,则会被覆盖去重)
                map.put(result,obj);
            }
        }
        //:::返回结果
        return map;
    }
复制代码

一对一链接接口定义:

/**
      * 一对一链接 :  beanKeyName <---> dataKeyName 做为链接条件
      *
      * @param beanList 须要被存放数据的beanList(主体)
      * @param beanKeyName   beanList中链接字段key的名字
      * @param beanModelName  beanList中用来存放匹配到的数据value的属性
      * @param dataList  须要被关联的data列表
      * @param dataKeyName 须要被关联的data中链接字段key的名字
      *
      * @throws Exception
      */
     public static void oneToOneLinked(List beanList, String beanKeyName, String beanModelName, List dataList, String dataKeyName) throws Exception { }
复制代码

若是带入上述一对一链接的例子,beanList是订单列表(List),beanKeyName是订单用于关联用户的字段名称(例如外键“OrderForm.customerID”),beanModelName是用于存放用户类的字段名称("例如OrderForm.customer"),dataList是顾客列表(List),dataKeyName是被关联数据的key(例如主键"Customer.id")。

  一对一链接代码实现:   

/**
     * 一对一链接 :  beanKeyName <---> dataKeyName 做为链接条件
     *
     * @param beanList 须要被存放数据的beanList(主体)
     * @param beanKeyName   beanList中链接字段key的名字
     * @param beanModelName  beanList中用来存放匹配到的数据value的属性
     * @param dataList  须要被关联的data列表
     * @param dataKeyName 须要被关联的data中链接字段key的名字
     *
     * @throws Exception
     */
    public static void oneToOneLinked(List beanList, String beanKeyName, String beanModelName, List dataList, String dataKeyName) throws Exception {
        //:::若是不须要转换,直接返回
        if(!needTrans(beanList,dataList)){
            return;
        }
        //:::将被关联的数据列表,以须要链接的字段为key,转换成map,加快查询的速度
        Map<String,Object> dataMap = beanListToMap(dataList,dataKeyName);

        //:::进行数据匹配链接
    &emsp;&emsp; matchedDataToBeanList(beanList,beanKeyName,beanModelName,dataMap);
&emsp;&emsp;}

/**
     * 将批量查询出来的数据集合,组装到对应的beanList之中
     * @param beanList 须要被存放数据的beanList(主体)
     * @param beanKeyName   beanList中用来匹配数据的属性
     * @param beanModelName  beanList中用来存放匹配到的数据的属性
     * @param dataMap  data结果集以某一字段做为key对应的map
     * @throws Exception
     */
    private static void matchedDataToBeanList(List beanList, String beanKeyName, String beanModelName, Map<String,Object> dataMap) throws Exception {
        //:::得到beanList中存放对象的key的get方法名
        String beanGetMethodName = makeGetMethodName(beanKeyName);
        //:::得到beanList中存放对象的model的set方法名
        String beanSetMethodName = makeSetMethodName(beanModelName);

        //:::遍历整个beanList
        for(Object bean : beanList){
            //:::得到bean中key的method对象
            Method beanGetMethod = bean.getClass().getMethod(beanGetMethodName);

            //:::调用得到当前的key
            String currentBeanKey = (String)beanGetMethod.invoke(bean);

            //:::从被关联的数据集map中找到匹配的数据
            Object matchedData = dataMap.get(currentBeanKey);

            //:::若是找到了匹配的对象
            if(matchedData != null){
                //:::得到bean中对应model的set方法
                Class clazz = matchedData.getClass();

                //:::若是匹配到的数据是hashMap
                if(clazz == HashMap.class){
                    //:::转为父类map class用来调用set方法
                    clazz = Map.class;
                }

                //:::得到主体bean用于存放被关联对象的set方法
                Method beanSetMethod = bean.getClass().getMethod(beanSetMethodName,clazz);
                //:::执行set方法,将匹配到的数据放入主体数据对应的model属性中
                beanSetMethod.invoke(bean,matchedData);
            }
        }
    }
复制代码

一对多链接接口定义:

/**
     * 一对多链接 :  oneKeyName <---> manyKeyName 做为链接条件
     *
     * @param oneDataList       '一方' 数据列表
     * @param oneKeyName        '一方' 链接字段key的名字
     * @param oneModelName      '一方' 用于存放 '多方'数据的列表属性名
     * @param manyDataList      '多方' 数据列表
     * @param manyKeyName       '多方' 链接字段key的名字
     *
     *  注意:  '一方' 存放 '多方'数据的属性oneModelName类型必须为List
     *
     * @throws Exception
     */
    public static void oneToManyLinked(List oneDataList,String oneKeyName,String oneModelName,List manyDataList,String manyKeyName) throws Exception {}
复制代码

若是带入上述一对多链接的例子,oneDataList是门店列表(List),oneKeyName是门店用于关联订单的字段名称(例如主键“Shop.id”),oneModelName是用于存放订单列表的字段名称(例如"Shop.orderFomrList"),manyDataList是多方列表(List),manyKeyName是被关联数据的key(例如外键"OrderFrom.shopID")。

  一对多链接代码实现:   

/**
     * 一对多链接 :  oneKeyName <---> manyKeyName 做为链接条件
     *
     * @param oneDataList       '一方' 数据列表
     * @param oneKeyName        '一方' 链接字段key的名字
     * @param oneModelName      '一方' 用于存放 '多方'数据的列表属性名
     * @param manyDataList      '多方' 数据列表
     * @param manyKeyName       '多方' 链接字段key的名字
     *
     *  注意:  '一方' 存放 '多方'数据的属性oneModelName类型必须为List
     *
     * @throws Exception
     */
    public static void oneToManyLinked(List oneDataList,String oneKeyName,String oneModelName,List manyDataList,String manyKeyName) throws Exception {
        if(!needTrans(oneDataList,manyDataList)){
            return;
        }
        //:::将'一方'数据,以链接字段为key,转成map,便于查询
        Map<String,Object> oneDataMap = beanListToMap(oneDataList,oneKeyName);

        //:::得到'一方'存放 '多方'数据字段的get方法名
        String oneDataModelGetMethodName = makeGetMethodName(oneModelName);
        //:::得到'一方'存放 '多方'数据字段的set方法名
        String oneDataModelSetMethodName = makeSetMethodName(oneModelName);

        //:::得到'多方'链接字段的get方法名
        String manyDataKeyGetMethodName = makeGetMethodName(manyKeyName);

        try {
            //:::遍历'多方'列表
            for (Object manyDataItem : manyDataList) {
                //:::'多方'对象链接key的值
                String manyDataItemKey;
                //:::判断当前'多方'对象的类型是不是 hashMap
                if(manyDataItem.getClass() == HashMap.class){
                    //:::若是是hashMap类型的,先转为Map对象
                    Map manyDataItemMap = (Map)manyDataItem;

                    //:::经过参数key 直接获取对象key链接字段的值
                    manyDataItemKey = (String)manyDataItemMap.get(manyKeyName);
                }else{
                    //:::若是是普通的pojo对象,则经过反射得到get方法来获取key链接字段的值

                    //:::得到'多方'数据中key的method对象
                    Method manyDataKeyGetMethod = manyDataItem.getClass().getMethod(manyDataKeyGetMethodName);
                    //:::调用'多方'数据的get方法得到当前'多方'数据链接字段key的值
                    manyDataItemKey = (String) manyDataKeyGetMethod.invoke(manyDataItem);
                }
                //:::经过'多方'的链接字段key从 '一方' map集合中查找出链接key相同的 '一方'数据对象
                Object matchedOneData = oneDataMap.get(manyDataItemKey);

                //:::若是匹配到了数据,才进行操做
                if(matchedOneData != null){
                    //:::将当前迭代的 '多方'数据 放入 '一方' 的对应的列表中
                    setManyDataToOne(matchedOneData,manyDataItem,oneDataModelGetMethodName,oneDataModelSetMethodName);
                }
            }
        }catch(Exception e){
            throw new Exception(e);
        }
    }

 /**
     * 将 '多方' 数据存入 '一方' 列表中
     * @param oneData 匹配到的'一方'数据
     * @param manyDataItem  当前迭代的 '多方数据'
     * @param oneDataModelGetMethodName 一方列表的get方法名
     * @param oneDataModelSetMethodName 一方列表的set方法名
     * @throws Exception
     */
    private static void setManyDataToOne(Object oneData,Object manyDataItem,String oneDataModelGetMethodName,String oneDataModelSetMethodName) throws Exception {
        //:::得到 '一方' 数据中存放'多方'数据属性的get方法
        Method oneDataModelGetMethod = oneData.getClass().getMethod(oneDataModelGetMethodName);
        //::: '一方' 数据中存放'多方'数据属性的set方法
        Method oneDataModelSetMethod;
        try {
            //::: '一方' set方法对象
            oneDataModelSetMethod = oneData.getClass().getMethod(oneDataModelSetMethodName,List.class);
        }catch(NoSuchMethodException e){
            throw new Exception("未找到知足条件的'一方'set方法");
        }

        //:::得到存放'多方'数据get方法返回值类型
        Class modelType = oneDataModelGetMethod.getReturnType();
        //::: get方法返回值必须是List
        if(modelType.equals(List.class)){
            //:::调用get方法,得到数据列表
            List modelList = (List)oneDataModelGetMethod.invoke(oneData);

            //:::若是当前成员变量为null
            if(modelList == null){
                //:::建立一个新的List
                List newList = new ArrayList<>();
                //:::将当前的'多方'数据存入list
                newList.add(manyDataItem);
                //:::将这个新建立出的List赋值给 '一方'的对象
                oneDataModelSetMethod.invoke(oneData,newList);
            }else{
                //:::若是已经存在了List

                //:::直接将'多方'数据存入list
                modelList.add(manyDataItem);
            }
        }else{
            throw new Exception("一对多链接时,一方指定的model对象必须是list类型");
        }
    }
复制代码

测试用例在个人github上面 github.com/1399852153/…

相关文章
相关标签/搜索