Mybatis如何使用注解优化代码

Z时代
2024-01-10
分类：综合

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();
	}
}