mybatis+spring的TODO小项目记录(六)使用mybatis进行数据库操作

来源:互联网 发布:淘宝pc端优惠券地址 编辑:程序博客网 时间:2024/06/06 13:22

在前面的准备工作完成之后,我们终于可以使用mybatis进行数据库操作啦。在第一篇中介绍了三张简单的表结构:userprojectstep。鉴于数据表很简单并且很相似,在这里只以用户表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类的属性之后,我们还需要为每个属性设置settergetter方法,可以使用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.xmlXxxDao接口的方式,即在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语句:

<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"><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不是(也不能)直接由接口创建,而是通过sessiongetMapper方法构造出了的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>

这里存在两个问题:

  1. 如果很多方法都是返回User对象,那么每个resultType都需要填写完整的包名路径,十分繁琐,也容易出错。不过,当然也有解决办法,就是使用mybatis提供的别名,我们可以再次打开Configuration.xml文件,在environments节点之前添加以下节点:

    <typeAliases>        <typeAlias type="com.loveqh.pojo.User" alias="User"></typeAlias></typeAliases>

    这样我们便可以把包路径替换成别名User

  2. 前面提到过,由于java的驼峰式命名法,使得对象属性与数据表列名可能不一致,例如这里的lastLoginlast_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<User> 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的数据库逻辑。接下来,我们就可以安心进行界面实现了。

0 0