Add examples using Virtual Threads.

Closes #665
This commit is contained in:
Mark Paluch
2023-10-27 11:01:21 +02:00
parent cdefadefdd
commit e90be0c65b
11 changed files with 366 additions and 4 deletions

4
redis/pubsub/README.md Normal file
View File

@@ -0,0 +1,4 @@
# Spring Data Redis Pub/Sub Example
This project contains samples of specific features of Spring Data Redis.

27
redis/pubsub/pom.xml Normal file
View File

@@ -0,0 +1,27 @@
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<artifactId>spring-data-redis-pubsub</artifactId>
<name>Spring Data Redis - Pub/Sub</name>
<parent>
<groupId>org.springframework.data.examples</groupId>
<artifactId>spring-data-redis-examples</artifactId>
<version>2.0.0.BUILD-SNAPSHOT</version>
<relativePath>../pom.xml</relativePath>
</parent>
<dependencies>
<dependency>
<groupId>${project.groupId}</groupId>
<artifactId>spring-data-redis-example-utils</artifactId>
<version>${project.version}</version>
<scope>test</scope>
</dependency>
</dependencies>
</project>

View File

@@ -0,0 +1,109 @@
/*
* Copyright 2023 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package example.springdata.redis;
import static org.assertj.core.api.Assertions.*;
import java.util.Collection;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.concurrent.TimeUnit;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.listener.ChannelTopic;
import org.springframework.data.redis.listener.RedisMessageListenerContainer;
import org.springframework.data.redis.listener.adapter.MessageListenerAdapter;
import org.springframework.data.redis.serializer.StringRedisSerializer;
/**
* Show usage of Redis Pub/Sub operations.
*
* @author Mark Paluch
*/
@SpringBootTest
public class PubSubTests {
@Autowired RedisConnectionFactory connectionFactory;
@Autowired StringRedisTemplate redisTemplate;
@Test
void shouldListenToPubSubEvents() throws Exception {
BlockingQueue<String> events = new LinkedBlockingDeque<>();
RedisMessageListenerContainer container = new RedisMessageListenerContainer();
container.setConnectionFactory(connectionFactory);
container.afterPropertiesSet();
container.addMessageListener(
(message, pattern) -> events.add(String.format("%s@%s", new String(message.getBody()), new String(pattern))),
ChannelTopic.of("my-channel"));
container.start();
redisTemplate.convertAndSend("my-channel", "Hello, world!");
String event = events.poll(5, TimeUnit.SECONDS);
container.stop();
container.destroy();
assertThat(event).isEqualTo("Hello, world!@my-channel");
}
@Test
void shouldNotifyListener() throws Exception {
BlockingQueue<String> events = new LinkedBlockingDeque<>();
RedisMessageListenerContainer container = new RedisMessageListenerContainer();
container.setConnectionFactory(connectionFactory);
container.afterPropertiesSet();
MessageListenerAdapter messageListenerAdapter = new MessageListenerAdapter(new MyListener(events));
messageListenerAdapter.afterPropertiesSet();
messageListenerAdapter.setSerializer(StringRedisSerializer.UTF_8);
container.addMessageListener(messageListenerAdapter, ChannelTopic.of("my-channel"));
container.start();
redisTemplate.convertAndSend("my-channel", "Hello, world!");
String event = events.poll(5, TimeUnit.SECONDS);
container.stop();
container.destroy();
assertThat(event).isEqualTo("Hello, world!@my-channel");
}
static class MyListener {
private final Collection<String> events;
public MyListener(Collection<String> events) {
this.events = events;
}
public void handleMessage(String message, String channel) {
events.add(String.format("%s@%s", message, channel));
}
}
}

View File

@@ -0,0 +1,76 @@
/*
* Copyright 2023 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package example.springdata.redis;
import static org.assertj.core.api.Assertions.*;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.concurrent.TimeUnit;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledOnJre;
import org.junit.jupiter.api.condition.JRE;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.core.task.AsyncTaskExecutor;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.listener.ChannelTopic;
import org.springframework.data.redis.listener.RedisMessageListenerContainer;
/**
* Show usage of Redis Pub/Sub operations using Virtual Threads.
*
* @author Mark Paluch
*/
@SpringBootTest(properties = "spring.threads.virtual.enabled=true")
@EnabledOnJre(JRE.JAVA_21)
public class PubSubVirtualThreadsTests {
@Autowired RedisConnectionFactory connectionFactory;
@Autowired AsyncTaskExecutor taskExecutor;
@Autowired StringRedisTemplate redisTemplate;
@Test
void shouldListenToPubSubEvents() throws Exception {
BlockingQueue<String> events = new LinkedBlockingDeque<>();
RedisMessageListenerContainer container = new RedisMessageListenerContainer();
container.setConnectionFactory(connectionFactory);
container.setTaskExecutor(taskExecutor);
container.afterPropertiesSet();
container.addMessageListener(
(message, pattern) -> events
.add(String.format("%s on Thread %s", new String(message.getBody()), Thread.currentThread())),
ChannelTopic.of("my-channel"));
container.start();
redisTemplate.convertAndSend("my-channel", "Hello, world!");
String event = events.poll(5, TimeUnit.SECONDS);
container.stop();
container.destroy();
assertThat(event).isNotNull().contains("Hello, world!").contains("VirtualThread");
}
}

View File

@@ -0,0 +1,24 @@
/*
* Copyright 2014-2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package example.springdata.redis;
import org.springframework.boot.autoconfigure.SpringBootApplication;
/**
* @author Mark Paluch
*/
@SpringBootApplication
public class RedisTestConfiguration {}

View File

@@ -0,0 +1 @@
logging.level.root=WARN