This CL implements the StopLabServer backend API for FE v6.

PiperOrigin-RevId: 925254824

Author: OmniLab Team

src/java/com/google/devtools/mobileharness/fe/v6/service/host/HostServiceLogicImpl.java

@@ -28,6 +28,9 @@
 import com.google.devtools.mobileharness.fe.v6.service.host.handlers.PreflightLabServerReleaseHandler;
 import com.google.devtools.mobileharness.fe.v6.service.host.handlers.ReleaseLabServerHandler;
 import com.google.devtools.mobileharness.fe.v6.service.host.handlers.RemoteControlDevicesHandler;
+import com.google.devtools.mobileharness.fe.v6.service.host.handlers.RestartLabServerHandler;
+import com.google.devtools.mobileharness.fe.v6.service.host.handlers.StartLabServerHandler;
+import com.google.devtools.mobileharness.fe.v6.service.host.handlers.StopLabServerHandler;
 import com.google.devtools.mobileharness.fe.v6.service.host.handlers.UpdatePassThroughFlagsHandler;
 import com.google.devtools.mobileharness.fe.v6.service.proto.host.CheckRemoteControlEligibilityRequest;
 import com.google.devtools.mobileharness.fe.v6.service.proto.host.CheckRemoteControlEligibilityResponse;
