From 78a29c0d4110d2c4560c1b7a74213d983de6193c Mon Sep 17 00:00:00 2001 From: Jayapal Date: Mon, 7 Apr 2014 15:57:00 +0530 Subject: [PATCH] CLOUDSTACK-6327 updated uuid to id mapping --- .../user/loadbalancer/AssignToLoadBalancerRuleCmd.java | 7 ++++++- .../user/loadbalancer/RemoveFromLoadBalancerRuleCmd.java | 8 +++++++- .../cloud/network/lb/LoadBalancingRulesManagerImpl.java | 2 +- 3 files changed, 14 insertions(+), 3 deletions(-) diff --git a/api/src/org/apache/cloudstack/api/command/user/loadbalancer/AssignToLoadBalancerRuleCmd.java b/api/src/org/apache/cloudstack/api/command/user/loadbalancer/AssignToLoadBalancerRuleCmd.java index eb26ce652a6..a5199937f4e 100644 --- a/api/src/org/apache/cloudstack/api/command/user/loadbalancer/AssignToLoadBalancerRuleCmd.java +++ b/api/src/org/apache/cloudstack/api/command/user/loadbalancer/AssignToLoadBalancerRuleCmd.java @@ -137,7 +137,12 @@ public class AssignToLoadBalancerRuleCmd extends BaseAsyncCmd { String vmId = idIpsMap.get("vmid"); String vmIp = idIpsMap.get("vmip"); - Long longVmId = new Long(vmId); + VirtualMachine lbvm = _entityMgr.findByUuid(VirtualMachine.class, vmId); + if (lbvm == null) { + throw new InvalidParameterValueException("Unable to find virtual machine ID: " + vmId); + } + + Long longVmId = lbvm.getId(); List ipsList = null; if (vmIdIpsMap.containsKey(longVmId)) { diff --git a/api/src/org/apache/cloudstack/api/command/user/loadbalancer/RemoveFromLoadBalancerRuleCmd.java b/api/src/org/apache/cloudstack/api/command/user/loadbalancer/RemoveFromLoadBalancerRuleCmd.java index b3eb159ebfc..13765563fd9 100644 --- a/api/src/org/apache/cloudstack/api/command/user/loadbalancer/RemoveFromLoadBalancerRuleCmd.java +++ b/api/src/org/apache/cloudstack/api/command/user/loadbalancer/RemoveFromLoadBalancerRuleCmd.java @@ -23,6 +23,7 @@ import java.util.Collection; import java.util.Iterator; import java.util.ArrayList; +import com.cloud.vm.VirtualMachine; import org.apache.log4j.Logger; import org.apache.cloudstack.api.APICommand; @@ -126,7 +127,12 @@ public class RemoveFromLoadBalancerRuleCmd extends BaseAsyncCmd { String vmId = idIpsMap.get("vmid"); String vmIp = idIpsMap.get("vmip"); - Long longVmId = new Long(vmId); + VirtualMachine lbvm = _entityMgr.findByUuid(VirtualMachine.class, vmId); + if (lbvm == null) { + throw new InvalidParameterValueException("Unable to find virtual machine ID: " + vmId); + } + + Long longVmId = lbvm.getId(); List ipsList = null; if (vmIdIpsMap.containsKey(longVmId)) { diff --git a/server/src/com/cloud/network/lb/LoadBalancingRulesManagerImpl.java b/server/src/com/cloud/network/lb/LoadBalancingRulesManagerImpl.java index 51c36b6e530..db2817fc57c 100755 --- a/server/src/com/cloud/network/lb/LoadBalancingRulesManagerImpl.java +++ b/server/src/com/cloud/network/lb/LoadBalancingRulesManagerImpl.java @@ -1382,7 +1382,7 @@ public class LoadBalancingRulesManagerImpl extends ManagerBase implements s_logger.warn("Unable to apply the load balancer config because resource is unavaliable.", e); } if (!success) { - CloudRuntimeException ex = new CloudRuntimeException("Failed to remove specified load balancer rule id for vms " + instanceIds); + CloudRuntimeException ex = new CloudRuntimeException("Failed to remove specified load balancer rule id for vms " + vmIds); ex.addProxyObject(loadBalancer.getUuid(), "loadBalancerId"); throw ex; }