基础使用
# Spring JDBC
Spring JDBC 框架提供了多种访问数据库的方法,其中最著名的就是使用JdbcTemplate
这个类。这也是主要的用于管理数据库连接与异常处理的类。要使用 Spring JDBC 的话,首先需要在 pom.xml 文件中配置依赖项:
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-context</artifactId>
<version>${spring.version}</version>
</dependency>
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>5.1.26</version>
</dependency>
基础使用
数据源配置
在我们访问数据库的时候,需要先配置一个数据源,下面分别介绍一下几种不同的数据库配置方式。首先,为了连接数据库需要引入 jdbc 支持,在 pom.xml 中引入如下配置:
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>
嵌入式数据库通常用于开发和测试环境,不推荐用于生产环境。Spring Boot 提供自动配置的嵌入式数据库有 H2、HSQL、Derby,你不需要提供任何连接配置就能使用。比如,我们可以在 pom.xml 中引入如下配置使用 HSQL:
<dependency>
<groupId>org.hsqldb</groupId>
<artifactId>hsqldb</artifactId>
<scope>runtime</scope>
</dependency>
以 MySQL 数据库为例,先引入 MySQL 连接的依赖包,在 pom.xml 中加入:
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
</dependency>
在 src/main/resources/application.properties 中配置数据源信息:
spring.datasource.url=jdbc:mysql://localhost:3306/test
spring.datasource.username=dbuser
spring.datasource.password=dbpass
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
注意:因为 Spring Boot 2.1.x 默认使用了 MySQL 8.0 的驱动,所以这里采用 com.mysql.cj.jdbc.Driver,而不是老的 com.mysql.jdbc.Driver。
使用 JdbcTemplate 操作数据库
Spring 的 JdbcTemplate 是自动配置的,你可以直接使用@Autowired 或构造函数(推荐)来注入到你自己的 bean 中来使用。先创建 User 表,包含属性 name、age。可以通过执行下面的建表语句:
CREATE TABLE `User` (
`name` varchar(100) COLLATE utf8mb4_general_ci NOT NULL,
`age` int NOT NULL
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci
根据数据库中创建的 User 表,创建领域对象:
@Data
@NoArgsConstructor
public class User {
private String name;
private Integer age;
}
然后定义包含有插入、删除、查询的抽象接口 UserService:
public interface UserService {
/**
* 新增一个用户
*
* @param name
* @param age
*/
int create(String name, Integer age);
/**
* 根据name查询用户
*
* @param name
* @return
*/
List<User> getByName(String name);
/**
* 根据name删除用户
*
* @param name
*/
int deleteByName(String name);
/**
* 获取用户总量
*/
int getAllUsers();
/**
* 删除所有用户
*/
int deleteAllUsers();
}
最后通过 JdbcTemplate 实现 UserService 中定义的数据访问操作:
@Service
public class UserServiceImpl implements UserService {
private JdbcTemplate jdbcTemplate;
UserServiceImpl(JdbcTemplate jdbcTemplate) {
this.jdbcTemplate = jdbcTemplate;
}
@Override
public int create(String name, Integer age) {
return jdbcTemplate.update("insert into USER(NAME, AGE) values(?, ?)", name, age);
}
@Override
public List<User> getByName(String name) {
List<User> users = jdbcTemplate.query("select NAME, AGE from USER where NAME = ?", (resultSet, i) -> {
User user = new User();
user.setName(resultSet.getString("NAME"));
user.setAge(resultSet.getInt("AGE"));
return user;
}, name);
return users;
}
@Override
public int deleteByName(String name) {
return jdbcTemplate.update("delete from USER where NAME = ?", name);
}
@Override
public int getAllUsers() {
return jdbcTemplate.queryForObject("select count(1) from USER", Integer.class);
}
@Override
public int deleteAllUsers() {
return jdbcTemplate.update("delete from USER");
}
}
单元测试
创建对 UserService 的单元测试用例,通过创建、删除和查询来验证数据库操作的正确性。
@RunWith(SpringRunner.class)
@SpringBootTest
public class Chapter31ApplicationTests {
@Autowired
private UserService userSerivce;
@Before
public void setUp() {
// 准备,清空user表
userSerivce.deleteAllUsers();
}
@Test
public void test() throws Exception {
// 插入5个用户
userSerivce.create("Tom", 10);
userSerivce.create("Mike", 11);
userSerivce.create("Didispace", 30);
userSerivce.create("Oscar", 21);
userSerivce.create("Linda", 17);
// 查询名为Oscar的用户,判断年龄是否匹配
List<User> userList = userSerivce.getByName("Oscar");
Assert.assertEquals(21, userList.get(0).getAge().intValue());
// 查数据库,应该有5个用户
Assert.assertEquals(5, userSerivce.getAllUsers());
// 删除两个用户
userSerivce.deleteByName("Tom");
userSerivce.deleteByName("Mike");
// 查数据库,应该有5个用户
Assert.assertEquals(3, userSerivce.getAllUsers());
}
}
数据查询
@SuppressWarnings({ "unchecked", "rawtypes" })
public Employee findById(int id){
String sql = "SELECT * FROM EMPLOYEE WHERE ID = ?";
jdbcTemplate = new JdbcTemplate(dataSource);
Employee employee = (Employee) jdbcTemplate.queryForObject(
sql, new Object[] { id }, new BeanPropertyRowMapper(Employee.class));
return employee;
}
在 query 中,最后需要传入一个继承自 RowMapper 的实现类,有时候方便起见,也可以直接传入一个 Entity。如果是采用的 RowMapper 模式,需要实现如下的映射器类:
@SuppressWarnings("rawtypes")
public class EmployeeRowMapper implements RowMapper {
public Object mapRow(ResultSet rs, int rowNum) throws SQLException {
Employee employee = new Employee();
employee.setId(rs.getInt("ID"));
employee.setName(rs.getString("NAME"));
employee.setAge(rs.getInt("AGE"));
return employee;
}
}
最后在调用的时候,把映射器作为最后一个参数传入:
Employee employee = (Employee) jdbcTemplate.queryForObject(sql, new Object[] { id }, new EmployeeRowMapper());
插入与修改
Insert
public void insert(Employee employee) {
String sql = "INSERT INTO EMPLOYEE " +
"(ID, NAME, AGE) VALUES (?, ?, ?)";
jdbcTemplate = new JdbcTemplate(dataSource);
jdbcTemplate.update(sql, new Object[] {
employee.getId(),
employee.getName(), employee.getAge()
});
}
有时候需要在插入之后,将插入行自动生成的主键返回,可以使用 jdbcTemplate 中提供的 KeyHolder 来实现:
public class ExampleDao {
@Autowired
private JdbcTemplate jdbcTemplate;
public long addNew(final String name) {
final PreparedStatementCreator psc = new PreparedStatementCreator() {
@Override
public PreparedStatement createPreparedStatement(final Connection connection) throws SQLException {
final PreparedStatement ps = connection.prepareStatement("INSERT INTO `names` (`name`) VALUES (?)",
Statement.RETURN_GENERATED_KEYS);
ps.setString(1, name);
return ps;
}
};
// The newly generated key will be saved in this object
final KeyHolder holder = new GeneratedKeyHolder();
jdbcTemplate.update(psc, holder);
final long newNameId = holder.getKey().longValue();
return newNameId;
}
}