1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
package com.yanzuoguang.db.impl;
import com.yanzuoguang.db.DbExecute;
import com.yanzuoguang.extend.ConfigDb;
import com.yanzuoguang.util.vo.MapRow;
import com.yanzuoguang.util.vo.Ref;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowCallbackHandler;
import org.springframework.jdbc.support.rowset.SqlRowSet;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
/**
* 数据库操作类
*
* @author 颜佐光
*/
@Component
public class DbExecuteImpl implements DbExecute {
@Resource
@Qualifier("jdbcTemplate")
private JdbcTemplate jdbc;
@Resource
private DbPrintSql printSql;
@Resource
private ConfigDb configDb;
/**
* 更新SQL语句的执行
*
* @param targetClass 触发类
* @param sqlName SQL语句名称
* @param sql SQL语句
* @param paras 参数信息
* @return
*/
@Override
public int update(Class targetClass, String sqlName, String sql, Object... paras) {
int row = 0;
long start = System.currentTimeMillis();
try {
sql = this.handleParas(sql, paras);
row = jdbc.update(sql, paras);
return row;
} finally {
printSql.print(targetClass, sqlName, start, row, sql, paras);
}
}
/**
* 查询数据
*
* @param targetClass 触发类
* @param sqlName SQL语句名称
* @param cls 查询的结果的类型
* @param rowHandle 通过该类来处理结果
* @param sql 需要查询的SQL语句
* @param paras 查询语句的参数
* @param <T> 返回的集合的类型
* @return 集合
*/
@Override
public <T extends Object> void query(Class targetClass, Class<T> cls, DbRow<T> rowHandle, String sqlName, String sql, Object... paras) {
Ref<Integer> row = new Ref<Integer>(0);
long start = System.currentTimeMillis();
try {
sql = this.handleParas(sql, paras);
RowCallbackHandler rowCallbackHandler = new RowCallbackHandler() {
@Override
public void processRow(ResultSet rs) throws SQLException {
AllBeanRowMapper<T> rowMap = AllBeanRowMapper.getInstance(cls, configDb);
T data = rowMap.mapRow(rs, row.value);
rowHandle.handle(data);
row.value++;
}
};
jdbc.query(sql, rowCallbackHandler, paras);
} finally {
printSql.print(targetClass, sqlName, start, row.value, sql, paras);
}
}
/**
* 查询数据，并返回集合
*
* @param targetClass 触发类
* @param sqlName SQL语句名称
* @param cls 查询的结果的类型
* @param sql 需要查询的SQL语句
* @param paras 查询语句的参数
* @param <T> 返回的集合的类型
* @return 集合
*/
@Override
public <T extends Object> List<T> query(Class targetClass, Class<T> cls, String sqlName, String sql, Object... paras) {
int row = 0;
long start = System.currentTimeMillis();
try {
sql = this.handleParas(sql, paras);
List<T> ret = jdbc.query(sql, paras, AllBeanRowMapper.getInstance(cls, configDb));
if (ret == null) {
ret = new ArrayList<T>();
}
row = ret.size();
return ret;
} finally {
printSql.print(targetClass, sqlName, start, row, sql, paras);
}
}
/**
* 查询数据，并返回集合
*
* @param targetClass 触发类
* @param sqlName SQL语句名称
* @param sql 需要查询的SQL语句
* @param paras 查询语句的参数
* @return 集合
*/
@Override
public List<MapRow> query(Class targetClass, String sqlName, String sql, Object... paras) {
return query(targetClass, MapRow.class, sqlName, sql, paras);
}
/**
* 查询第一个单元格的信息
*
* @param targetClass 触发类
* @param sqlName SQL语句名称
* @param sql SQL语句
* @param paras 参数信息
* @return 第一个单元格的数据
*/
@Override
public Object queryCell(Class targetClass, String sqlName, String sql, Object... paras) {
int row = 0;
long start = System.currentTimeMillis();
try {
sql = this.handleParas(sql, paras);
SqlRowSet rowSet = jdbc.queryForRowSet(sql, paras);
while (rowSet.next()) {
row = 1;
return rowSet.getObject(1);
}
return null;
} finally {
printSql.print(targetClass, sqlName, start, row, sql, paras);
}
}
/**
* 处理SQL语句和参数值
*
* @param sql
* @param paras
* @return
*/
protected String handleParas(String sql, Object... paras) {
return sql.replaceAll("1\\s*?=\\s*?1\\s*?(?i)AND", "")
.replaceAll("(?i)WHERE\\s*?1\\s*?=\\s*?1", "")
.replaceAll("((?i)ORDER\\s*?(?i)BY\\s*?)1\\s*?,", "$1")
.replaceAll("(?i)ORDER\\s*?(?i)BY\\s*?1\\s*?", "")
.replaceAll("((?i)GROUP\\s*?(?i)BY\\s*?)1\\s*?,", "$1")
.replaceAll("(?i)GROUP\\s*?(?i)BY\\s*?1\\s*?", "");
}
}