@effect/cluster 0.45.0 → 0.46.1

package/src/ShardManager.ts

@@ -373,12 +373,26 @@ export const makeClientRpc: Effect.Effect<
 
   return ShardManagerClient.of({
     register: (address, groups) =>
-      client.Register({ runner: Runner.make({ address, version: config.serverVersion, groups }) }),
-    unregister: (address) => client.Unregister({ address }),
-    notifyUnhealthyRunner: (address) => client.NotifyUnhealthyRunner({ address }),
-    getAssignments: client.GetAssignments(),
-    shardingEvents: client.ShardingEvents(void 0, { asMailbox: true }),
-    getTime: client.GetTime()
+      client.Register({ runner: Runner.make({ address, version: config.serverVersion, groups }) }).pipe(
+        Effect.orDie
+      ),
+    unregister: (address) => Effect.orDie(client.Unregister({ address })),
+    notifyUnhealthyRunner: (address) => Effect.orDie(client.NotifyUnhealthyRunner({ address })),
+    getAssignments: Effect.orDie(client.GetAssignments()),
+    shardingEvents: Mailbox.make<ShardingEvent>().pipe(
+      Effect.tap(Effect.fnUntraced(
+        function*(mailbox) {
+          const events = yield* client.ShardingEvents(void 0, { asMailbox: true })
+          const take = Effect.orDie(events.takeAll)
+          while (true) {
+            mailbox.unsafeOfferAll((yield* take)[0])
+          }
+        },
+        (effect, mb) => Mailbox.into(effect, mb),
+        Effect.forkScoped
+      ))
+    ),
+    getTime: Effect.orDie(client.GetTime())
   })
 })
 

package/src/Sharding.ts

@@ -37,6 +37,7 @@ import {
   RunnerUnavailable
 } from "./ClusterError.js"
 import { Persisted, Uninterruptible } from "./ClusterSchema.js"
+import * as ClusterSchema from "./ClusterSchema.js"
 import type { CurrentAddress, CurrentRunnerAddress, Entity, HandlersFrom } from "./Entity.js"
 import { EntityAddress } from "./EntityAddress.js"
 import { EntityId } from "./EntityId.js"
@@ -950,6 +951,7 @@ const make = Effect.gen(function*() {
   > = yield* ResourceMap.make(Effect.fnUntraced(function*(entity: Entity<string, any>) {
     const client = yield* RpcClient.makeNoSerialization(entity.protocol, {
       spanPrefix: `${entity.type}.client`,
+      disableTracing: !Context.get(entity.protocol.annotations, ClusterSchema.ClientTracingEnabled),
       supportsAck: true,
       generateRequestId: () => RequestId(snowflakeGen.unsafeNext()),
       flatten: true,