@@ -81,6 +84,9 @@ public final class HostServiceLogicImpl implements HostServiceLogic {
   private final DecommissionMissingDevicesHandler decommissionMissingDevicesHandler;
   private final PreflightLabServerReleaseHandler preflightLabServerReleaseHandler;
   private final ReleaseLabServerHandler releaseLabServerHandler;
+  private final StartLabServerHandler startLabServerHandler;
+  private final RestartLabServerHandler restartLabServerHandler;
+  private final StopLabServerHandler stopLabServerHandler;
   private final UpdatePassThroughFlagsHandler updatePassThroughFlagsHandler;
   private final UniverseFactory universeFactory;
 
@@ -94,6 +100,9 @@ public final class HostServiceLogicImpl implements HostServiceLogic {
       DecommissionMissingDevicesHandler decommissionMissingDevicesHandler,
       PreflightLabServerReleaseHandler preflightLabServerReleaseHandler,
       ReleaseLabServerHandler releaseLabServerHandler,
+      StartLabServerHandler startLabServerHandler,
+      RestartLabServerHandler restartLabServerHandler,
+      StopLabServerHandler stopLabServerHandler,
       UpdatePassThroughFlagsHandler updatePassThroughFlagsHandler,
       UniverseFactory universeFactory) {
     this.getHostOverviewHandler = getHostOverviewHandler;
@@ -104,6 +113,9 @@ public final class HostServiceLogicImpl implements HostServiceLogic {
     this.decommissionMissingDevicesHandler = decommissionMissingDevicesHandler;
     this.preflightLabServerReleaseHandler = preflightLabServerReleaseHandler;
     this.releaseLabServerHandler = releaseLabServerHandler;
+    this.startLabServerHandler = startLabServerHandler;
+    this.restartLabServerHandler = restartLabServerHandler;
+    this.stopLabServerHandler = stopLabServerHandler;
     this.updatePassThroughFlagsHandler = updatePassThroughFlagsHandler;
     this.universeFactory = universeFactory;
   }
@@ -247,30 +259,36 @@ public ListenableFuture<ReleaseLabServerResponse> releaseLabServer(
 
   @Override
   public ListenableFuture<StartLabServerResponse> startLabServer(StartLabServerRequest request) {
-    // TODO: Use the universe parameter.
-    @SuppressWarnings("unused")
-    String universe = request.getUniverse();
-    // TODO: Implement this method.
-    return immediateFuture(StartLabServerResponse.getDefaultInstance());
+    UniverseScope universe;
+    try {
+      universe = universeFactory.create(request.getUniverse());
+    } catch (IllegalArgumentException e) {
+      return immediateFailedFuture(e);
+    }
+    return startLabServerHandler.startLabServer(request, universe);
   }
 
   @Override
   public ListenableFuture<RestartLabServerResponse> restartLabServer(
       RestartLabServerRequest request) {
-    // TODO: Use the universe parameter.
-    @SuppressWarnings("unused")
-    String universe = request.getUniverse();
-    // TODO: Implement this method.
-    return immediateFuture(RestartLabServerResponse.getDefaultInstance());
+    UniverseScope universe;
+    try {
+      universe = universeFactory.create(request.getUniverse());
+    } catch (IllegalArgumentException e) {
+      return immediateFailedFuture(e);
+    }
+    return restartLabServerHandler.restartLabServer(request, universe);
   }
 
   @Override
   public ListenableFuture<StopLabServerResponse> stopLabServer(StopLabServerRequest request) {
-    // TODO: Use the universe parameter.
-    @SuppressWarnings("unused")
-    String universe = request.getUniverse();
-    // TODO: Implement this method.
-    return immediateFuture(StopLabServerResponse.getDefaultInstance());
+    UniverseScope universe;
+    try {
+      universe = universeFactory.create(request.getUniverse());
+    } catch (IllegalArgumentException e) {
+      return immediateFailedFuture(e);
+    }
+    return stopLabServerHandler.stopLabServer(request, universe);
   }
 
   @Override

src/java/com/google/devtools/mobileharness/fe/v6/service/host/handlers/LabServerVersionResolver.java

@@ -0,0 +1,110 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.host.handlers;
+
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.ListeningExecutorService;
+import com.google.devtools.mobileharness.api.query.proto.FilterProto.LabFilter;
+import com.google.devtools.mobileharness.api.query.proto.FilterProto.StringMatchCondition;
+import com.google.devtools.mobileharness.api.query.proto.FilterProto.StringMatchCondition.Include;
+import com.google.devtools.mobileharness.api.query.proto.LabQueryProto.LabData;
+import com.google.devtools.mobileharness.api.query.proto.LabQueryProto.LabInfo;
+import com.google.devtools.mobileharness.api.query.proto.LabQueryProto.LabQuery;
+import com.google.devtools.mobileharness.api.query.proto.LabQueryProto.LabQuery.Filter;
+import com.google.devtools.mobileharness.api.query.proto.LabQueryProto.LabQuery.LabViewRequest;
+import com.google.devtools.mobileharness.fe.v6.service.host.provider.HostAuxiliaryInfoProvider;
+import com.google.devtools.mobileharness.fe.v6.service.host.provider.HostReleaseInfo;
+import com.google.devtools.mobileharness.fe.v6.service.host.util.HostVersionUtil;
+import com.google.devtools.mobileharness.fe.v6.service.shared.providers.LabInfoProvider;
+import com.google.devtools.mobileharness.fe.v6.service.util.UniverseScope;
+import com.google.devtools.mobileharness.shared.labinfo.proto.LabInfoServiceProto.GetLabInfoRequest;
+import com.google.devtools.mobileharness.shared.labinfo.proto.LabInfoServiceProto.GetLabInfoResponse;
+import java.util.Optional;
+import javax.inject.Inject;
+import javax.inject.Singleton;
+
+/** Resolves the current lab server version for a host. */
+@Singleton
+public class LabServerVersionResolver {
+
+  private final LabInfoProvider labInfoProvider;
+  private final HostAuxiliaryInfoProvider hostAuxiliaryInfoProvider;
+  private final ListeningExecutorService executor;
+
+  @Inject
+  LabServerVersionResolver(
+      LabInfoProvider labInfoProvider,
+      HostAuxiliaryInfoProvider hostAuxiliaryInfoProvider,
+      ListeningExecutorService executor) {
+    this.labInfoProvider = labInfoProvider;
+    this.hostAuxiliaryInfoProvider = hostAuxiliaryInfoProvider;
+    this.executor = executor;
+  }
+
+  /** Resolves the current version of the lab server running on the specified host. */
+  public ListenableFuture<String> resolveCurrentVersion(String hostName, UniverseScope universe) {
+    ListenableFuture<GetLabInfoResponse> labInfoFuture =
+        labInfoProvider.getLabInfoAsync(createGetLabInfoRequest(hostName), universe);
+
+    ListenableFuture<Optional<HostReleaseInfo>> hostReleaseInfoFuture =
+        hostAuxiliaryInfoProvider.getHostReleaseInfo(hostName, universe);
+
+    return Futures.whenAllSucceed(labInfoFuture, hostReleaseInfoFuture)
+        .call(
+            () -> {
+              GetLabInfoResponse labInfoResponse = Futures.getDone(labInfoFuture);
+              Optional<HostReleaseInfo> hostReleaseInfoOpt = Futures.getDone(hostReleaseInfoFuture);
+
+              Optional<LabInfo> labInfoOpt =
+                  labInfoResponse.getLabQueryResult().getLabView().getLabDataList().stream()
+                      .map(LabData::getLabInfo)
+                      .findFirst();
+
+              Optional<String> currentVersionOpt =
+                  HostVersionUtil.resolveCurrentVersion(labInfoOpt, hostReleaseInfoOpt);
+
+              return currentVersionOpt.orElseThrow(
+                  () ->
+                      new IllegalStateException(
+                          "Failed to resolve current lab server version for host: " + hostName));
+            },
+            executor);
+  }
+
+  private GetLabInfoRequest createGetLabInfoRequest(String hostName) {
+    return GetLabInfoRequest.newBuilder()
+        .setLabQuery(
+            LabQuery.newBuilder()
+                .setFilter(
+                    Filter.newBuilder()
+                        .setLabFilter(
+                            LabFilter.newBuilder()
+                                .addLabMatchCondition(
+                                    LabFilter.LabMatchCondition.newBuilder()
+                                        .setLabHostNameMatchCondition(
+                                            LabFilter.LabMatchCondition.LabHostNameMatchCondition
+                                                .newBuilder()
+                                                .setCondition(
+                                                    StringMatchCondition.newBuilder()
+                                                        .setInclude(
+                                                            Include.newBuilder()
+                                                                .addExpected(hostName)))))))
+                .setLabViewRequest(LabViewRequest.getDefaultInstance()))
+        .build();
+  }
+}

src/java/com/google/devtools/mobileharness/fe/v6/service/host/handlers/NoOpRestartLabServerActionHelper.java

@@ -0,0 +1,38 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.host.handlers;
+
+import static com.google.common.util.concurrent.Futures.immediateFuture;
+
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.RestartLabServerRequest;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.RestartLabServerResponse;
+import com.google.devtools.mobileharness.fe.v6.service.util.UniverseScope;
+import javax.inject.Inject;
+
+/** No-op implementation of {@link RestartLabServerActionHelper}. */
+public final class NoOpRestartLabServerActionHelper implements RestartLabServerActionHelper {
+
+  @Inject
+  NoOpRestartLabServerActionHelper() {}
+
+  @Override
+  public ListenableFuture<RestartLabServerResponse> restartLabServer(
+      RestartLabServerRequest request, UniverseScope universe) {
+    return immediateFuture(RestartLabServerResponse.getDefaultInstance());
+  }
+}

src/java/com/google/devtools/mobileharness/fe/v6/service/host/handlers/NoOpStartLabServerActionHelper.java

@@ -0,0 +1,38 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.host.handlers;
+
+import static com.google.common.util.concurrent.Futures.immediateFuture;
+
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.StartLabServerRequest;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.StartLabServerResponse;
+import com.google.devtools.mobileharness.fe.v6.service.util.UniverseScope;
+import javax.inject.Inject;
+
+/** No-op implementation of {@link StartLabServerActionHelper}. */
+public final class NoOpStartLabServerActionHelper implements StartLabServerActionHelper {
+
+  @Inject
+  NoOpStartLabServerActionHelper() {}
+
+  @Override
+  public ListenableFuture<StartLabServerResponse> startLabServer(
+      StartLabServerRequest request, UniverseScope universe) {
+    return immediateFuture(StartLabServerResponse.getDefaultInstance());
+  }
+}

src/java/com/google/devtools/mobileharness/fe/v6/service/host/handlers/NoOpStopLabServerActionHelper.java

@@ -0,0 +1,38 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.host.handlers;
+
+import static com.google.common.util.concurrent.Futures.immediateFuture;
+
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.StopLabServerRequest;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.StopLabServerResponse;
+import com.google.devtools.mobileharness.fe.v6.service.util.UniverseScope;
+import javax.inject.Inject;
+
+/** No-op implementation of {@link StopLabServerActionHelper}. */
+public final class NoOpStopLabServerActionHelper implements StopLabServerActionHelper {
+
+  @Inject
+  NoOpStopLabServerActionHelper() {}
+
+  @Override
+  public ListenableFuture<StopLabServerResponse> stopLabServer(
+      StopLabServerRequest request, UniverseScope universe) {
+    return immediateFuture(StopLabServerResponse.getDefaultInstance());
+  }
+}

src/java/com/google/devtools/mobileharness/fe/v6/service/host/handlers/RestartLabServerActionHelper.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.host.handlers;
+
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.RestartLabServerRequest;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.RestartLabServerResponse;
+import com.google.devtools.mobileharness.fe.v6.service.util.UniverseScope;
+
+/** Interface for helper to handle RestartLabServer RPCs. */
+public interface RestartLabServerActionHelper {
+  ListenableFuture<RestartLabServerResponse> restartLabServer(
+      RestartLabServerRequest request, UniverseScope universe);
+}

src/java/com/google/devtools/mobileharness/fe/v6/service/host/handlers/RestartLabServerHandler.java

@@ -0,0 +1,38 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.fe.v6.service.host.handlers;
+
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.RestartLabServerRequest;
+import com.google.devtools.mobileharness.fe.v6.service.proto.host.RestartLabServerResponse;
+import com.google.devtools.mobileharness.fe.v6.service.util.UniverseScope;
+import javax.inject.Inject;
+
+/** Handler for RestartLabServer RPCs. */
+public final class RestartLabServerHandler {
+  private final RestartLabServerActionHelper actionHelper;
+
+  @Inject
+  RestartLabServerHandler(RestartLabServerActionHelper actionHelper) {
+    this.actionHelper = actionHelper;
+  }
+
+  public ListenableFuture<RestartLabServerResponse> restartLabServer(
+      RestartLabServerRequest request, UniverseScope universe) {
+    return actionHelper.restartLabServer(request, universe);
+  }
+}