X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=renderer%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Ftransportpce%2Frenderer%2Fprovisiondevice%2Ftasks%2FOlmPowerSetupRollbackTask.java;h=2555fb0319a0be23f9a1b6966c4ddc8e3186bc52;hb=b1b3bafd549bb501937cea5c976d5344608b6ed3;hp=fbd70782e1bc1c62919fbcd1ac7e1cb3a29984c4;hpb=e36b6989280be0cf36416ed1b32e583aaf226994;p=transportpce.git

diff --git a/renderer/src/main/java/org/opendaylight/transportpce/renderer/provisiondevice/tasks/OlmPowerSetupRollbackTask.java b/renderer/src/main/java/org/opendaylight/transportpce/renderer/provisiondevice/tasks/OlmPowerSetupRollbackTask.java
index fbd70782e..2555fb031 100644
--- a/renderer/src/main/java/org/opendaylight/transportpce/renderer/provisiondevice/tasks/OlmPowerSetupRollbackTask.java
+++ b/renderer/src/main/java/org/opendaylight/transportpce/renderer/provisiondevice/tasks/OlmPowerSetupRollbackTask.java
@@ -9,27 +9,27 @@ package org.opendaylight.transportpce.renderer.provisiondevice.tasks;
 
 import java.util.concurrent.Future;
 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerSetupInput;
+import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerTurndown;
 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerTurndownInput;
 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerTurndownInputBuilder;
 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerTurndownOutput;
-import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.TransportpceOlmService;
 import org.opendaylight.yangtools.yang.common.RpcResult;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 public class OlmPowerSetupRollbackTask extends RollbackTask {
-
     private static final Logger LOG = LoggerFactory.getLogger(OlmPowerSetupRollbackTask.class);
+
     private static final String FAILED = "Failed";
     private final boolean isRollbackNecessary;
-    private final TransportpceOlmService olmService;
+    private final ServicePowerTurndown servicePowerTurndown;
     private final ServicePowerSetupInput powerSetupInput;
 
-    public OlmPowerSetupRollbackTask(String id, boolean isRollbackNecessary, TransportpceOlmService olmService,
-                                     ServicePowerSetupInput powerSetupInput) {
+    public OlmPowerSetupRollbackTask(String id, boolean isRollbackNecessary, ServicePowerTurndown servicePowerTurndown,
+            ServicePowerSetupInput powerSetupInput) {
         super(id);
         this.isRollbackNecessary = isRollbackNecessary;
-        this.olmService = olmService;
+        this.servicePowerTurndown = servicePowerTurndown;
         this.powerSetupInput = powerSetupInput;
     }
 
@@ -49,7 +49,7 @@ public class OlmPowerSetupRollbackTask extends RollbackTask {
                 .build();
 
         Future<RpcResult<ServicePowerTurndownOutput>> powerTurndownResultFuture =
-                this.olmService.servicePowerTurndown(powerTurndownInput);
+                servicePowerTurndown.invoke(powerTurndownInput);
         RpcResult<ServicePowerTurndownOutput> powerTurndownResult = powerTurndownResultFuture.get();
         if (FAILED.equals(powerTurndownResult.getResult().getResult())) {
             LOG.warn("Olmp power setup rollback for {} was not successful!", this.getId());