-------前篇:手写DAO框架(二)-开发前的最后准备---------
前言
上一篇主要是温习了一下基础知识,然后将整个项目按照模块进行了划分。因为是个人项目,一个人开发,本人采用了自底向上的开发。
本篇会介绍连接层的代码,包括三部分:pom配置、数据库连接和数据库连接池。
pom配置
1
2 xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
3 4.0.0
4
5 me.lovegao
6 gdao
7 0.0.2-SNAPSHOT
8 jar
9
10 gdao
11 http://maven.apache.org
12
13
14 UTF-8
15 4.12
16
17
18
19
20 junit
21 junit
22 ${junit.version}
23 test
24
25
26 org.slf4j
27 slf4j-api
28 1.7.12
29
30
31 com.alibaba
32 fastjson
33 1.2.31
34
35
36 org.apache.commons
37 commons-lang3
38 3.2.1
39
40
41 mysql
42 mysql-connector-java
43 5.1.31
44
45
46
47 me_lovegao_gdao
48
49
50 org.apache.maven.plugins
51 maven-compiler-plugin
52 3.7.0
53
54 1.8
55 1.8
56
57
58
59
60
maven定义了jdk的版本,免得导入IDE的时候变成1.5,又得修改。
数据库连接
数据库连接主要就是为了获取数据库连接,这个模块不关注连接的使用、关闭等,只提供获取连接,其他逻辑由调用方负责。
书上、网上都说,要面向接口编程。
本来我是定义了一个接口的,但是最终实现的时候没有用上。回头看这段代码,感觉用上接口之后,依赖关系有点复杂,所以就没改了。
而且,这里的获取连接,定位的是一个工具类,所以直接写了静态方法。
1、连接获取类:ConnectionGetor
1 packageme.lovegao.gdao.connection;2
3 importjava.sql.Connection;4 importjava.sql.DriverManager;5 importjava.util.Properties;6
7 importorg.slf4j.Logger;8 importorg.slf4j.LoggerFactory;9
10 importme.lovegao.gdao.bean.SystemConstant;11
12 public classConnectionGetor {13 private final static Logger log = LoggerFactory.getLogger(ConnectionGetor.class);14
15 public static Connection createConnection(Properties properties) throwsException {16 Connection conn = null;17 String driverName =properties.getProperty(SystemConstant.STR_DRIVER_NAME);18 String url =properties.getProperty(SystemConstant.STR_CONNECTION_URL);19 String userName =properties.getProperty(SystemConstant.STR_USER_NAME);20 String passwd =properties.getProperty(SystemConstant.STR_USER_PASSWORD);21 try{22 Class.forName(driverName);23 conn =DriverManager.getConnection(url, userName, passwd);24 } catch(Exception e) {25 log.error("createConnectionException", e);26 }27 returnconn;28 }29
30 }
为了保证通用性,入参我选择了Properties类型。这样,配置既可以从本地加载,也可以从流加载,确保了灵活性。
为了能够支持多数据源,所以这个方法里没有定义和数据库连接相关的局部变量或者常量。
2、数据库实例配置:Properties
说到了配置,我就把配置的字段先列一下。毕竟框架再好用,不教人怎么配置就是扯。
##驱动名称
driverName=com.mysql.jdbc.Driver
##连接url
connectionUrl=jdbc:mysql://localhost:3306/simple?useServerPrepStmts=false&rewriteBatchedStatements=true&connectTimeout=1000&useUnicode=true&characterEncoding=utf-8
##用户名
userName=name
##用户密码
userPassword=123456
##初始化连接数
initConnectionNum=10
##最大连接数
maxConnectionNum=50
##最大查询等待时间-秒
maxQueryTime=3
配置这块就是这样的。为什么定义了这几个配置,在前两篇文章里也有涉及,这里不再赘述。
数据库连接池
因为数据库连接的创建是需要时间的,我们可以在项目初始化的时候先创建一批连接,在需要使用连接的时候可以节省创建连接的等待时间,从而提高程序响应速度。而且,连接是可以复用的,所以,项目中使用连接池是有意义的。
要面向接口编程,所以这里先定义接口。
1、连接池接口:IConnectionPool
1 packageme.lovegao.gdao.connection;2
3 importjava.sql.Connection;4
5 /**
6 * 连接池7 *@authorsimple8 *9 */
10 public interfaceIConnectionPool {11 /**
12 * 获取连接13 *@return
14 */
15 Connection getConnection() throwsException;16 /**
17 * 归还连接18 *@paramconn19 */
20 voidreturnConnection(Connection conn);21
22 /**
23 * 获取查询超时时间24 *@return
25 */
26 intgetQueryTimeoutSecond();27 }
从代码可以看到,连接池的主要功能包括:获取连接,归还连接。
“获取查询超时时间”这个接口是我后来加的,作为一个框架,应该保证不因为一个查询耗时超过正常区间仍然义无反顾的等待。只不过加在这里,感觉不太优雅,暂时也没有想到更好的写法,所以还是暂时放在这里吧。(有建议欢迎提出)
2、连接池的简单实现:SimpleConnectionPool
1 packageme.lovegao.gdao.connection;2
3 importjava.sql.Connection;4 importjava.util.Map;5 importjava.util.Properties;6 importjava.util.Queue;7 importjava.util.concurrent.ConcurrentHashMap;8 importjava.util.concurrent.ConcurrentLinkedQueue;9 importjava.util.concurrent.atomic.AtomicInteger;10
11 importorg.slf4j.Logger;12 importorg.slf4j.LoggerFactory;13
14 importme.lovegao.gdao.bean.SystemConstant;15
16 public class SimpleConnectionPool implementsIConnectionPool {17 private final static Logger log = LoggerFactory.getLogger(SimpleConnectionPool.class);18 /**配置**/
19 privateProperties properties;20 /**保存连接的map**/
21 private final Map CONNECTION_MAP_POOL = newConcurrentHashMap();22 /**连接池的连接索引**/
23 private final Queue CONNECTION_KEY_POOL = newConcurrentLinkedQueue();24 /**连接池初始连接数量**/
25 private intPOOL_INIT_NUM;26 /**连接池最大连接数量**/
27 private intPOOL_MAX_NUM;28 /**已创建连接数量**/
29 private AtomicInteger POOL_CREATE_NUM = new AtomicInteger(0);30 /**查询超时时间-秒**/
31 private intQUERY_TIMEOUT_SECONDS;32
33 public SimpleConnectionPool(Properties properties) throwsException {34 this.properties =properties;35 this.POOL_INIT_NUM =Integer.parseInt(properties.getProperty(SystemConstant.STR_INIT_CONNECTION_NUM));36 this.POOL_MAX_NUM =Integer.parseInt(properties.getProperty(SystemConstant.STR_MAX_CONNECTION_NUM));37 this.QUERY_TIMEOUT_SECONDS =Integer.parseInt(properties.getProperty(SystemConstant.STR_QUERY_TIME));38 for(int i=0; i
46 @Override47 public Connection getConnection() throwsException {48 Connection conn = null;49 Integer connKey =CONNECTION_KEY_POOL.poll();50 if(connKey == null) {51 if(POOL_CREATE_NUM.intValue()
64 if(conn == null) {65 throw new NullPointerException("连接池连接用完");66 }67 returnconn;68 }69
70 @Override71 public voidreturnConnection(Connection conn) {72 if(conn != null) {73 try{74 if(conn.isClosed()) {75 CONNECTION_MAP_POOL.remove(conn.hashCode());76 POOL_CREATE_NUM.decrementAndGet();77 } else{78 CONNECTION_KEY_POOL.add(conn.hashCode());79 }80 } catch(Exception e) {81 log.error("returnConnectionException", e);82 }83 }84 }85
86 @Override87 public intgetQueryTimeoutSecond() {88 returnQUERY_TIMEOUT_SECONDS;89 }90
91 }
逻辑简介:初始化的时候,先创建指定数量的连接存起来。在获取连接的时候,如果还有连接,就返回;如果没有连接了,就判断连接数是否到了最大上限,没有到就创建并返回,到上限了就返回空。
其他说明:
为了保证并发安全,这里是通过POOL_CREATE_NUM变量来保证的。
为了支持多数据源,这里同样没有静态变量。
为了保存连接,这里定义了两个容器,分别是Map CONNECTION_MAP_POOL和Queue CONNECTION_KEY_POOL。为了并发安全,使用了ConcurrentHashMap和ConcurrentLinkedQueue来存储。
其中,队列是用于取出连接和返回连接,这里队列存放的是连接对应的key。MAP是为了保存对所有连接的引用,防止一些连接没有归还而系统却不知道,同时也可以知道连接的总数。也有一点考虑,是为了以后可以有单独线程来定时判断哪些线程应该关闭,主要是为了以后扩展用。
最初,我想把连接池做成那种:获取连接的时候,如果没有连接,就等待一段时间(指定时间,即获取连接等待时间)之后再试。查了一下对应的技术栈,综合考虑了一下,感觉有些不足之处,比如:如果连接用完了,可能出现大量线程等待的现象,反而会影响性能。
后来参考了一下其他连接池的实现,最终做成目前的这样:获取连接的时候,没有连接,看能不能创建,能创建就创建,不能创建就返回失败。逻辑简单,容易实现,快速错误。大概这就是fast-fail吧!
--本文内容到这里就结束了,欢迎指导交流--