上一篇深入Java日记——自己写一个ORM框架（1）中提到了如何将数据库的表生成对象的JavaBean,这篇博客就讲一下如何利用JavaBean增删改查，主要行为是拼接sql语句
1.增
首先要写一个最基础的执行sql语句的方法
我们使用的PreparedStatement来执行，添加参数可以通过一个工具类来实现
public class JDBCUtils {
public static void handleParams(PreparedStatement ps,Object[]params){
if (params!=null){
for (int i=0;i<params.length;i++){
try {
//参数是从1开始
ps.setObject(i+1,params[i]);
} catch (SQLException e) {
e.printStackTrace();
}
}
}
}
}
执行sql语句方法的主体
public int executeDML(String sql,Object[] params){
Connection connection=DBManager.getConn();
int count=0;
PreparedStatement ps=null;
try {
ps=connection.prepareStatement(sql);
JDBCUtils.handleParams(ps,params);
count=ps.executeUpdate();
} catch (SQLException e) {
e.printStackTrace();
}finally {
DBManager.close(ps,connection);
}
return count;
}
数据库的名字和字段的名字都是采用下划线分割法，但字段名是采用小驼峰，所以要进行一次转换
public static String camel2Underline(String line){
StringBuilder result = new StringBuilder();
if (line != null && line.length() > 0) {
// 将第一个字符处理成小写
result.append(line.substring(0, 1).toLowerCase());
// 循环处理其余字符
for (int i = 1; i < line.length(); i++) {
String s = line.substring(i, i + 1);
// 在大写字母前添加下划线
if (s.equals(s.toUpperCase()) && !Character.isDigit(s.charAt(0))) {
result.append("_");
}
// 其他字符直接转成大写
result.append(s.toLowerCase());
}
}
return result.toString();
}
插入时获取属性值主要通过反射来进行
public static Object getFieldValue(String field,Object obj){
Class clazz=obj.getClass();
Field idField= null;
Object id=null;
try {
idField = clazz.getDeclaredField(field);
idField.setAccessible(true);
id=idField.get(obj);
} catch (NoSuchFieldException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
}
return id;
}
插入方法的主要步骤是：拼接前置sql语句，转换和拼接属性名，拼接后置sql语句
/**
* 插入对象到数据库，拼接的sql如 insert into users(id,name) values(?,?)
* @param obj 要储存的数据
*/
public void insert(Object obj){
Class clazz = obj.getClass();
//通过Class获取TableInfo
TableInfo tableInfo = TableContext.poClassTableMap.get(clazz);
//拼接前置sql
StringBuilder stringBuilder=new StringBuilder("insert into " + tableInfo.getTname() + "(");
Field[]fields=clazz.getDeclaredFields();
Object[] fieldValue=new Object[fields.length];
//遍历所有属性，获取他们在要插入对象的值
for (int i=0;i<fields.length;i++){
Field field=fields[i];
fieldValue[i]=ReflectUtils.getFieldValue(field.getName(),obj);
stringBuilder.append(StringUtils.camel2Underline(field.getName())+",");
}
//删除最后一个逗号
stringBuilder.delete(stringBuilder.length()-1,stringBuilder.length());
//拼接后置sql
stringBuilder.append(")");
stringBuilder.append(" values(");
for (int i=0;i<fields.length;i++){
stringBuilder.append("?,");
}
stringBuilder.delete(stringBuilder.length()-1,stringBuilder.length());
stringBuilder.append(")");
String sql = stringBuilder.toString();
System.out.println(sql);
executeDML(sql, fieldValue);
}
2.删
删除我们采用两种方式，第一种是通过类和主键，另一种是通过对象实例删除
通过类和主键删除
public void delete(Class clazz,Object id){
//通过Class获取TableInfo
TableInfo tableInfo = TableContext.poClassTableMap.get(clazz);
//获得主键
ColumnInfo columnInfo = tableInfo.getOnlyPriKey();
String sql = "delete from " + tableInfo.getTname() + " where " + columnInfo.getName() + " =?";
executeDML(sql, new Object[]{id});
}
通过对象实例删除
public void delete(Object obj){
Class clazz = obj.getClass();
//通过Class获取TableInfo
TableInfo tableInfo = TableContext.poClassTableMap.get(clazz);
//获得主键
ColumnInfo columnInfo = tableInfo.getOnlyPriKey();
String sql = "delete from " + tableInfo.getTname() + " where " + columnInfo.getName() + " =?";
//通过反射获取主键
Object id = ReflectUtils.getFieldValue(StringUtils.underlineToSmallCamel(columnInfo.getName()), obj);
executeDML(sql, new Object[]{id});
}
3.改
改的sql拼接方式有点类似插入
/**
* 只更新指定字段 拼接的sql如 update users set name = ? where id = ?
* @param obj 所要更新的对象
* @param fieldNames 要更新的字段名
* @return 执行sql语句后影响的行数
*/
public int update(Object obj,String[] fieldNames){
Class clazz = obj.getClass();
//通过Class获取TableInfo
TableInfo tableInfo = TableContext.poClassTableMap.get(clazz);
//获得主键
ColumnInfo columnInfo = tableInfo.getOnlyPriKey();
List<Object> params=new ArrayList<>();
StringBuilder stringBuilder=new StringBuilder();
//拼接前置sql语句
stringBuilder.append("update " + tableInfo.getTname()+" set ");
//遍历要修改的属性，获取他们在要插入对象的值
for (String field:fieldNames){
stringBuilder.append(StringUtils.camel2Underline(field)+" = ?");
params.add(ReflectUtils.getFieldValue(field,obj));
stringBuilder.append(",");
}
//删除最后一个逗号
stringBuilder.delete(stringBuilder.length()-1,stringBuilder.length());
//拼接后置sql
stringBuilder.append(" where "+tableInfo.getOnlyPriKey().getName()+"=?");
String sql = stringBuilder.toString();
//通过反射获取主键
Object id = ReflectUtils.getFieldValue(StringUtils.underlineToSmallCamel(columnInfo.getName()), obj);
if (id==null){
return 0;
}
System.out.println(sql);
params.add(id);
return executeDML(sql, params.toArray());
}
4.查
查的话有三种查询，第一种是查询多行记录，第二种是查询某个记录，第三种是查询某个值
查询的时候要将记录转换为对应JavaBean的List，需要用到JavaBean的set方法，我们可以通过反射来进行
public static void invokeSet(String field,Object value,Object obj){
Class clazz=obj.getClass();
try {
Method method=clazz.getDeclaredMethod("set"+StringUtils.underlineToBigCamel(field),value.getClass());
method.invoke(obj,value);
} catch (NoSuchMethodException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (InvocationTargetException e) {
e.printStackTrace();
}
}
查询多行记录
/**
* 返回多行记录，并将每行记录封装到指定的类中
* @param sql 查询语句
* @param clazz 封装数据的javabean类的Class对象
* @param params sql的参数
* @return 查询到的结果
*/
public List queryRow(final String sql,final Class clazz,final Object[] params){
Connection connection=DBManager.getConn();
ResultSet resultSet=null;
try {
PreparedStatement ps=connection.prepareStatement(sql);
JDBCUtils.handleParams(ps,params);
resultSet=ps.executeQuery();
List result=new ArrayList();
ResultSetMetaData resultSetMetaData= null;
try {
resultSetMetaData = resultSet.getMetaData();
while (resultSet.next()){
Object rowObj=clazz.newInstance();
//获得遍历该行的字段,并将其赋值给对象
for(int i=0;i<resultSetMetaData.getColumnCount();i++){
//获取字段名
String columnName=resultSetMetaData.getColumnLabel(i+1);
//获取字段属性
Object columnValue=resultSet.getObject(i+1);
//赋值
ReflectUtils.invokeSet(columnName,columnValue,rowObj);
}
result.add(rowObj);
}
} catch (SQLException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (InstantiationException e) {
e.printStackTrace();
}finally {
DBManager.close(preparedStatement,connection);
}
return result;
} catch (SQLException e) {
e.printStackTrace();
return null;
}
}
查询某个记录
/**
* 返回一行记录，并将一行记录封装到指定的类中
* @param sql 查询语句
* @param clazz 封装数据的javabean类的Class对象
* @param params sql的参数
* @return 查询到的结果
*/
public Object queryUniqueRow(String sql,Class clazz,Object[] params){
List result=queryRow(sql,clazz,params);
if (result!=null&&result.size()>0){
return result.get(0);
}
return null;
}
查询某个值
/**
* 返回一个值（一行一列）
* @param sql 查询语句
* @param params sql的参数
* @return 查询到的结果
*/
public Object queryValue(String sql,Object[] params){
Connection connection=DBManager.getConn();
ResultSet resultSet=null;
try {
PreparedStatement ps=connection.prepareStatement(sql);
JDBCUtils.handleParams(ps,params);
resultSet=ps.executeQuery();
try {
while (resultSet.next()){
//只返回第一条记录的第一个字段的值
return resultSet.getObject(1);
}
} catch (SQLException e) {
e.printStackTrace();
}finally {
DBManager.close(preparedStatement,connection);
}
return null;
}
}
以上是核心代码，之后我们还可以自定义一个连接池和自定义Query对象的工厂来适配不同数据库，这部分的代码留给你们去实现
网友评论