文章目錄

• 簡介
• 依賴關系
• 數據庫連接配置
• 數據庫初始化
• DAO操作
• Service操作和Transaction
• controller
• 測試

簡介

上篇文章我們講到了怎么在Spring webFlux中使用r2dbc,今天我們看一下怎么使用spring-data-r2dbc這個Spring data對r2dbc的封裝來進行r2dbc操作。

依賴關系

要使用Spring-datea-r2dbc需要配置下面的依賴關系：

<dependencies><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-r2dbc</artifactId><version>2.3.3.RELEASE</version></dependency><!-- R2DBC H2 Driver --><dependency><groupId>io.r2dbc</groupId><artifactId>r2dbc-h2</artifactId><version>${r2dbc-h2.version}</version></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-webflux</artifactId></dependency>

數據庫連接配置

Spring-data-r2dbc使用的是R2dbcProperties來進行配置文件的讀取：

簡單看一下R2dbcProperties的定義：

@ConfigurationProperties(prefix = "spring.r2dbc") public class R2dbcProperties {/*** Database name. Set if no name is specified in the url. Default to "testdb" when* using an embedded database.*/private String name;/*** Whether to generate a random database name. Ignore any configured name when* enabled.*/private boolean generateUniqueName;/*** R2DBC URL of the database. database name, username, password and pooling options* specified in the url take precedence over individual options.*/private String url;/*** Login username of the database. Set if no username is specified in the url.*/private String username;/*** Login password of the database. Set if no password is specified in the url.*/private String password;

相應的，看一下我們的配置文件是怎么樣的：

spring.r2dbc.url=r2dbc:h2:mem://./testdb spring.r2dbc.username=sa spring.r2dbc.password=password

這里，我們使用的是H2數據庫。

數據庫初始化

數據庫初始化過程中，我們需要創建一個users表格，這里我們在一個initDatabase bean中實現：

@Beanpublic ApplicationRunner initDatabase(DatabaseClient client, UsersDao usersDao) {List<String> statements = Arrays.asList("DROP TABLE IF EXISTS USERS;","CREATE TABLE IF NOT EXISTS USERS ( id SERIAL PRIMARY KEY, firstname VARCHAR(100) NOT NULL, lastname VARCHAR(100) NOT NULL);");statements.forEach(sql -> executeSql(client,sql).doOnSuccess(count -> log.info("Schema created, rows updated: {}", count)).doOnError(error -> log.error("got error : {}",error.getMessage(),error)).subscribe());return args ->getUser().flatMap(usersDao::save).subscribe(user -> log.info("User saved: {}", user));}private Flux<Users> getUser() {return Flux.just(new Users(null, "John", "Doe"), new Users(null, "Jane", "Doe"));}private Mono<Integer> executeSql(DatabaseClient client, String sql) {return client.execute(sql).fetch().rowsUpdated();}

上面的例子中，我們創建數據庫分為了兩部分，第一部分是數據庫schema的操作，我們執行了drop和create table的操作。

第二部分是向數據庫表中插入數據。

注意，上面的兩部分操作都需要執行subscribe操作，以觸發真正的Reactive操作。

DAO操作

在DAO操作之前，我們需要創建Users entity：

@Data @AllArgsConstructor public class Users {@Idprivate Integer id;private String firstname;private String lastname;boolean hasId() {return id != null;} }

再看一下我們創建的UserDao：

public interface UsersDao extends ReactiveCrudRepository<Users, Long> {@Query("select id, firstname, lastname from users c where c.lastname = :lastname")Flux<Users> findByLastname(String lastname); }

注意，這里并不需要添加@Component注解，因為我們繼承了ReactiveCrudRepository，會自動幫我們創建UsersDao的實例，我們直接使用就可以了。

ReactiveCrudRepository為我們封裝了一些DAO的基本操作，比如save，saveAll, findById ,existsById等基礎操作。

當然，我們也可以自定義自己的SQL語句，比如上面的findByLastname。

Service操作和Transaction

我們看一下怎么使用UserDao來進行具體的方法操作：

@Component public class UsersService {@Resourceprivate UsersDao usersDao;@Transactionalpublic Mono<Users> save(Users user) {return usersDao.save(user).map(it -> {if (it.getFirstname().equals("flydean")) {throw new IllegalStateException();} else {return it;}});}}

上面我們創建了一個save方法，用來保存相應的User對象。

controller

最后，我們創建一個controller來對外暴露相應的方法：

@RestController @RequiredArgsConstructor public class UsersController {private final UsersDao usersDao;@GetMapping("/users")public Flux<Users> findAll() {return usersDao.findAll();} }

測試

好了，現在我們的程序已經寫好了，可以進行測試了。

運行程序，執行：

curl "localhost:8080/users" [{"id":1,"firstname":"John","lastname":"Doe"},{"id":2,"firstname":"Jane","lastname":"Doe"}]%

可以看到取出了相應的結果。

完美，實驗成功。

本文的代碼：spring-data-r2dbc

本文作者：flydean程序那些事

本文鏈接：http://www.flydean.com/spring-data-r2dbc/

本文來源：flydean的博客

歡迎關注我的公眾號:「程序那些事」最通俗的解讀，最深刻的干貨，最簡潔的教程，眾多你不知道的小技巧等你來發現！

總結

以上是生活随笔為你收集整理的在Spring data中使用r2dbc的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。