造一个方形的轮子5--数据库支持

造一个方形轮子文章目录：造一个方形的轮子

01、先把车正过

在上一篇《造一个方形的轮子4--依赖注入》的最后提出了一个问题,按类型简称（不带包名）以及按注解上设置的Bean名字去初始化Bean的时候都会有覆盖问题，比如不同包下的相同的类，或者在注解上设置了相同Bean名字的类，解决方法不允许重复就可以了，暴力一点有重名的直接抛异常。

BeansInitUtil.java 修改loadClass方法：

    //......上略
    private static void loadClass(File file, Map map){
        ....
                // 按类设置bean
                map.put(beanObject.getClassName(), beanObject);
                String simpleName = firstToLowerCase(beanObject.getSimpleName());
                // 这里添加判断
                if(map.get(simpleName) != null){
                    throw new SquareBeanInitException("There are duplicate beans ，beanName:"+simpleName);
                }
                map.put(simpleName, beanObject);
                // 按注解输入value设置bean
                for (Annotation annotation : annotations) {
                    String tmp_name = "";
                    if(annotation instanceof Service){
                        tmp_name = ((Service)annotation).value();
                    } else if(annotation instanceof Component) {
                        tmp_name = ((Component)annotation).value();
                    }
                    if(tmp_name != null && !tmp_name.equals("")) {
                        // 这里添加判断
                        if(map.get(tmp_name) != null){
                            throw new SquareBeanInitException("There are duplicate beans ，beanName:"+tmp_name);
                        }
                        map.put(tmp_name, beanObject);
                    }
                }
            }
        } catch (Exception e) {
            log.error("init bean error:{}", file.getPath(), e);
        }
    }
    //......下略

复制一个com.jisuye.service.impl.AbcImpl到com.jisuye.service包启动程序查看日志输出：

12:48:17.058 [main] ERROR com.jisuye.core.SquareApplication - There are duplicate beans ，beanName:abcImpl
12:48:17.060 [main] ERROR com.jisuye.core.SquareApplication - Application startup failed...

看日志输出了重复的bean异常提示，现在删除刚复制的AbcImpl.java，将原来com.jisuye.service.impl.DefImpl及com.jisuye.service.impl.DefImpl都使用@Service("def") 注解，然后启动程序查看日志输出：

12:53:53.801 [main] ERROR com.jisuye.core.SquareApplication - There are duplicate beans ，beanName:def
12:53:53.803 [main] ERROR com.jisuye.core.SquareApplication - Application startup failed...

也输出了输出了重复的bean异常提示，验证完毕。

02、数据库支持准备

数据库支持就以mysql为例，添加一个jdbcTemplate类似的功能。

首先引入数据库链接池，这们就不用管链接的问题了，使用的是HikariCP链接池，在pom.xml中添加依赖

        
            mysql
            mysql-connector-java
            ${mysql.version}
        
        
            com.zaxxer
            HikariCP
            ${hikariCP.version}
        

配置文件中添加数据库相关配置

square:
  datasource:
    url: jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf8&useSSL=false&autoReconnect=true
    username: root
    password: 123456

03、DbUtil工具类

将创建数据库链接池及相关方法封装到统一的工具类DbUtil.java：

package com.jisuye.util;
//import ...
/**
 * 数据库操作工具
 * @author ixx
 * @date 2019-07-01
 */
public class DbUtil {
    private static final Logger log = LoggerFactory.getLogger(DbUtil.class);
    private static Connection connection;

    /** 初始化方法*/
    public static void init(){
        try {
            String url = ApplicationContext.getConf("square.datasource.url").toString();
            String username = ApplicationContext.getConf("square.datasource.username").toString();
            String password = ApplicationContext.getConf("square.datasource.password").toString();

            HikariDataSource ds = new HikariDataSource();
            ds.setJdbcUrl(url);
            ds.setUsername(username);
            ds.setPassword(password);
            // HikariCP提供的优化设置
            ds.addDataSourceProperty("cachePrepStmts", "true");
            ds.addDataSourceProperty("prepStmtCacheSize", "250");
            ds.addDataSourceProperty("prepStmtCacheSqlLimit", "2048");
            connection = ds.getConnection();
        } catch (Exception e) {
            log.error("mysql connection init error..", e);
            throw new SquareBeanInitException("mysql connection init error....");
        }
    }

    /** install/update 带参数占位符方法*/
    public static boolean update(String sql, Object... params){
        PreparedStatement statement;
        try {
            statement = connection.prepareStatement(sql);
            if(params != null) {
                for (int i = 1; i <= params.length; i++) {
                    statement.setObject(i, params[i - 1]);
                }
            }
            return statement.execute();
        } catch (Exception e) {
            log.error("install/update exception.", e);
        }
        return false;
    }
    /** install/update 无参数占位符方法*/
    public static boolean update(String sql){
        return update(sql, null);
    }

    /**
     * 通用查询方法
     * @param sql sql语句
     * @param clazz 返回列表类型
     * @param params 参数列表
     * @param  返回列表类型
     * @return
     */
    public static  List select(String sql,Class clazz, Object... params){
        List list = new ArrayList<>();
        PreparedStatement statement;
        try {
            statement = connection.prepareStatement(sql);
            for(int i=1; i<= params.length; i++){
                statement.setObject(i, params[i-1]);
            }
            ResultSet rs = statement.executeQuery();
            while(rs.next()){
                T t = clazz.newInstance();
                Method[] methods = clazz.getMethods();
                for (Method method : methods) {
                    if(method.getName().startsWith("set")){
                        String field = BeansInitUtil.firstToLowerCase(method.getName().substring(3));
                        method.invoke(t, rs.getObject(field));
                    }
                }
                list.add(t);
            }
        } catch (Exception e) {
            log.error("select exception.", e);
        }
        return list.size()>0 ? list : null;
    }
}

init方法用来初始化链接池，update方法执行insert/update/delete语句，select处理查询并简单使用反射封装了一下Entity。

04、添加JdbcTemplate

JdbcTemplate类，只是封装了一下DbUtil的方法，提供给其它service使用，代码如下：

package com.jisuye.core;
//import ...
/**
 * 添加数据库支持(默认不加载该Bean，有其它Bean引用时再加载)
 * @author ixx
 * @date 2019-07-05
 */
public class JdbcTemplate {

    public int insert(String sql){
        return DbUtil.update(sql) ? 1 : 0;
    }
    public int insert(String sql, Object... params){
        return DbUtil.update(sql, params) ? 1 : 0;
    }

    public int update(String sql, Object... params){
        return DbUtil.update(sql, params) ? 1 : 0;
    }
    public int update(String sql){
        return DbUtil.update(sql) ? 1 : 0;
    }

    public int delete(String sql){
        return DbUtil.update(sql) ? 1 : 0;
    }
    public int delete(String sql, Object... params){
        return DbUtil.update(sql, params) ? 1 : 0;
    }

    public  List select(String sql,Class clazz, Object... param){
        return DbUtil.select(sql, clazz, param);
    }
}

修改依赖注入部分代码，因为做的是框架，要考虑不使用DB的情况，所以jdbcTemplate默认不初始化，如果有其它Service添加了JdbcTemplate的依赖，再去初始化。防止直接初始化时，没有配置DB相关配置报错的情况。

BeansInitUtil.initDI()方法做如下修改(只保留了大体结构，具体查看=======标记中间的部分)：

private static void initDI(Map map){
        List