Skip to content

Commit

Permalink
Merge branch 'redis:doctests' into doctests
Browse files Browse the repository at this point in the history
  • Loading branch information
tishun authored Nov 21, 2024
2 parents d616b16 + bf4c683 commit 2c7fd42
Show file tree
Hide file tree
Showing 2 changed files with 251 additions and 0 deletions.
130 changes: 130 additions & 0 deletions src/test/java/io/redis/examples/async/StringExample.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,130 @@
// EXAMPLE: set_tutorial
package io.redis.examples.async;

import io.lettuce.core.*;
import io.lettuce.core.api.async.RedisAsyncCommands;
import io.lettuce.core.api.StatefulRedisConnection;

// REMOVE_START
import org.junit.Test;
// REMOVE_END

import java.util.*;
import java.util.concurrent.CompletableFuture;

// REMOVE_START
import static org.junit.Assert.assertEquals;
// REMOVE_END

public class StringExample {
// REMOVE_START
@Test
// REMOVE_END
public void run() {
RedisClient redisClient = RedisClient.create("redis://localhost:6379");

try (StatefulRedisConnection<String, String> connection = redisClient.connect()) {
RedisAsyncCommands<String, String> asyncCommands = connection.async();

// STEP_START set_get
CompletableFuture<Void> setAndGet = asyncCommands.set("bike:1", "Deimos")
.thenCompose(v -> {
System.out.println(v); // OK
// REMOVE_START
assertEquals("OK", v);
// REMOVE_END
return asyncCommands.get("bike:1");
})
// REMOVE_START
.thenApply(res -> {
assertEquals("Deimos", res);
return res;
})
// REMOVE_END
.thenAccept(System.out::println) // Deimos
.toCompletableFuture();
// STEP_END

// STEP_START setnx_xx
CompletableFuture<Void> setnx = asyncCommands.setnx("bike:1", "bike")
.thenCompose(v -> {
System.out.println(v); // false (because key already exists)
// REMOVE_START
assertEquals(false, v);
// REMOVE_END
return asyncCommands.get("bike:1");
})
// REMOVE_START
.thenApply(res -> {
assertEquals("Deimos", res);
return res;
})
// REMOVE_END
.thenAccept(System.out::println) // Deimos (value is unchanged)
.toCompletableFuture();

// set the value to "bike" if it already exists
CompletableFuture<Void> setxx = asyncCommands.set("bike:1", "bike", SetArgs.Builder.xx())
//REMOVE_START
.thenApply(res -> {
assertEquals("OK", res);
return res;
})
// REMOVE_END
.thenAccept(System.out::println) // OK
.toCompletableFuture();
// STEP_END

// STEP_START mset
Map<String, String> bikeMap = new HashMap<>();
bikeMap.put("bike:1", "Deimos");
bikeMap.put("bike:2", "Ares");
bikeMap.put("bike:3", "Vanth");

CompletableFuture<Void> mset = asyncCommands.mset(bikeMap)
.thenCompose(v -> {
System.out.println(v); // OK
return asyncCommands.mget("bike:1", "bike:2", "bike:3");
})
// REMOVE_START
.thenApply(res -> {
List<KeyValue<String, String>> expected = new ArrayList<>(Arrays.asList(
KeyValue.just("bike:1", "Deimos"),
KeyValue.just("bike:2", "Ares"),
KeyValue.just("bike:3", "Vanth")
));
assertEquals(expected, res);
return res;
})
// REMOVE_END
.thenAccept(System.out::println) // [KeyValue[bike:1, Deimos], KeyValue[bike:2, Ares], KeyValue[bike:3, Vanth]]
.toCompletableFuture();
// STEP_END

// STEP_START incr
CompletableFuture<Void> incrby = asyncCommands.set("total_crashes", "0")
.thenCompose(v -> asyncCommands.incr("total_crashes"))
.thenCompose(v -> {
System.out.println(v); // 1
// REMOVE_START
assertEquals(1L, v.longValue());
// REMOVE_END
return asyncCommands.incrby("total_crashes", 10);
})
// REMOVE_START
.thenApply(res -> {
assertEquals(11L, res.longValue());
return res;
})
// REMOVE_END
.thenAccept(System.out::println) // 11
.toCompletableFuture();
// STEP_END

CompletableFuture.allOf(setAndGet, setnx, setxx, mset, incrby).join();

} finally {
redisClient.shutdown();
}
}
}
121 changes: 121 additions & 0 deletions src/test/java/io/redis/examples/reactive/StringExample.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,121 @@
// EXAMPLE: set_tutorial
package io.redis.examples.reactive;

