首页 > 编程语言 > Mybatis如何使用注解优化代码
2020
11-03

Mybatis如何使用注解优化代码

entity层代码不变,因为是优化,所以在dao层新增了一个通过id查询用户的功能,现在来演示一下优化前后的代码对比,希望你们喜欢

entity层代码不变,dao层中的UserMapper中添加一个方法

/**
* 根据id查询用户
*/
User findUserById(int id);

在UserMapper.xml映射文件中添加

  <select id="findUserById" parameterType="int" resultType="com.vg.entity.User">
    select * from user where id = #{id};
  </select>


在test层中的test中使用没有优化的代码展示

package com.vg.test;



import com.vg.dao.UserMapper;
import com.vg.entity.User;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.Test;

import java.io.InputStream;
import java.util.List;

/**
 * 测试类
 */
public class TestUserMapper {

	/**
	 * 测试方法:查询所有学生
	 * SqlSessionFactoryBuilder => SqlSessionFactory => SqlSession:相当于JDBC中的Connection对象,提供了操作数据库的CRUD方法
	 */
	@Test
	public void testFindAllUsers()throws Exception{
		// 1. 通过Resources类,加载sqlMapConfig.xml,得到文件输入流对象
		// Resources类作用:专门读取src目录下的资源文件
		InputStream in = Resources.getResourceAsStream("sqlMapConfig.xml");
		// 2. 实例化会话工厂创建类SqlSessionFactoryBuilder
		SqlSessionFactoryBuilder builder = new SqlSessionFactoryBuilder();
		// 3. 通过会话工厂创建类,读取上面的输入流,得到会话工厂SqlSessionFactory类
		SqlSessionFactory sqlSessionFactory = builder.build(in);
		// 4. 使用SqlSessionFactory对象,创建SqlSession对象
		// a) 它相当于JDBC中的Connection对象,提供了操作数据库的CRUD方法
		// b) 它提供了一个getMapper()方法,获取接口的实现对象。
		SqlSession sqlSession = sqlSessionFactory.openSession();

		// 5. 获取接口 UserMapper 实现类对象(代理对象)
		UserMapper userMapper = sqlSession.getMapper(UserMapper.class);
		System.out.println("userMapper = " +userMapper);
		// 6. 执行数据库的查询操作,输出用户信息
		List<User> userList = userMapper.findAllUsers();
		for (User user : userList) {
			System.out.println(user);
		}
		// 7. 关闭会话,释放资源。
		sqlSession.close();
	}
	@Test
	public void testFindUserById()throws Exception{
		// 1. 通过Resources类,加载sqlMapConfig.xml,得到文件输入流对象
		// Resources类作用:专门读取src目录下的资源文件
		InputStream in = Resources.getResourceAsStream("sqlMapConfig.xml");
		// 2. 实例化会话工厂创建类SqlSessionFactoryBuilder
		SqlSessionFactoryBuilder builder = new SqlSessionFactoryBuilder();
		// 3. 通过会话工厂创建类,读取上面的输入流,得到会话工厂SqlSessionFactory类
		SqlSessionFactory sqlSessionFactory = builder.build(in);
		// 4. 使用SqlSessionFactory对象,创建SqlSession对象
		// a) 它相当于JDBC中的Connection对象,提供了操作数据库的CRUD方法
		// b) 它提供了一个getMapper()方法,获取接口的实现对象。
		SqlSession sqlSession = sqlSessionFactory.openSession();

		// 5. 获取接口 UserMapper 实现类对象(代理对象)
		UserMapper userMapper = sqlSession.getMapper(UserMapper.class);
		System.out.println("userMapper = " +userMapper);
		// 6. 调用接口的方法根据id查询用户
		User user = userMapper.findUserById(1);
		System.out.println("user = " + user);
		// 7. 关闭会话,释放资源。
		sqlSession.close();
	}
}

你会发现在这俩个方法中有很多相似的代码,每次写都要重新写,关键是还没有啥软用,现在来看看优化之后的代码

package com.vg.test;

import com.vg.dao.UserMapper;
import com.vg.entity.User;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;

/**
 * 测试类
 */
public class TestUserMapper {
	private static SqlSessionFactory sqlSessionFactory =null;
	private SqlSession sqlSession =null;
	private UserMapper userMapper =null;

	// 该方法在所有测试方法执行之前执行1次
	@BeforeClass
	public static void init() throws Exception {
		// 1. 通过Resources类,加载sqlMapConfig.xml,得到文件输入流对
		InputStream in = Resources.getResourceAsStream("sqlMapConfig.xml");
		// 2. 实例化会话工厂创建类SqlSessionFactoryBuilder对象
		SqlSessionFactoryBuilder builder = new SqlSessionFactoryBuilder();
		// 3. 根据字节输入流获取SqlSessionFactory对象
		sqlSessionFactory = builder.build(in);
	}
	// 该方法会在每个测试方法执行之前执行1次
	@Before
	public void before(){
		// 4. 获取SqlSession对象,等价连接对象
		// true:事务自动提交,false:不自动提交,默认值
		sqlSession = sqlSessionFactory.openSession();
		// 5. 创建接口 实现类对象
		 userMapper = sqlSession.getMapper(UserMapper.class);

	}

	/**
	 * 测试方法:查询所有学生
	 * SqlSessionFactoryBuilder => SqlSessionFactory => SqlSession:相当于JDBC中的Connection对象,提供了操作数据库的CRUD方法
	 */
	@Test
	public void testFindAllUsers() throws Exception{
		List<User> userList = userMapper.findAllUsers();
		for (User user : userList) {
			System.out.println(user);
		}
	}

	@Test
	public void testFindUserById()throws Exception{
		// 6. 调用接口的方法根据id查询用户
		User user = userMapper.findUserById(1);
		System.out.println("user = " + user);

	}

	@After
	public void after() {
		// 7. 关闭会话,释放资源。

		sqlSession.close();
	}
}

这样子是不是比以前的代码简单起来了呢?

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持自学编程网。

编程技巧