在前面的准备工作完成之后,我们终于可以使用mybatis进行数据库操作啦。在第一篇中介绍了三张简单的表结构:user、project、step。鉴于数据表很简单并且很相似,在这里只以用户表user为例进行说明。
第一步,我们首先创建与user表对应的pojo类(Plain Ordinary Java Object),命名为User,并且让数据表中的每一列都与类中的实例变量相对应。根据user表中的列id、name、password、email、last_login,我们相应创建了User类,如下:
package com.loveqh.pojo;
/**
* Created by WL on 2017-04-23.
*/
public class User {
private int id;
private String name;
private String password;
private String email;
private String lastLogin;
}
注意到,由于java一般采用驼峰式命名法,数据表中带下划线的列名last_login对应的是lastLogin属性。
设计完User类的属性之后,我们还需要为每个属性设置setter和getter方法,可以使用intellij快捷键Alt+Insert自动添加。添加的代码如下:
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
public String getEmail() {
return email;
}
public void setEmail(String email) {
this.email = email;
}
public String getLastLogin() {
return lastLogin;
}
public void setLastLogin(String lastLogin) {
this.lastLogin = lastLogin;
}
为了方便调试中打印对象信息,我们再添加对象的toString方法,如下:
@Override
public String toString() {
return "User{" +
"id=" + id +
", name='" + name + '\'' +
", password='" + password + '\'' +
", email='" + email + '\'' +
", lastLogin='" + lastLogin + '\'' +
'}';
}
第二步,使用mybatis进行数据库操作,可以采用xxxMapper.xml及XxxDao接口的方式,即在XxxDao接口中定义需要的数据操作方法,然后在xxxMapper.xml中使用相应的sql语句实现。
对应地,我们在com.loveqh.dao包中创建UserDao接口,并在mybatis-mapper文件夹下创建userMapper.xml文件。
首先,我们想实现查询总用户数的操作,那么就可以在UserDao接口中添加相应方法:
package com.loveqh.dao;
import com.loveqh.pojo.User;
/**
* Created by WL on 2017-04-23.
*/
public interface UserDao {
int getAllUsersCount();
}
对应地,在userMapper.xml中添加相应的sql语句:
<mapper namespace="com.loveqh.dao.UserDao">
<select id="getAllUsersCount" resultType="int">
select count(*) from user
select>
<mapper>
在该文件中,我们注意两点:
1. mapper节点的namespace属性要对应UserDao接口。
2. select节点的id对应接口中的方法名,然后设置方法相应的参数类型及返回类型。
当然,这个mapper的xml文件需要在mybatis的配置文件中声明才能够使用。打开Configuration.xml文件,在environments节点后面添加以下节点:
<mappers>
<mapper resource="mybatis-mapper/userMapper.xml"/>
mappers>
为了测试该方法能否成功执行,我们可以在user数据表中先插入一条测试数据,然后使用junit进行单元测试。我们在com.loveqh.test包中新建UserTest测试类,以测试getAllUsersCount方法。代码如下:
package com.loveqh.test;
import com.loveqh.dao.UserDao;
import com.loveqh.pojo.User;
import com.loveqh.util.MyBatisSessionFactory;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
/**
* Created by WL on 2017-04-23.
*/
public class UserTest {
private SqlSession session;
private UserDao userDao;
@Before
public void setUp() {
SqlSessionFactory sessionFactory = MyBatisSessionFactory.getSessionFactory();
session = sessionFactory.openSession();
userDao = session.getMapper(UserDao.class);
}
@Test
public void getAllUsersCount() {
int count = userDao.getAllUsersCount();
Assert.assertEquals(count, 1);
}
@After
public void tearDown() {
session.close();
}
}
增加以下几点说明:
1. 使用mybatis访问数据库,我们需要根据数据库配置文件构造出SqlSessionFactory对象,然后得到SqlSession对象。MyBatisSessionFactory类是为了方便获取SqlSessionFactory对象而创建的工具类,代码稍后贴出。
2. 在使用时需要首先创建SqlSession对象,操作完毕之后就要进行关闭,因此这两步我们分别在@Before和@After中进行。
3. 对象userDao不是(也不能)直接由接口创建,而是通过session的getMapper方法构造出了的UserDao接口的实例。
4. 在@Test声明的测试方法中,我们调用UserDao接口中的getAllUsersCount方法,由于我们插入了一条测试数据,因此可以使用断言判断总用户数是否为1,如果是,则测试通过。
MyBatisSessionFactory类的代码如下:
package com.loveqh.util;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import java.io.InputStream;
/**
* Created by WL on 2017-04-23.
*/
public class MyBatisSessionFactory {
public static SqlSessionFactory getSessionFactory() {
SqlSessionFactory sessionFactory = null;
String resource = "Configuration.xml";
try {
InputStream inputStream = Resources.getResourceAsStream(resource);
sessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
} catch (Exception e) {
e.printStackTrace();
}
return sessionFactory;
}
}
在用户管理中,我们可能会需要根据用户的id来得到用户的详细信息,因此在UserDao类中需要一个findUserById方法,如下:
User findUserById(int id);
对应地,userMapper.xml中添加相应的sql语句实现:
<select id="findUserById" resultType="com.loveqh.pojo.User" parameterType="int">
select * from user where id=#{value}
select>
这里存在两个问题:
如果很多方法都是返回User对象,那么每个resultType都需要填写完整的包名路径,十分繁琐,也容易出错。不过,当然也有解决办法,就是使用mybatis提供的别名,我们可以再次打开Configuration.xml文件,在environments节点之前添加以下节点:
<typeAliases>
<typeAlias type="com.loveqh.pojo.User" alias="User">typeAlias>
typeAliases>
这样我们便可以把包路径替换成别名User。
前面提到过,由于java的驼峰式命名法,使得对象属性与数据表列名可能不一致,例如这里的lastLogin和last_login。这时,我们可以采用resultMap方式,将数据表的列名与类的属性名进行映射。打开userMapper.xml文件,添加以下节点:
<resultMap id="userResultMap" type="User">
<id property="id" column="id">id>
<result property="name" column="name">result>
<result property="password" column="password">result>
<result property="email" column="email">result>
<result property="lastLogin" column="last_login">result>
resultMap>
然后在select节点中使用resultMap替换掉resultType,修改如下:
<select id="findUserById" resultMap="userResultMap" parameterType="int">
select * from user where id=#{value}
select>
在项目中,我还增加了根据字符串进行模糊查询的操作,由于sql语句需要使用%通配符,所以在字符串拼接时不能使用#{},这里改用${},如下:
List findUsersLikeName(String name);
<select id="findUsersLikeName" resultMap="userResultMap" parameterType="String">
select * from user where name like '%${value}%'
select>
注意到这里返回类型仍然是User,当查询结果多于一条时,mybatis会自动帮我们保存到List中,是不是很方便?关于mybatis中#和$的区别,可以参考mybatis中的#和$的区别。
数据库的基本原子操作是CURD,即创建(Create)、更新(Update)、读取(Retrieve)和删除(Delete)操作。上面只介绍了读取操作,而其他三个操作也类似,只不过它们都会对数据库进行修改,因此在执行完操作后会返回影响的记录数,并且需要执行commit操作。示例如下:
@Test
public void deleteUserById() {
int id = 2;
int count = userDao.deleteUserById(id); //返回影响的记录数
session.commit(); //提交
Assert.assertEquals(count, 1);
findAllUsers();
}
具体的代码实现可以参考提交到github上的本项目,用户管理对应的文件分别是:
- User.java
- UserDao.java
- userMapper.xml
- UserTest.java
类似地,我们可以根据需求实现project和step的数据库逻辑。接下来,我们就可以安心进行界面实现了。