Implement CitcOperations via intellij-ext citc operations service

PiperOrigin-RevId: 612341088
bazelbuild · Mar 4, 2024 · e5acdcc · e5acdcc
1 parent 204ebdc
commit e5acdcc
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/base/src/com/google/idea/blaze/base/ext/IntelliJExtManager.java b/base/src/com/google/idea/blaze/base/ext/IntelliJExtManager.java
@@ -61,6 +61,9 @@ public class IntelliJExtManager {
   private static final BoolExperiment CRITIQUE_SERVICE =
       new BoolExperiment("use.intellij.ext.critiqueservice", false);
 
+  private static final BoolExperiment CITC_OPERATIONS_SERVICE =
+      new BoolExperiment("use.intellij.ext.citcoperationsservice", false);
+
   private static final BoolExperiment CODESEARCH =
       new BoolExperiment("use.intellij.ext.codesearch", false);
 
@@ -189,6 +192,10 @@ public boolean isCritiqueServiceEnabled() {
     return isEnabled() && CRITIQUE_SERVICE.getValue();
   }
 
+  public boolean isCitcOperationsServiceEnabled() {
+    return isEnabled() && CITC_OPERATIONS_SERVICE.getValue();
+  }
+
   public boolean isCodeSearchEnabled() {
     return isEnabled() && CODESEARCH.getValue();
   }

diff --git a/ext/src/com/google/idea/blaze/ext/IntelliJExtClient.java b/ext/src/com/google/idea/blaze/ext/IntelliJExtClient.java
@@ -19,7 +19,7 @@
 import com.google.idea.blaze.ext.BuildServiceGrpc.BuildServiceBlockingStub;
 import com.google.idea.blaze.ext.BuildServiceGrpc.BuildServiceFutureStub;
 import com.google.idea.blaze.ext.ChatBotModelGrpc.ChatBotModelBlockingStub;
-import com.google.idea.blaze.ext.CitcOperationsServiceGrpc.CitcOperationsServiceFutureStub;
+import com.google.idea.blaze.ext.CitcOperationsServiceGrpc.CitcOperationsServiceBlockingStub;
 import com.google.idea.blaze.ext.CodeSearchGrpc.CodeSearchFutureStub;
 import com.google.idea.blaze.ext.CritiqueServiceGrpc.CritiqueServiceBlockingStub;
 import com.google.idea.blaze.ext.DepServerGrpc.DepServerFutureStub;
@@ -133,8 +133,8 @@ public DepServerFutureStub getDependencyService() {
     return DepServerGrpc.newFutureStub(channel);
   }
 
-  public CitcOperationsServiceFutureStub getCitcOperationsService() {
-    return CitcOperationsServiceGrpc.newFutureStub(channel);
+  public CitcOperationsServiceBlockingStub getCitcOperationsService() {
+    return CitcOperationsServiceGrpc.newBlockingStub(channel);
   }
 
   public CodeSearchFutureStub getCodeSearchService() {

diff --git a/ext/src/com/google/idea/blaze/ext/IntelliJExtService.java b/ext/src/com/google/idea/blaze/ext/IntelliJExtService.java
@@ -20,7 +20,7 @@
 import com.google.idea.blaze.ext.BuildServiceGrpc.BuildServiceBlockingStub;
 import com.google.idea.blaze.ext.BuildServiceGrpc.BuildServiceFutureStub;
 import com.google.idea.blaze.ext.ChatBotModelGrpc.ChatBotModelBlockingStub;
-import com.google.idea.blaze.ext.CitcOperationsServiceGrpc.CitcOperationsServiceFutureStub;
+import com.google.idea.blaze.ext.CitcOperationsServiceGrpc.CitcOperationsServiceBlockingStub;
 import com.google.idea.blaze.ext.CodeSearchGrpc.CodeSearchFutureStub;
 import com.google.idea.blaze.ext.CritiqueServiceGrpc.CritiqueServiceBlockingStub;
 import com.google.idea.blaze.ext.DepServerGrpc.DepServerFutureStub;
@@ -210,7 +210,7 @@ public CritiqueServiceBlockingStub getCritiqueService() throws IOException {
     return client.getCritiqueService();
   }
 
-  public CitcOperationsServiceFutureStub getCitcOperationsService() throws IOException {
+  public CitcOperationsServiceBlockingStub getCitcOperationsService() throws IOException {
     IntelliJExtBlockingStub unused = connect();
     return client.getCitcOperationsService();
   }