Injecting BlockNodeContext using Dagger

Signed-off-by: Alfredo Gutierrez <[email protected]>
hashgraph · Aug 21, 2024 · 4d3dfea · 4d3dfea
1 parent d1b5d62
commit 4d3dfea
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/server/src/main/java/com/hedera/block/server/BlockNodeApp.java b/server/src/main/java/com/hedera/block/server/BlockNodeApp.java
@@ -17,7 +17,6 @@
 package com.hedera.block.server;
 
 import com.hedera.block.server.config.BlockNodeContext;
-import com.hedera.block.server.config.BlockNodeContextFactory;
 import com.hedera.block.server.data.ObjectEvent;
 import com.hedera.block.server.health.HealthService;
 import com.hedera.block.server.mediator.LiveStreamMediatorBuilder;
@@ -46,6 +45,7 @@ public class BlockNodeApp {
     private static final System.Logger LOGGER = System.getLogger(Server.class.getName());
     private final ServiceStatus serviceStatus;
     private final HealthService healthService;
+    private final BlockNodeContext blockNodeContext;
 
     /**
      * Has all needed dependencies to start the server and initialize the context.
@@ -55,9 +55,12 @@ public class BlockNodeApp {
      */
     @Inject
     public BlockNodeApp(
-            @NonNull ServiceStatus serviceStatus, @NonNull HealthService healthService) {
+            @NonNull ServiceStatus serviceStatus,
+            @NonNull HealthService healthService,
+            @NonNull BlockNodeContext blockNodeContext) {
         this.serviceStatus = serviceStatus;
         this.healthService = healthService;
+        this.blockNodeContext = blockNodeContext;
     }
 
     /**
@@ -66,8 +69,6 @@ public BlockNodeApp(
      * @throws IOException if the server cannot be started
      */
     public void startServer() throws IOException {
-        // init context, metrics, and configuration.
-        @NonNull final BlockNodeContext blockNodeContext = BlockNodeContextFactory.create();
 
         @NonNull
         final BlockWriter<com.hedera.block.protos.BlockStreamService.BlockItem> blockWriter =

diff --git a/server/src/main/java/com/hedera/block/server/BlockNodeAppInjectionModule.java b/server/src/main/java/com/hedera/block/server/BlockNodeAppInjectionModule.java
@@ -16,9 +16,14 @@
 
 package com.hedera.block.server;
 
+import com.hedera.block.server.config.BlockNodeContext;
+import com.hedera.block.server.config.BlockNodeContextFactory;
 import dagger.Binds;
 import dagger.Module;
+import dagger.Provides;
+
 import javax.inject.Singleton;
+import java.io.IOException;
 
 /**
  * A Dagger Module for interfaces that are at the BlockNodeApp Level, should be temporary and
@@ -36,4 +41,18 @@ public interface BlockNodeAppInjectionModule {
     @Singleton
     @Binds
     ServiceStatus bindServiceStatus(ServiceStatusImpl serviceStatus);
+
+    /**
+     * Provides a block node context singleton using the factory.
+     * @return a block node context singleton
+     */
+    @Provides
+    @Singleton
+    static BlockNodeContext provideBlockNodeContext() {
+        try {
+            return BlockNodeContextFactory.create();
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
 }