New replacement classes for Read-Only build (#11)

* new MarquezApp replacement (0.50.0) * new MarquezContext replacement (0.50.0)
nubank · Nov 19, 2024 · a6d4011 · a6d4011
1 parent f400525
commit a6d4011
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/api/read-only-replacements/MarquezApp.java b/api/read-only-replacements/MarquezApp.java
@@ -28,7 +28,7 @@
 import marquez.api.filter.JobRedirectFilter;
 import marquez.api.filter.exclusions.Exclusions;
 import marquez.api.filter.exclusions.ExclusionsConfig;
-import marquez.cli.DbMigrationCommand;
+import marquez.cli.DbMigrateCommand;
 import marquez.cli.DbRetentionCommand;
 import marquez.cli.MetadataCommand;
 import marquez.cli.SeedCommand;
@@ -90,6 +90,7 @@ public void initialize(@NonNull Bootstrap<MarquezConfig> bootstrap) {
             new EnvironmentVariableSubstitutor(ERROR_ON_UNDEFINED)));
 
     // Add CLI commands
+    bootstrap.addCommand(new DbMigrateCommand());
     bootstrap.addCommand(new DbRetentionCommand());
     bootstrap.addCommand(new MetadataCommand());
     bootstrap.addCommand(new SeedCommand());
@@ -202,12 +203,6 @@ public void registerResources(
     }
   }
 
-  @Override
-  protected void addDefaultCommands(Bootstrap<MarquezConfig> bootstrap) {
-    bootstrap.addCommand(new DbMigrationCommand<>(this));
-    super.addDefaultCommands(bootstrap);
-  }
-
   private void registerServlets(@NonNull Environment env) {
     log.debug("Registering servlets...");
 
@@ -224,4 +219,4 @@ private void registerFilters(@NonNull Environment env, MarquezContext marquezCon
         .getResourceConfig()
         .register(new JobRedirectFilter(marquezContext.getJobService()));
   }
-}
+}
diff --git a/api/read-only-replacements/MarquezContext.java b/api/read-only-replacements/MarquezContext.java
@@ -155,7 +155,7 @@ private MarquezContext(
     this.tagService = new TagService(baseDao);
     // this.tagService.init(tags);
     this.openLineageService = new OpenLineageService(baseDao, runService);
-    this.lineageService = new LineageService(lineageDao, jobDao);
+    this.lineageService = new LineageService(lineageDao, jobDao, runDao);
     this.columnLineageService = new ColumnLineageService(columnLineageDao, datasetFieldDao);
     this.searchService = new SearchService(searchConfig);
     this.statsService = new StatsService(statsDao);
@@ -254,4 +254,4 @@ public MarquezContext build() {
       return new MarquezContext(jdbi, searchConfig, tags, runTransitionListeners);
     }
   }
-}
+}