模仿QueryRunner的底层实现

最新推荐文章于 2022-05-14 20:46:53 发布

原创最新推荐文章于 2022-05-14 20:46:53 发布 · 519 阅读

0 ·

CC 4.0 BY-SA版权

java基础专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一个名为MyQueryRunner的实用工具类，该类用于简化Java应用程序中的数据库操作。通过使用MyQueryRunner，开发者可以方便地执行SQL查询和更新操作，并通过自定义的结果集处理器将查询结果转换为特定类型的对象。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package cn.xiechengxu.dbutils;

import javax.sql.DataSource;
import java.sql.*;

/**
 * Created by zhangqiang on 17/6/23.
 */
public class MyQueryRunner {

    private DataSource ds;

    public MyQueryRunner() {
    }

    public MyQueryRunner(DataSource ds){

        this.ds=ds;
    }



    public <T> T query(Connection con,String sql ,MyResultSetHandler<T> mrs, Object ...params) throws SQLException {

        PreparedStatement pst=con.prepareStatement(sql);

        ParameterMetaData pmd=pst.getParameterMetaData();

        int count=pmd.getParameterCount();


        for(int i=1;i<=count;i++){
            pst.setObject(i,params[i-1]);
        }


        ResultSet rs=pst.executeQuery();//得到结果集，要将结果集封装成用户想要的对象，但是，工具不知道用户需求

        return mrs.handle(rs);
    }


    public int update(Connection con,String sql , Object ...params) throws SQLException {

        PreparedStatement pst=con.prepareStatement(sql);

        ParameterMetaData pmd=pst.getParameterMetaData();

        int count=pmd.getParameterCount();


        for(int i=1;i<=count;i++){
            pst.setObject(i,params[i-1]);
        }

        int row= pst.executeUpdate();

        //关闭资源
        pst.close();

        return row;

    }




    public int update(String sql , Object ...params) throws SQLException {


        Connection con=ds.getConnection();

        PreparedStatement pst=con.prepareStatement(sql);

        ParameterMetaData pmd=pst.getParameterMetaData();

        int count=pmd.getParameterCount();


        for(int i=1;i<=count;i++){
            pst.setObject(i,params[i-1]);
        }

        int row= pst.executeUpdate();

        //关闭资源
        pst.close();
        con.close();
        return row;

    }

}


package cn.xiechengxu.dbutils.domain;

import cn.xiechengxu.dbutils.MyQueryRunner;
import cn.xiechengxu.dbutils.MyResultSetHandler;
import cn.xiechengxu.utils.DataSourceUtils;
import org.junit.Test;

import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * Created by zhangqiang on 17/6/23.
 */
public class MyQueryRunnerTest {


    @Test
    public void updateTest() throws SQLException{

        String sql="delete from account where id=?";

        MyQueryRunner mqr=new MyQueryRunner();

        mqr.update(DataSourceUtils.getConnection(),sql,3);

    }


    @Test
    public void selectTest() throws SQLException {

        String sql="select * from account where id=?";

        MyQueryRunner mqr=new MyQueryRunner();

       Account a=mqr.query(DataSourceUtils.getConnection(), sql, new MyResultSetHandler<Account>() {
            @Override
            public Account handle(ResultSet rs) throws SQLException {
                Account a=null;

                if(rs.next()){
                    a=new Account();
                    a.setId(rs.getInt("id"));
                    a.setName(rs.getString("name"));
                    a.setMoney(rs.getDouble("money"));
                }

                return a;
            }
        }, 2);



       System.out.print(a);
    }
}

public interface MyResultSetHandler<T> {

    public T handle(ResultSet rs)throws SQLException;
}