MyBatis直接执行SQL的工具SqlMapper

jopen 4年前

可能有些人也有过类似需求,一般都会选择使用其他的方式如Spring-JDBC等方式解决。

能否通过MyBatis实现这样的功能呢?

为了让通用Mapper更彻底的支持多表操作以及更灵活的操作,在2.2.0版本增加了一个可以直接执行SQL的新类SqlMapper。

通过这篇博客,我们来了解一下SqlMapper

SqlMapper提供的方法

SqlMapper提供了以下这些公共方法:

  • Map<String,Object> selectOne(String sql)

  • Map<String,Object> selectOne(String sql, Object value)

  • <T> T selectOne(String sql, Class<T> resultType)

  • <T> T selectOne(String sql, Object value, Class<T> resultType)

  • List<Map<String,Object>> selectList(String sql)

  • List<Map<String,Object>> selectList(String sql, Object value)

  • <T> List<T> selectList(String sql, Class<T> resultType)

  • <T> List<T> selectList(String sql, Object value, Class<T> resultType)

  • int insert(String sql)

  • int insert(String sql, Object value)

  • int update(String sql)

  • int update(String sql, Object value)

  • int delete(String sql)

  • int delete(String sql, Object value)

一共14个方法,这些方法的命名和参数和SqlSession接口的很像,只是基本上第一个参数都成了sql。

其中Object value为入参,入参形式和SqlSession中的入参一样,带有入参的方法,在使用时sql可以包含#{param}或${param}形式的参数,这些参数需要通过入参来传值。需要的参数过多的时候,参数可以使用Map类型。另外这种情况下的sql还支持下面这种复杂形式:

String sql = "<script>select * from sys_user where 1=1"  +           "<if test=\"usertype != null\">usertype = #{usertype}</if></script>";

这种情况用的比较少,不多说。

不带有Object value的所有方法,sql中如果有参数需要手动拼接成一个可以直接执行的sql语句。

在selectXXX方法中,使用Class<T> resultType可以指定返回类型,否则就是Map<String,Object>类型。

实例化SqlMapper

SqlMapper构造参数public SqlMapper(SqlSession sqlSession),需要一个入参SqlSession sqlSession,在一般系统中,可以按照下面的方式获取:

SqlSession sqlSession = (...);//通过某些方法获取sqlSession  //创建sqlMapper  SqlMapper sqlMapper = new SqlMapper(sqlSession);

如果使用的Spring,那么可以按照下面的方式配置<bean>:

<bean id="sqlMapper" class="com.github.abel533.sql.SqlMapper" scope="prototype">    <constructor-arg ref="sqlSession"/>  </bean>

在Service中使用的时候可以直接使用@Autowired注入。

简单例子

在src/test/java目录的com.github.abel533.sql包中包含这些方法的测试。

下面挑几个看看如何使用。

selectList

//查询,返回List<Map>  List<Map<String, Object>> list = sqlMapper.selectList("select * from country where id < 11");    //查询,返回指定的实体类  List<Country> countryList = sqlMapper.selectList("select * from country where id < 11", Country.class);    //查询,带参数  countryList = sqlMapper.selectList("select * from country where id < #{id}", 11, Country.class);    //复杂点的查询,这里参数和上面不同的地方,在于传入了一个对象  Country country = new Country();  country.setId(11);  countryList = sqlMapper.selectList("<script>" +          "select * from country " +          "   <where>" +          "       <if test=\"id != null\">" +          "           id < #{id}" +          "       </if>" +          "   </where>" +          "</script>", country, Country.class);

selectOne

Map<String, Object> map = sqlMapper.selectOne("select * from country where id = 35");    map = sqlMapper.selectOne("select * from country where id = #{id}", 35);    Country country = sqlMapper.selectOne("select * from country where id = 35", Country.class);    country = sqlMapper.selectOne("select * from country where id = #{id}", 35, Country.class);

insert,update,delete

//insert  int result = sqlMapper.insert("insert into country values(1921,'天朝','TC')");    Country tc = new Country();  tc.setId(1921);  tc.setCountryname("天朝");  tc.setCountrycode("TC");  //注意这里的countrycode和countryname故意写反的  result = sqlMapper.insert("insert into country values(#{id},#{countrycode},#{countryname})"                            , tc);      //update  result = sqlMapper.update("update country set countryname = '天朝' where id = 35");    tc = new Country();  tc.setId(35);  tc.setCountryname("天朝");    int result = sqlMapper.update("update country set countryname = #{countryname}" +              " where id in(select id from country where countryname like 'A%')", tc);      //delete  result = sqlMapper.delete("delete from country where id = 35");  result = sqlMapper.delete("delete from country where id = #{id}", 35);

注意

通过上面这些例子应该能对此有个基本的了解,但是如果你使用参数方式,建议阅读下面的文章:

深入了解MyBatis参数

实现原理

2015-03-09:最初想要设计这个功能的时候,感觉会很复杂,想的也复杂,需要很多个类,因此当时没有实现。

2015-03-10:突发奇想,设计了现在的这种方式。并且有种强烈的感觉就是幸好昨天没有尝试去实现,因为昨天晚上思考这个问题的时候是晚上10点多,而今天(10号)是7点开始思考。我很庆幸在一个更清醒的状态下去写这段代码。

下面简单说思路和实现方式。

在写MyBatis分页插件的时候熟悉了MappedStatement类。

在写通用Mapper的时候熟悉了xml转SqlNode结构。

如果我根据SQL动态的创建一个MappedStatement,然后使用MappedStatement的id在sqlSession中执行不就可以了吗?

想到这一点,一切就简单了。

看看下面select查询创建MappedStatement的代码:

/**   * 创建一个查询的MS   *   * @param msId   * @param sqlSource 执行的sqlSource   * @param resultType 返回的结果类型   */  private void newSelectMappedStatement(String msId, SqlSource sqlSource, final Class<?> resultType) {      MappedStatement ms = new MappedStatement.Builder(              configuration, msId, sqlSource, SqlCommandType.SELECT)          .resultMaps(new ArrayList<ResultMap>() {              {                  add(new ResultMap.Builder(configuration,                          "defaultResultMap",                          resultType,                          new ArrayList<ResultMapping>(0)).build());              }          })          .build();      //缓存      configuration.addMappedStatement(ms);  }

代码是不是很简单,这段代码的关键是参数sqlSource,下面是创建SqlSource的方法,分为两种。

一种是一个完整的sql,不需要参数的,可以直接执行的:

StaticSqlSource sqlSource = new StaticSqlSource(configuration, sql);

其中configuration从sqlSession中获取,sql就是用户传入到sql语句,是不是也很简单?

另一种是支持动态sql的,支持参数的SqlSource:

SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, parameterType);

是不是也很简单?这个方法其实可以兼容上面的StaticSqlSource,这里比上面多了一个parameterType,因为这儿是可以传递参数的,另外languageDriver是从configuration中获取的。

是不是很简单?

我一开始也没想到MyBatis直接执行sql实现起来会这么的容易。

insert,delete,update方法的创建更容易,因为他们的返回值都是int,所以处理起来更简单,有兴趣的可以去通用Mapper下的包com.github.abel533.sql中查看SqlMapper的源码。