This is an automated email from the ASF dual-hosted git repository.

mayanks pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-pinot.git


The following commit(s) were added to refs/heads/master by this push:
     new 04e70da  Initialize ControllerFilePathProvider in PerfBenchmarkDriver 
(#5383)
04e70da is described below

commit 04e70da72444e7e0810de33c14604f00809085eb
Author: Jialiang Li <[email protected]>
AuthorDate: Wed May 13 20:31:17 2020 -0700

    Initialize ControllerFilePathProvider in PerfBenchmarkDriver (#5383)
    
    Co-authored-by: Jack Li(Analytics Engineering) <[email protected]>
---
 .../src/main/java/org/apache/pinot/perf/BenchmarkQueryEngine.java       | 2 ++
 .../src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkDriver.java  | 2 +-
 .../src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java  | 2 ++
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git 
a/pinot-perf/src/main/java/org/apache/pinot/perf/BenchmarkQueryEngine.java 
b/pinot-perf/src/main/java/org/apache/pinot/perf/BenchmarkQueryEngine.java
index 1f447a0..7fe56e3 100644
--- a/pinot-perf/src/main/java/org/apache/pinot/perf/BenchmarkQueryEngine.java
+++ b/pinot-perf/src/main/java/org/apache/pinot/perf/BenchmarkQueryEngine.java
@@ -27,6 +27,7 @@ import org.apache.helix.ZNRecord;
 import org.apache.helix.manager.zk.ZNRecordSerializer;
 import org.apache.pinot.broker.requesthandler.OptimizationFlags;
 import org.apache.pinot.common.request.BrokerRequest;
+import org.apache.pinot.controller.api.resources.ControllerFilePathProvider;
 import org.apache.pinot.core.segment.index.metadata.SegmentMetadataImpl;
 import org.apache.pinot.tools.perf.PerfBenchmarkDriver;
 import org.apache.pinot.tools.perf.PerfBenchmarkDriverConf;
@@ -98,6 +99,7 @@ public class BenchmarkQueryEngine {
     conf.setConfigureResources(false);
     _perfBenchmarkDriver = new PerfBenchmarkDriver(conf);
     _perfBenchmarkDriver.run();
+    ControllerFilePathProvider.init(_perfBenchmarkDriver.getControllerConf());
 
     File[] segments = new File(DATA_DIRECTORY, TABLE_NAME).listFiles();
     for (File segmentDir : segments) {
diff --git 
a/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkDriver.java
 
b/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkDriver.java
index 13d6277..ec60358 100644
--- 
a/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkDriver.java
+++ 
b/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkDriver.java
@@ -204,7 +204,7 @@ public class PerfBenchmarkDriver {
     _controllerStarter.start();
   }
 
-  private ControllerConf getControllerConf() {
+  public ControllerConf getControllerConf() {
     ControllerConf conf = new ControllerConf();
     conf.setHelixClusterName(_clusterName);
     conf.setZkStr(_zkAddress);
diff --git 
a/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java
 
b/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java
index 959826b..8f24621 100644
--- 
a/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java
+++ 
b/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java
@@ -25,6 +25,7 @@ import java.util.List;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.TimeUnit;
+import org.apache.pinot.controller.api.resources.ControllerFilePathProvider;
 import org.apache.pinot.core.segment.index.metadata.SegmentMetadataImpl;
 import org.apache.pinot.tools.AbstractBaseCommand;
 import org.apache.pinot.tools.Command;
@@ -160,6 +161,7 @@ public class PerfBenchmarkRunner extends 
AbstractBaseCommand implements Command
     boolean tableConfigured = false;
     File[] segments = new File(dataDir, tableName).listFiles();
     Preconditions.checkNotNull(segments);
+    ControllerFilePathProvider.init(driver.getControllerConf());
     for (File segment : segments) {
       SegmentMetadataImpl segmentMetadata = new SegmentMetadataImpl(segment);
       if (!tableConfigured) {


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to