Add Distributed Tracing support

.gitignore

@@ -1,5 +1,6 @@
 .DS_Store
-/.build
+.build
+.swiftpm
 /Packages
 xcuserdata/
 DerivedData/
@@ -10,4 +11,4 @@ DerivedData/
 Package.resolved
 .benchmarkBaselines/
 .swift-version
-.docc-build
+.docc-build

Package.swift

@@ -19,12 +19,14 @@ let package = Package(
     ],
     traits: [
         .trait(name: "ServiceLifecycleSupport"),
-        .default(enabledTraits: ["ServiceLifecycleSupport"]),
+        .trait(name: "DistributedTracingSupport"),
+        .default(enabledTraits: ["ServiceLifecycleSupport", "DistributedTracingSupport"]),
     ],
     dependencies: [
         .package(url: "https://github.com/apple/swift-atomics.git", from: "1.0.0"),
         .package(url: "https://github.com/apple/swift-collections.git", from: "1.1.4"),
         .package(url: "https://github.com/apple/swift-log.git", from: "1.6.3"),
+        .package(url: "https://github.com/apple/swift-distributed-tracing.git", from: "1.0.0"),
         .package(url: "https://github.com/apple/swift-nio.git", from: "2.81.0"),
         .package(url: "https://github.com/apple/swift-nio-ssl.git", from: "2.29.0"),
         .package(url: "https://github.com/apple/swift-nio-transport-services.git", from: "1.23.0"),
@@ -39,6 +41,7 @@ let package = Package(
                 .byName(name: "_ValkeyConnectionPool"),
                 .product(name: "DequeModule", package: "swift-collections"),
                 .product(name: "Logging", package: "swift-log"),
+                .product(name: "Tracing", package: "swift-distributed-tracing", condition: .when(traits: ["DistributedTracingSupport"])),
                 .product(name: "NIOCore", package: "swift-nio"),
                 .product(name: "NIOPosix", package: "swift-nio"),
                 .product(name: "NIOSSL", package: "swift-nio-ssl"),

Sources/Valkey/Commands/BitmapCommands.swift

@@ -82,6 +82,8 @@ public struct BITCOUNT: ValkeyCommand {
     }
     public typealias Response = Int
 
+    @inlinable public static var name: String { "BITCOUNT" }
+
     public var key: ValkeyKey
     public var range: Range?
 
@@ -247,6 +249,8 @@ public struct BITFIELD: ValkeyCommand {
     }
     public typealias Response = RESPToken.Array
 
+    @inlinable public static var name: String { "BITFIELD" }
+
     public var key: ValkeyKey
     public var operations: [Operation]
 
@@ -287,6 +291,8 @@ public struct BITFIELDRO: ValkeyCommand {
     }
     public typealias Response = [Int]
 
+    @inlinable public static var name: String { "BITFIELD_RO" }
+
     public var key: ValkeyKey
     public var getBlocks: [GetBlock]
 
@@ -328,6 +334,8 @@ public struct BITOP: ValkeyCommand {
     }
     public typealias Response = Int
 
+    @inlinable public static var name: String { "BITOP" }
+
     public var operation: Operation
     public var destkey: ValkeyKey
     public var keys: [ValkeyKey]
@@ -405,6 +413,8 @@ public struct BITPOS: ValkeyCommand {
     }
     public typealias Response = Int
 
+    @inlinable public static var name: String { "BITPOS" }
+
     public var key: ValkeyKey
     public var bit: Int
     public var range: Range?
@@ -429,6 +439,8 @@ public struct BITPOS: ValkeyCommand {
 public struct GETBIT: ValkeyCommand {
     public typealias Response = Int
 
+    @inlinable public static var name: String { "GETBIT" }
+
     public var key: ValkeyKey
     public var offset: Int
 
@@ -451,6 +463,8 @@ public struct GETBIT: ValkeyCommand {
 public struct SETBIT: ValkeyCommand {
     public typealias Response = Int
 
+    @inlinable public static var name: String { "SETBIT" }
+
     public var key: ValkeyKey
     public var offset: Int
     public var value: Int

Sources/Valkey/Commands/ClusterCommands.swift

@@ -28,6 +28,8 @@ public enum CLUSTER {
     /// Assigns new hash slots to a node.
     @_documentation(visibility: internal)
     public struct ADDSLOTS: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER ADDSLOTS" }
+
         public var slots: [Int]
 
         @inlinable public init(slots: [Int]) {
@@ -62,6 +64,8 @@ public enum CLUSTER {
                 endSlot.encode(into: &commandEncoder)
             }
         }
+        @inlinable public static var name: String { "CLUSTER ADDSLOTSRANGE" }
+
         public var ranges: [Range]
 
         @inlinable public init(ranges: [Range]) {
@@ -78,6 +82,8 @@ public enum CLUSTER {
     public struct BUMPEPOCH: ValkeyCommand {
         public typealias Response = ByteBuffer
 
+        @inlinable public static var name: String { "CLUSTER BUMPEPOCH" }
+
         @inlinable public init() {
         }
 
@@ -91,6 +97,8 @@ public enum CLUSTER {
     public struct COUNTFAILUREREPORTS<NodeId: RESPStringRenderable>: ValkeyCommand {
         public typealias Response = Int
 
+        @inlinable public static var name: String { "CLUSTER COUNT-FAILURE-REPORTS" }
+
         public var nodeId: NodeId
 
         @inlinable public init(nodeId: NodeId) {
@@ -107,6 +115,8 @@ public enum CLUSTER {
     public struct COUNTKEYSINSLOT: ValkeyCommand {
         public typealias Response = Int
 
+        @inlinable public static var name: String { "CLUSTER COUNTKEYSINSLOT" }
+
         public var slot: Int
 
         @inlinable public init(slot: Int) {
@@ -121,6 +131,8 @@ public enum CLUSTER {
     /// Sets hash slots as unbound for a node.
     @_documentation(visibility: internal)
     public struct DELSLOTS: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER DELSLOTS" }
+
         public var slots: [Int]
 
         @inlinable public init(slots: [Int]) {
@@ -155,6 +167,8 @@ public enum CLUSTER {
                 endSlot.encode(into: &commandEncoder)
             }
         }
+        @inlinable public static var name: String { "CLUSTER DELSLOTSRANGE" }
+
         public var ranges: [Range]
 
         @inlinable public init(ranges: [Range]) {
@@ -184,6 +198,8 @@ public enum CLUSTER {
                 }
             }
         }
+        @inlinable public static var name: String { "CLUSTER FAILOVER" }
+
         public var options: Options?
 
         @inlinable public init(options: Options? = nil) {
@@ -198,6 +214,8 @@ public enum CLUSTER {
     /// Deletes all slots information from a node.
     @_documentation(visibility: internal)
     public struct FLUSHSLOTS: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER FLUSHSLOTS" }
+
         @inlinable public init() {
         }
 
@@ -209,6 +227,8 @@ public enum CLUSTER {
     /// Removes a node from the nodes table.
     @_documentation(visibility: internal)
     public struct FORGET<NodeId: RESPStringRenderable>: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER FORGET" }
+
         public var nodeId: NodeId
 
         @inlinable public init(nodeId: NodeId) {
@@ -225,6 +245,8 @@ public enum CLUSTER {
     public struct GETKEYSINSLOT: ValkeyCommand {
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER GETKEYSINSLOT" }
+
         public var slot: Int
         public var count: Int
 
@@ -243,6 +265,8 @@ public enum CLUSTER {
     public struct HELP: ValkeyCommand {
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER HELP" }
+
         @inlinable public init() {
         }
 
@@ -256,6 +280,8 @@ public enum CLUSTER {
     public struct INFO: ValkeyCommand {
         public typealias Response = ByteBuffer
 
+        @inlinable public static var name: String { "CLUSTER INFO" }
+
         @inlinable public init() {
         }
 
@@ -269,6 +295,8 @@ public enum CLUSTER {
     public struct KEYSLOT<Key: RESPStringRenderable>: ValkeyCommand {
         public typealias Response = Int
 
+        @inlinable public static var name: String { "CLUSTER KEYSLOT" }
+
         public var key: Key
 
         @inlinable public init(_ key: Key) {
@@ -285,6 +313,8 @@ public enum CLUSTER {
     public struct LINKS: ValkeyCommand {
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER LINKS" }
+
         @inlinable public init() {
         }
 
@@ -296,6 +326,8 @@ public enum CLUSTER {
     /// Forces a node to handshake with another node.
     @_documentation(visibility: internal)
     public struct MEET<Ip: RESPStringRenderable>: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER MEET" }
+
         public var ip: Ip
         public var port: Int
         public var clusterBusPort: Int?
@@ -316,6 +348,8 @@ public enum CLUSTER {
     public struct MYID: ValkeyCommand {
         public typealias Response = ByteBuffer
 
+        @inlinable public static var name: String { "CLUSTER MYID" }
+
         @inlinable public init() {
         }
 
@@ -329,6 +363,8 @@ public enum CLUSTER {
     public struct MYSHARDID: ValkeyCommand {
         public typealias Response = ByteBuffer
 
+        @inlinable public static var name: String { "CLUSTER MYSHARDID" }
+
         @inlinable public init() {
         }
 
@@ -342,6 +378,8 @@ public enum CLUSTER {
     public struct NODES: ValkeyCommand {
         public typealias Response = ByteBuffer
 
+        @inlinable public static var name: String { "CLUSTER NODES" }
+
         @inlinable public init() {
         }
 
@@ -355,6 +393,8 @@ public enum CLUSTER {
     public struct REPLICAS<NodeId: RESPStringRenderable>: ValkeyCommand {
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER REPLICAS" }
+
         public var nodeId: NodeId
 
         @inlinable public init(nodeId: NodeId) {
@@ -369,6 +409,8 @@ public enum CLUSTER {
     /// Configure a node as replica of a primary node.
     @_documentation(visibility: internal)
     public struct REPLICATE<NodeId: RESPStringRenderable>: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER REPLICATE" }
+
         public var nodeId: NodeId
 
         @inlinable public init(nodeId: NodeId) {
@@ -398,6 +440,8 @@ public enum CLUSTER {
                 }
             }
         }
+        @inlinable public static var name: String { "CLUSTER RESET" }
+
         public var resetType: ResetType?
 
         @inlinable public init(resetType: ResetType? = nil) {
@@ -412,6 +456,8 @@ public enum CLUSTER {
     /// Forces a node to save the cluster configuration to disk.
     @_documentation(visibility: internal)
     public struct SAVECONFIG: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER SAVECONFIG" }
+
         @inlinable public init() {
         }
 
@@ -423,6 +469,8 @@ public enum CLUSTER {
     /// Sets the configuration epoch for a new node.
     @_documentation(visibility: internal)
     public struct SETCONFIGEPOCH: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER SET-CONFIG-EPOCH" }
+
         public var configEpoch: Int
 
         @inlinable public init(configEpoch: Int) {
@@ -463,6 +511,8 @@ public enum CLUSTER {
                 }
             }
         }
+        @inlinable public static var name: String { "CLUSTER SETSLOT" }
+
         public var slot: Int
         public var subcommand: Subcommand
         public var timeout: Int?
@@ -481,6 +531,8 @@ public enum CLUSTER {
     /// Returns the mapping of cluster slots to shards.
     @_documentation(visibility: internal)
     public struct SHARDS: ValkeyCommand {
+        @inlinable public static var name: String { "CLUSTER SHARDS" }
+
         @inlinable public init() {
         }
 
@@ -494,6 +546,8 @@ public enum CLUSTER {
     public struct SLAVES<NodeId: RESPStringRenderable>: ValkeyCommand {
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER SLAVES" }
+
         public var nodeId: NodeId
 
         @inlinable public init(nodeId: NodeId) {
@@ -588,6 +642,8 @@ public enum CLUSTER {
         }
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER SLOT-STATS" }
+
         public var filter: Filter
 
         @inlinable public init(filter: Filter) {
@@ -604,6 +660,8 @@ public enum CLUSTER {
     public struct SLOTS: ValkeyCommand {
         public typealias Response = RESPToken.Array
 
+        @inlinable public static var name: String { "CLUSTER SLOTS" }
+
         @inlinable public init() {
         }
 
@@ -617,6 +675,8 @@ public enum CLUSTER {
 /// Signals that a cluster client is following an -ASK redirect.
 @_documentation(visibility: internal)
 public struct ASKING: ValkeyCommand {
+    @inlinable public static var name: String { "ASKING" }
+
     @inlinable public init() {
     }
 
@@ -628,6 +688,8 @@ public struct ASKING: ValkeyCommand {
 /// Enables read-only queries for a connection to a Valkey replica node.
 @_documentation(visibility: internal)
 public struct READONLY: ValkeyCommand {
+    @inlinable public static var name: String { "READONLY" }
+
     @inlinable public init() {
     }
 
@@ -639,6 +701,8 @@ public struct READONLY: ValkeyCommand {
 /// Enables read-write queries for a connection to a Valkey replica node.
 @_documentation(visibility: internal)
 public struct READWRITE: ValkeyCommand {
+    @inlinable public static var name: String { "READWRITE" }
+
     @inlinable public init() {
     }