前言

上篇文章我們講到了怎么在Spring webFlux中使用r2dbc,今天我們看一下怎么使用spring-data-r2dbc這個(gè)Spring data對(duì)r2dbc的封裝來(lái)進(jìn)行r2dbc操作。

依賴關(guān)系

要使用Spring-datea-r2dbc需要配置下面的依賴關(guān)系：

<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>

數(shù)據(jù)庫(kù)連接配置

Spring-data-r2dbc使用的是R2dbcProperties來(lái)進(jìn)行配置文件的讀?。?/p>

簡(jiǎn)單看一下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;

相應(yīng)的，看一下我們的配置文件是怎么樣的：

spring.r2dbc.url=r2dbc:h2:mem://./testdbspring.r2dbc.username=saspring.r2dbc.password=password

這里，我們使用的是H2數(shù)據(jù)庫(kù)。

數(shù)據(jù)庫(kù)初始化

數(shù)據(jù)庫(kù)初始化過(guò)程中，我們需要?jiǎng)?chuàng)建一個(gè)users表格，這里我們?cè)谝粋€(gè)initDatabase bean中實(shí)現(xiàn)：

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

上面的例子中，我們創(chuàng)建數(shù)據(jù)庫(kù)分為了兩部分，第一部分是數(shù)據(jù)庫(kù)schema的操作，我們執(zhí)行了drop和create table的操作。

第二部分是向數(shù)據(jù)庫(kù)表中插入數(shù)據(jù)。

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

DAO操作

在DAO操作之前，我們需要?jiǎng)?chuàng)建Users entity：

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

再看一下我們創(chuàng)建的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注解，因?yàn)槲覀兝^承了ReactiveCrudRepository，會(huì)自動(dòng)幫我們創(chuàng)建UsersDao的實(shí)例，我們直接使用就可以了。

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

當(dāng)然，我們也可以自定義自己的SQL語(yǔ)句，比如上面的findByLastname。

Service操作和Transaction

我們看一下怎么使用UserDao來(lái)進(jìn)行具體的方法操作：

@Componentpublic class UsersService { @Resource private UsersDao usersDao; @Transactional public Mono<Users> save(Users user) { return usersDao.save(user).map(it -> { if (it.getFirstname().equals('flydean')) { throw new IllegalStateException(); } else { return it; } }); }}

上面我們創(chuàng)建了一個(gè)save方法，用來(lái)保存相應(yīng)的User對(duì)象。

controller

最后，我們創(chuàng)建一個(gè)controller來(lái)對(duì)外暴露相應(yīng)的方法：

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

好了，現(xiàn)在我們的程序已經(jīng)寫好了，可以進(jìn)行測(cè)試了。

運(yùn)行程序，執(zhí)行：

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

可以看到取出了相應(yīng)的結(jié)果。

完美，實(shí)驗(yàn)成功。

本文的代碼： spring-data-r2dbc

總結(jié)

到此這篇關(guān)于如何在Spring data中使用r2dbc的文章就介紹到這了,更多相關(guān)Spring data使用r2dbc內(nèi)容請(qǐng)搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！