Update classes in example plugin (#761)

Yelp · Oct 9, 2024 · 0c58fce · 0c58fce
1 parent dfd991f
commit 0c58fce
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/example-plugin/src/main/java/com/yelp/nrtsearch/plugins/example/ExamplePlugin.java b/example-plugin/src/main/java/com/yelp/nrtsearch/plugins/example/ExamplePlugin.java
@@ -15,8 +15,8 @@
  */
 package com.yelp.nrtsearch.plugins.example;
 
-import com.yelp.nrtsearch.server.config.LuceneServerConfiguration;
-import com.yelp.nrtsearch.server.luceneserver.analysis.AnalysisProvider;
+import com.yelp.nrtsearch.server.analysis.AnalysisProvider;
+import com.yelp.nrtsearch.server.config.NrtsearchConfig;
 import com.yelp.nrtsearch.server.plugins.AnalysisPlugin;
 import com.yelp.nrtsearch.server.plugins.CustomRequestPlugin;
 import com.yelp.nrtsearch.server.plugins.Plugin;
@@ -30,7 +30,7 @@ public class ExamplePlugin extends Plugin implements AnalysisPlugin, CustomReque
   private final String availableAnalyzers = String.join(",", getAnalyzers().keySet());
 
   // Constructor that accepts LuceneServerConfiguration object is required
-  public ExamplePlugin(LuceneServerConfiguration configuration) {}
+  public ExamplePlugin(NrtsearchConfig configuration) {}
 
   @Override
   public String id() {

diff --git a/example-plugin/src/test/java/com/yelp/nrtsearch/plugins/example/ExamplePluginTest.java b/example-plugin/src/test/java/com/yelp/nrtsearch/plugins/example/ExamplePluginTest.java
@@ -17,7 +17,8 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 
-import com.yelp.nrtsearch.server.config.LuceneServerConfiguration;
+import com.yelp.nrtsearch.server.ServerTestCase;
+import com.yelp.nrtsearch.server.config.NrtsearchConfig;
 import com.yelp.nrtsearch.server.grpc.AddDocumentRequest;
 import com.yelp.nrtsearch.server.grpc.AddDocumentRequest.MultiValuedField;
 import com.yelp.nrtsearch.server.grpc.CustomRequest;
@@ -27,7 +28,6 @@
 import com.yelp.nrtsearch.server.grpc.Query;
 import com.yelp.nrtsearch.server.grpc.SearchRequest;
 import com.yelp.nrtsearch.server.grpc.SearchResponse;
-import com.yelp.nrtsearch.server.luceneserver.ServerTestCase;
 import com.yelp.nrtsearch.server.plugins.Plugin;
 import io.grpc.testing.GrpcCleanupRule;
 import java.io.ByteArrayInputStream;
@@ -46,7 +46,7 @@ public class ExamplePluginTest extends ServerTestCase {
   private static final ExamplePlugin examplePlugin = new ExamplePlugin(getConfig());
 
   @Override
-  protected List<Plugin> getPlugins(LuceneServerConfiguration configuration) {
+  protected List<Plugin> getPlugins(NrtsearchConfig configuration) {
     return List.of(examplePlugin);
   }
 
@@ -121,8 +121,8 @@ public void testAnalysisForSearch() {
     assertThat(response.getHitsCount()).isEqualTo(1);
   }
 
-  private static LuceneServerConfiguration getConfig() {
+  private static NrtsearchConfig getConfig() {
     String config = "nodeName: \"lucene_server_foo\"";
-    return new LuceneServerConfiguration(new ByteArrayInputStream(config.getBytes()));
+    return new NrtsearchConfig(new ByteArrayInputStream(config.getBytes()));
   }
 }