543 lines
14 KiB
Groovy
543 lines
14 KiB
Groovy
import com.lambdaworks.redis.ClientOptions
|
|
import com.lambdaworks.redis.RedisClient
|
|
import com.lambdaworks.redis.RedisFuture
|
|
import com.lambdaworks.redis.RedisURI
|
|
import com.lambdaworks.redis.api.StatefulConnection
|
|
import com.lambdaworks.redis.api.async.RedisAsyncCommands
|
|
import com.lambdaworks.redis.api.sync.RedisCommands
|
|
import com.lambdaworks.redis.protocol.AsyncCommand
|
|
import datadog.trace.agent.test.AgentTestRunner
|
|
import datadog.trace.agent.test.utils.PortUtils
|
|
import datadog.trace.api.DDSpanTypes
|
|
import datadog.trace.bootstrap.instrumentation.api.Tags
|
|
|
|
import redis.embedded.RedisServer
|
|
import spock.lang.Shared
|
|
import spock.util.concurrent.AsyncConditions
|
|
import com.lambdaworks.redis.codec.Utf8StringCodec
|
|
|
|
import java.util.concurrent.CancellationException
|
|
import java.util.concurrent.TimeUnit
|
|
import java.util.function.BiConsumer
|
|
import java.util.function.BiFunction
|
|
import java.util.function.Consumer
|
|
import java.util.function.Function
|
|
|
|
import com.lambdaworks.redis.RedisConnectionException
|
|
|
|
import static datadog.trace.instrumentation.lettuce.InstrumentationPoints.AGENT_CRASHING_COMMAND_PREFIX
|
|
|
|
class LettuceAsyncClientTest extends AgentTestRunner {
|
|
public static final String HOST = "127.0.0.1"
|
|
public static final int DB_INDEX = 0
|
|
// Disable autoreconnect so we do not get stray traces popping up on server shutdown
|
|
public static final ClientOptions CLIENT_OPTIONS = new ClientOptions.Builder().autoReconnect(false).build()
|
|
|
|
@Shared
|
|
int port
|
|
@Shared
|
|
int incorrectPort
|
|
@Shared
|
|
String dbAddr
|
|
@Shared
|
|
String dbAddrNonExistent
|
|
@Shared
|
|
String dbUriNonExistent
|
|
@Shared
|
|
String embeddedDbUri
|
|
|
|
@Shared
|
|
RedisServer redisServer
|
|
|
|
@Shared
|
|
Map<String, String> testHashMap = [
|
|
firstname: "John",
|
|
lastname : "Doe",
|
|
age : "53"
|
|
]
|
|
|
|
RedisClient redisClient
|
|
StatefulConnection connection
|
|
RedisAsyncCommands<String, ?> asyncCommands
|
|
RedisCommands<String, ?> syncCommands
|
|
|
|
def setupSpec() {
|
|
port = PortUtils.randomOpenPort()
|
|
incorrectPort = PortUtils.randomOpenPort()
|
|
dbAddr = HOST + ":" + port + "/" + DB_INDEX
|
|
dbAddrNonExistent = HOST + ":" + incorrectPort + "/" + DB_INDEX
|
|
dbUriNonExistent = "redis://" + dbAddrNonExistent
|
|
embeddedDbUri = "redis://" + dbAddr
|
|
|
|
redisServer = RedisServer.builder()
|
|
// bind to localhost to avoid firewall popup
|
|
.setting("bind " + HOST)
|
|
// set max memory to avoid problems in CI
|
|
.setting("maxmemory 128M")
|
|
.port(port).build()
|
|
}
|
|
|
|
def setup() {
|
|
redisClient = RedisClient.create(embeddedDbUri)
|
|
|
|
println "Using redis: $redisServer.args"
|
|
redisServer.start()
|
|
redisClient.setOptions(CLIENT_OPTIONS)
|
|
|
|
connection = redisClient.connect()
|
|
asyncCommands = connection.async()
|
|
syncCommands = connection.sync()
|
|
|
|
syncCommands.set("TESTKEY", "TESTVAL")
|
|
|
|
// 1 set + 1 connect trace
|
|
TEST_WRITER.waitForTraces(2)
|
|
TEST_WRITER.clear()
|
|
}
|
|
|
|
def cleanup() {
|
|
connection.close()
|
|
redisServer.stop()
|
|
}
|
|
|
|
def "connect using get on ConnectionFuture"() {
|
|
setup:
|
|
RedisClient testConnectionClient = RedisClient.create(embeddedDbUri)
|
|
testConnectionClient.setOptions(CLIENT_OPTIONS)
|
|
|
|
when:
|
|
StatefulConnection connection = testConnectionClient.connect(new Utf8StringCodec(),
|
|
new RedisURI(HOST, port, 3, TimeUnit.SECONDS))
|
|
|
|
then:
|
|
connection != null
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "CONNECT:" + dbAddr
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.PEER_HOSTNAME" HOST
|
|
"$Tags.PEER_PORT" port
|
|
"$Tags.DB_TYPE" "redis"
|
|
"db.redis.dbIndex" 0
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
cleanup:
|
|
connection.close()
|
|
}
|
|
|
|
def "connect exception inside the connection future"() {
|
|
setup:
|
|
RedisClient testConnectionClient = RedisClient.create(dbUriNonExistent)
|
|
testConnectionClient.setOptions(CLIENT_OPTIONS)
|
|
|
|
when:
|
|
StatefulConnection connection = testConnectionClient.connect(new Utf8StringCodec(),
|
|
new RedisURI(HOST, incorrectPort, 3, TimeUnit.SECONDS))
|
|
|
|
then:
|
|
connection == null
|
|
thrown RedisConnectionException
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "CONNECT:" + dbAddrNonExistent
|
|
errored true
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.PEER_HOSTNAME" HOST
|
|
"$Tags.PEER_PORT" incorrectPort
|
|
"$Tags.DB_TYPE" "redis"
|
|
"db.redis.dbIndex" 0
|
|
errorTags RedisConnectionException, String
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "set command using Future get with timeout"() {
|
|
setup:
|
|
RedisFuture<String> redisFuture = asyncCommands.set("TESTSETKEY", "TESTSETVAL")
|
|
String res = redisFuture.get(3, TimeUnit.SECONDS)
|
|
|
|
expect:
|
|
res == "OK"
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "SET"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "get command chained with thenAccept"() {
|
|
setup:
|
|
def conds = new AsyncConditions()
|
|
Consumer<String> consumer = new Consumer<String>() {
|
|
@Override
|
|
void accept(String res) {
|
|
conds.evaluate {
|
|
assert res == "TESTVAL"
|
|
}
|
|
}
|
|
}
|
|
|
|
when:
|
|
RedisFuture<String> redisFuture = asyncCommands.get("TESTKEY")
|
|
redisFuture.thenAccept(consumer)
|
|
|
|
then:
|
|
conds.await()
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "GET"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// to make sure instrumentation's chained completion stages won't interfere with user's, while still
|
|
// recording metrics
|
|
def "get non existent key command with handleAsync and chained with thenApply"() {
|
|
setup:
|
|
def conds = new AsyncConditions()
|
|
final String successStr = "KEY MISSING"
|
|
BiFunction<String, Throwable, String> firstStage = new BiFunction<String, Throwable, String>() {
|
|
@Override
|
|
String apply(String res, Throwable throwable) {
|
|
conds.evaluate {
|
|
assert res == null
|
|
assert throwable == null
|
|
}
|
|
return (res == null ? successStr : res)
|
|
}
|
|
}
|
|
Function<String, Object> secondStage = new Function<String, Object>() {
|
|
@Override
|
|
Object apply(String input) {
|
|
conds.evaluate {
|
|
assert input == successStr
|
|
}
|
|
return null
|
|
}
|
|
}
|
|
|
|
when:
|
|
RedisFuture<String> redisFuture = asyncCommands.get("NON_EXISTENT_KEY")
|
|
redisFuture.handleAsync(firstStage).thenApply(secondStage)
|
|
|
|
then:
|
|
conds.await()
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "GET"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "command with no arguments using a biconsumer"() {
|
|
setup:
|
|
def conds = new AsyncConditions()
|
|
BiConsumer<String, Throwable> biConsumer = new BiConsumer<String, Throwable>() {
|
|
@Override
|
|
void accept(String keyRetrieved, Throwable throwable) {
|
|
conds.evaluate {
|
|
assert keyRetrieved != null
|
|
}
|
|
}
|
|
}
|
|
|
|
when:
|
|
RedisFuture<String> redisFuture = asyncCommands.randomkey()
|
|
redisFuture.whenCompleteAsync(biConsumer)
|
|
|
|
then:
|
|
conds.await()
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "RANDOMKEY"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "hash set and then nest apply to hash getall"() {
|
|
setup:
|
|
def conds = new AsyncConditions()
|
|
|
|
when:
|
|
RedisFuture<String> hmsetFuture = asyncCommands.hmset("TESTHM", testHashMap)
|
|
hmsetFuture.thenApplyAsync(new Function<String, Object>() {
|
|
@Override
|
|
Object apply(String setResult) {
|
|
TEST_WRITER.waitForTraces(1) // Wait for 'hmset' trace to get written
|
|
conds.evaluate {
|
|
assert setResult == "OK"
|
|
}
|
|
RedisFuture<Map<String, String>> hmGetAllFuture = asyncCommands.hgetall("TESTHM")
|
|
hmGetAllFuture.exceptionally(new Function<Throwable, Map<String, String>>() {
|
|
@Override
|
|
Map<String, String> apply(Throwable throwable) {
|
|
println("unexpected:" + throwable.toString())
|
|
throwable.printStackTrace()
|
|
assert false
|
|
return null
|
|
}
|
|
})
|
|
hmGetAllFuture.thenAccept(new Consumer<Map<String, String>>() {
|
|
@Override
|
|
void accept(Map<String, String> hmGetAllResult) {
|
|
conds.evaluate {
|
|
assert testHashMap == hmGetAllResult
|
|
}
|
|
}
|
|
})
|
|
return null
|
|
}
|
|
})
|
|
|
|
then:
|
|
conds.await()
|
|
assertTraces(2) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "HMSET"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
trace(1, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "HGETALL"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "command completes exceptionally"() {
|
|
setup:
|
|
// turn off auto flush to complete the command exceptionally manually
|
|
asyncCommands.setAutoFlushCommands(false)
|
|
def conds = new AsyncConditions()
|
|
RedisFuture redisFuture = asyncCommands.del("key1", "key2")
|
|
boolean completedExceptionally = ((AsyncCommand) redisFuture).completeExceptionally(new IllegalStateException("TestException"))
|
|
redisFuture.exceptionally({
|
|
throwable ->
|
|
conds.evaluate {
|
|
assert throwable != null
|
|
assert throwable instanceof IllegalStateException
|
|
assert throwable.getMessage() == "TestException"
|
|
}
|
|
throw throwable
|
|
})
|
|
|
|
when:
|
|
// now flush and execute the command
|
|
asyncCommands.flushCommands()
|
|
redisFuture.get()
|
|
|
|
then:
|
|
conds.await()
|
|
completedExceptionally == true
|
|
thrown Exception
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "DEL"
|
|
errored true
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
errorTags(IllegalStateException, "TestException")
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "cancel command before it finishes"() {
|
|
setup:
|
|
asyncCommands.setAutoFlushCommands(false)
|
|
def conds = new AsyncConditions()
|
|
RedisFuture redisFuture = asyncCommands.sadd("SKEY", "1", "2")
|
|
redisFuture.whenCompleteAsync({
|
|
res, throwable ->
|
|
conds.evaluate {
|
|
assert throwable != null
|
|
assert throwable instanceof CancellationException
|
|
}
|
|
})
|
|
|
|
when:
|
|
boolean cancelSuccess = redisFuture.cancel(true)
|
|
asyncCommands.flushCommands()
|
|
|
|
then:
|
|
conds.await()
|
|
cancelSuccess == true
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "SADD"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
"db.command.cancelled" true
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def "debug segfault command (returns void) with no argument should produce span"() {
|
|
setup:
|
|
asyncCommands.debugSegfault()
|
|
|
|
expect:
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName AGENT_CRASHING_COMMAND_PREFIX + "DEBUG"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
def "shutdown command (returns void) should produce a span"() {
|
|
setup:
|
|
asyncCommands.shutdown(false)
|
|
|
|
expect:
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
serviceName "redis"
|
|
operationName "redis.query"
|
|
spanType DDSpanTypes.REDIS
|
|
resourceName "SHUTDOWN"
|
|
errored false
|
|
|
|
tags {
|
|
"$Tags.COMPONENT" "redis-client"
|
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
|
"$Tags.DB_TYPE" "redis"
|
|
defaultTags()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|