import io.lettuce.core.*;
import io.lettuce.core.api.reactive.RedisReactiveCommands;
import io.lettuce.core.api.StatefulRedisConnection;
// REMOVE_START
import org.junit.Test;
// REMOVE_END
import reactor.core.publisher.Mono;

import java.util.*;

// REMOVE_START
import static org.junit.Assert.assertEquals;
// REMOVE_END

public class StringExample {
// REMOVE_START
@Test
// REMOVE_END
public void run() {
RedisClient redisClient = RedisClient.create("redis://localhost:6379");

try (StatefulRedisConnection<String, String> connection = redisClient.connect()) {
RedisReactiveCommands<String, String> reactiveCommands = connection.reactive();

// STEP_START set_get
Mono<Void> setAndGet = reactiveCommands.set("bike:1", "Deimos")
.doOnNext(v -> {
System.out.println(v); // OK
// REMOVE_START
assertEquals("OK", v);
// REMOVE_END
})
.flatMap(v -> reactiveCommands.get("bike:1"))
.doOnNext(res -> {
// REMOVE_START
assertEquals("Deimos", res);
// REMOVE_END
System.out.println(res); // Deimos
})
.then();
// STEP_END

// STEP_START setnx_xx
Mono<Void> setnx = reactiveCommands.setnx("bike:1", "bike")
.doOnNext(v -> {
System.out.println(v); // false (because key already exists)
// REMOVE_START
assertEquals(false, v);
// REMOVE_END
})
.flatMap(v -> reactiveCommands.get("bike:1"))
.doOnNext(res -> {
// REMOVE_START
assertEquals("Deimos", res);
// REMOVE_END
System.out.println(res); // Deimos (value is unchanged)
})
.then();

Mono<Void> setxx = reactiveCommands.set("bike:1", "bike", SetArgs.Builder.xx())
.doOnNext(res -> {
// REMOVE_START
assertEquals("OK", res);
// REMOVE_END
System.out.println(res); // OK
})
.then();
// STEP_END

// STEP_START mset
Map<String, String> bikeMap = new HashMap<>();
bikeMap.put("bike:1", "Deimos");
bikeMap.put("bike:2", "Ares");
bikeMap.put("bike:3", "Vanth");

Mono<Void> mset = reactiveCommands.mset(bikeMap)
.doOnNext(System.out::println) // OK
.flatMap(v -> reactiveCommands.mget("bike:1", "bike:2", "bike:3").collectList())
.doOnNext(res -> {
List<KeyValue<String, String>> expected = new ArrayList<>(Arrays.asList(
KeyValue.just("bike:1", "Deimos"),
KeyValue.just("bike:2", "Ares"),
KeyValue.just("bike:3", "Vanth")
));
// REMOVE_START
assertEquals(expected, res);
// REMOVE_END
System.out.println(res); // [KeyValue[bike:1, Deimos], KeyValue[bike:2, Ares], KeyValue[bike:3, Vanth]]
})
.then();
// STEP_END

// STEP_START incr
Mono<Void> incrby = reactiveCommands.set("total_crashes", "0")
.flatMap(v -> reactiveCommands.incr("total_crashes"))
.doOnNext(v -> {
System.out.println(v); // 1
// REMOVE_START
assertEquals(1L, v.longValue());
// REMOVE_END
})
.flatMap(v -> reactiveCommands.incrby("total_crashes", 10))
.doOnNext(res -> {
// REMOVE_START
assertEquals(11L, res.longValue());
// REMOVE_END
System.out.println(res); // 11
})
.then();
// STEP_END

Mono.when(setAndGet, setnx, setxx, mset, incrby).block();

} finally {
redisClient.shutdown();
}
}
}

0 comments on commit 2c7fd42

Please sign in to comment.