Some refactoring / qol improvements

Author: rpanic

packages/api/src/metrics/OpenTelemetryServer.ts

@@ -32,17 +32,15 @@ export type OpenTelemetryServerConfig = {
 };
 
 @sequencerModule()
-export class OpenTelemetryServer
-  extends SequencerModule<OpenTelemetryServerConfig>
-  implements DependencyFactory
-{
+// implements DependencyFactory
+export class OpenTelemetryServer extends SequencerModule<OpenTelemetryServerConfig> {
   public constructor(
     @inject("Sequencer") private readonly sequencer: Sequencer<any>
   ) {
     super();
   }
 
-  public dependencies() {
+  public static dependencies() {
     return {
       Tracer: {
         useClass: OpenTelemetryTracer,

packages/api/src/metrics/OpenTelemetryTracer.ts

@@ -11,7 +11,8 @@ export class OpenTelemetryTracer implements Tracer {
     // We need to import this here, so that the OpenTelemetryServer will be resolved
     // before this module, and therefore will be already started when this module is
     // eventually consumed and used
-    @inject("OpenTelemetryServer") openTelemetryServer: OpenTelemetryServer
+    @inject("OpenTelemetryServer")
+    private readonly openTelemetryServer: OpenTelemetryServer
   ) {
     noop();
   }
@@ -24,6 +25,9 @@ export class OpenTelemetryTracer implements Tracer {
     f: () => Promise<T>,
     metadata?: Record<string, string | number>
   ) {
+    if (this.openTelemetryServer.config.tracing?.enabled !== true) {
+      return await f();
+    }
     if (this.tracer === undefined) {
       this.tracer = opentelemetry.trace.getTracer("protokit", "canary");
     }

packages/library/src/runtime/Withdrawals.ts

@@ -1,4 +1,5 @@
 import {
+  outgoingMessage,
   OutgoingMessages,
   runtimeModule,
   RuntimeModule,
@@ -28,6 +29,7 @@ export class Withdrawal extends Struct({
 
 @runtimeModule()
 export class Withdrawals extends RuntimeModule {
+  @outgoingMessage()
   messages = new OutgoingMessages({
     withdrawal: Withdrawal,
   });

packages/sequencer/src/sequencer/SequencerStartupModule.ts

@@ -151,6 +151,15 @@ export class SequencerStartupModule
 
     log.info("Compiling Protocol circuits, this can take a few minutes");
 
+    const timeout = setTimeout(
+      () => {
+        log.error(
+          "No response yet received from workers - have you configured a TaskQueue and a corresponding worker running (either in-process or somewhere else)"
+        );
+      },
+      this.areProofsEnabled.areProofsEnabled ? 20 * 60 * 1000 : 30 * 1000
+    );
+
     const root = await this.compileRuntime(flow);
 
     const protocolBridgeArtifacts = await this.compileProtocolAndBridge(
@@ -176,6 +185,8 @@ export class SequencerStartupModule
     });
 
     log.info("Protocol circuits compiled successfully, commencing startup");
+
+    clearTimeout(timeout);
   }
 
   public async close() {

packages/sequencer/src/worker/worker/FlowTaskWorker.ts

@@ -106,6 +106,8 @@ export class FlowTaskWorker<Tasks extends Task<any, any>[]>
       ...newWorkers,
     };
 
+    log.info("All tasks prepared and ready");
+
     this.prepareResolve!();
   }