`
yunchow
  • 浏览: 319785 次
  • 性别: Icon_minigender_1
  • 来自: 南京
社区版块
存档分类
最新评论

数据库通用模板类实现

    博客分类:
  • DBI
阅读更多
package com.yunchow.util;

import java.util.*;
import java.sql.*;

/**
 * jdbc操作模板
 * @author yunchow
 * @version 1.2  09/8/5
 */
final public class JdbcTemplate {
	
	/**
	 * 统计指定表的记录条数
	 * @param tabName 表名
	 * @return 记录条数
	 */
	public int count(String tabName) {
		return count(tabName, null);
	}
	/**
	 * 按条件查询记录条数
	 * @param tabName 表名
	 * @param sqlwhere 条件
	 * @return 记录条数
	 */
	public int count(String tabName, String sqlwhere) {
		String sql = "select count(*) from " + tabName ;
		if(sqlwhere!=null && !"".equals(sqlwhere.trim())) {
			sql += " where " + sqlwhere;
		}
		return find(sql, new RowMapper<Integer>(){
				
				public Integer mapperRow(ResultSet rs) throws SQLException {
					return rs.getInt(1);
				}
				
			});
	}
	/**
	 * 执行批量更新
	 * @param sqls sql语句数组,其中每个sql子句只支持statement格式的,
	 *		  不支持preparedStatement格式,必须自已将参数拼装好再传过来.
	 * @return 所影响的行数
	 */
	public int[] updateBatch(String[] sqls) {
		Connection conn = null;
		Statement stmt = null;
		int[] rtn = null;
		try {
			conn = JdbcUtils.getConnection();
			stmt = conn.createStatement();
			conn.setAutoCommit(false);
			if(sqls!=null && sqls.length>0) {
				for(String sql : sqls)
				{
					stmt.addBatch(sql);
					//System.out.println(sql);
				}
			}
			rtn = stmt.executeBatch(); // 批量执行
			conn.commit();
			conn.setAutoCommit(true); // 此句必不可少,不然会引发死锁现象
		} catch(Exception ex) {
		   ex.printStackTrace();
			rtn = null;
			try{
				conn.rollback();
			} catch(SQLException e){
			  e.printStackTrace();
				// ignore the rollback exception
			}
			throw new RuntimeException(ex);
		} finally {
			JdbcUtils.close(stmt, conn);
		}
		return rtn;
	}
	/**
	 * 查询一个对象的集合
	 * @param sql sql语句,必须符合PreparedStatement格式
	 * @param params 为sql语句准备的参数,必须与sql中的?号一一对应
	 * @param mapper 行映射器,将结果集映射到特定的类
	 * @return T 查询结果的java bean风格类封装的集合
	 */
	public <T> List<T> findList(String sql, RowMapper<T> mapper, Object... params) {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		List<T> list = new ArrayList<T>();
		try {
			
			conn = JdbcUtils.getConnection();
			ps = conn.prepareStatement(sql);
			if(params!=null && params.length>0) {
				for(int i=0; i<params.length; i++){
					ps.setObject(i+1, params[i]);
				}
			}
			rs = ps.executeQuery();
			while(rs.next()) { 
				list.add(mapper.mapperRow(rs));
			}
		} catch(SQLException ex) {
			throw new RuntimeException(ex.getMessage(), ex);
		} finally {
			//System.out.println("JdbcTemplate: close()");
			JdbcUtils.close(rs, ps, conn);
		}
		return list;
	}
	/**
	 * 查询一个对象
	 * @param sql sql语句,必须符合PreparedStatement格式
	 * @param params 为sql语句准备的参数,必须与sql中的?号一一对应
	 * @param mapper 行映射器,将结果集映射到特定的类
	 * @return T 查询结果的java bean风格类封装
	 */
	public <T> T find(String sql, RowMapper<T> mapper, Object... params) {
		List<T> list = findList(sql, mapper, params);
		if(list!=null && list.size()>0){
			return list.get(0);
		}
		return null;
	}
	/**
	 * 支持所有更新操作 增,删,改
	 * @param sql 传过来的sql语句,其中的参数用?代替
	 * @param params 传过来的参数信息, 必须与sql中的问号相对应
	 * @return  更新的行数
	 */
	public int update(String sql, Object... params) {
		return update(sql, false, params);
	}
	/**
	 * 添加信息,并返回自动生成的主键
	 * @param sql 传过来的sql语句,其中的参数用?代替
	 * @param params 传过来的参数信息, 必须与sql中的问号相对应
	 * @return int 主键值
	 */
	public int save(String sql, Object... params) {
		return update(sql, true, params);
	}
	/**
	 * 支持所有更新操作,增,删,改
	 * @param sql 传过来的sql语句,其中的参数用?代替
	 * @param params 传过来的参数信息, 必须与sql中的问号相对应
	 * @param boo 为true获取自动生成的主键,否则忽略
	 * @return true 更新成功
	 */
	protected int update(String sql, Boolean boo, Object... params) {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		int rtn = -1;
		try {
			conn = JdbcUtils.getConnection();
			ps = boo?conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS):conn.prepareStatement(sql);
			if(params!=null && params.length>0) { //有参数传过来才设置参数
				for(int i=0; i<params.length; i++) {
					ps.setObject(i+1, params[i]);
				}
			}
			rtn = ps.executeUpdate();
			// 获取主键
			if(boo) {
				rs = ps.getGeneratedKeys();
				if(rs.next()) {
					rtn = rs.getInt(1);
				}
			}
		} catch(SQLException ex) {
			throw new RuntimeException(ex);
		} finally {
			JdbcUtils.close(ps, conn);
		}
		return rtn;
	}
	
	
} 
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics