| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you 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 |
| // |
| // http://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 org.apache.cloudstack.api.command.user.loadbalancer; |
| |
| import java.util.List; |
| import java.util.HashMap; |
| import java.util.Map; |
| import java.util.Collection; |
| import java.util.Iterator; |
| import java.util.ArrayList; |
| |
| import com.cloud.vm.VirtualMachine; |
| |
| import org.apache.cloudstack.api.APICommand; |
| import org.apache.cloudstack.api.ApiConstants; |
| import org.apache.cloudstack.api.ApiErrorCode; |
| import org.apache.cloudstack.api.BaseAsyncCmd; |
| import org.apache.cloudstack.api.Parameter; |
| import org.apache.cloudstack.api.ServerApiException; |
| import org.apache.cloudstack.api.response.FirewallRuleResponse; |
| import org.apache.cloudstack.api.response.SuccessResponse; |
| import org.apache.cloudstack.api.response.UserVmResponse; |
| import org.apache.cloudstack.context.CallContext; |
| import org.apache.commons.lang3.StringUtils; |
| |
| import com.cloud.event.EventTypes; |
| import com.cloud.exception.InvalidParameterValueException; |
| import com.cloud.network.rules.LoadBalancer; |
| import com.cloud.user.Account; |
| |
| @APICommand(name = "removeFromLoadBalancerRule", |
| description = "Removes a virtual machine or a list of virtual machines from a load balancer rule.", |
| responseObject = SuccessResponse.class, |
| requestHasSensitiveInfo = false, |
| responseHasSensitiveInfo = false) |
| public class RemoveFromLoadBalancerRuleCmd extends BaseAsyncCmd { |
| |
| |
| ///////////////////////////////////////////////////// |
| //////////////// API parameters ///////////////////// |
| ///////////////////////////////////////////////////// |
| |
| @Parameter(name = ApiConstants.ID, |
| type = CommandType.UUID, |
| entityType = FirewallRuleResponse.class, |
| required = true, |
| description = "The ID of the load balancer rule") |
| private Long id; |
| |
| @Parameter(name = ApiConstants.VIRTUAL_MACHINE_IDS, |
| type = CommandType.LIST, |
| collectionType = CommandType.UUID, |
| entityType = UserVmResponse.class, |
| description = "the list of IDs of the virtual machines that are being removed from the load balancer rule (i.e. virtualMachineIds=1,2,3)") |
| private List<Long> virtualMachineIds; |
| |
| @Parameter(name = ApiConstants.VIRTUAL_MACHINE_ID_IP, |
| type = CommandType.MAP, |
| description = "VM ID and IP map, vmidipmap[0].vmid=1 vmidipmap[0].ip=10.1.1.75", |
| since = "4.4") |
| private Map vmIdIpMap; |
| |
| ///////////////////////////////////////////////////// |
| /////////////////// Accessors /////////////////////// |
| ///////////////////////////////////////////////////// |
| |
| public Long getId() { |
| return id; |
| } |
| |
| public List<Long> getVirtualMachineIds() { |
| return virtualMachineIds; |
| } |
| |
| public Map<Long, String> getVmIdIpMap() { |
| return vmIdIpMap; |
| } |
| |
| ///////////////////////////////////////////////////// |
| /////////////// API Implementation/////////////////// |
| ///////////////////////////////////////////////////// |
| |
| @Override |
| public long getEntityOwnerId() { |
| LoadBalancer lb = _entityMgr.findById(LoadBalancer.class, getId()); |
| if (lb == null) { |
| return Account.ACCOUNT_ID_SYSTEM; // bad id given, parent this command to SYSTEM so ERROR events are tracked |
| } |
| return lb.getAccountId(); |
| } |
| |
| @Override |
| public String getEventType() { |
| return EventTypes.EVENT_REMOVE_FROM_LOAD_BALANCER_RULE; |
| } |
| |
| |
| public Map<Long, List<String>> getVmIdIpListMap() { |
| Map<Long, List<String>> vmIdIpsMap = new HashMap<Long, List<String>>(); |
| if (vmIdIpMap != null && !vmIdIpMap.isEmpty()) { |
| Collection idIpsCollection = vmIdIpMap.values(); |
| Iterator iter = idIpsCollection.iterator(); |
| while (iter.hasNext()) { |
| HashMap<String, String> idIpsMap = (HashMap<String, String>)iter.next(); |
| String vmId = idIpsMap.get("vmid"); |
| String vmIp = idIpsMap.get("vmip"); |
| |
| 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<String> ipsList = null; |
| if (vmIdIpsMap.containsKey(longVmId)) { |
| ipsList = vmIdIpsMap.get(longVmId); |
| } else { |
| ipsList = new ArrayList<String>(); |
| } |
| ipsList.add(vmIp); |
| vmIdIpsMap.put(longVmId, ipsList); |
| |
| } |
| } |
| |
| return vmIdIpsMap; |
| } |
| |
| @Override |
| public String getEventDescription() { |
| return "removing instances from load balancer: " + getId() + " (ids: " + StringUtils.join(getVirtualMachineIds(), ",") + ")"; |
| } |
| |
| @Override |
| public void execute() { |
| CallContext.current().setEventDetails("Load balancer Id: " + getId() + " VmIds: " + StringUtils.join(getVirtualMachineIds(), ",")); |
| Map<Long, List<String>> vmIdIpsMap = getVmIdIpListMap(); |
| try { |
| boolean result = _lbService.removeFromLoadBalancer(id, virtualMachineIds, vmIdIpsMap, false); |
| if (result) { |
| SuccessResponse response = new SuccessResponse(getCommandName()); |
| this.setResponseObject(response); |
| } else { |
| throw new ServerApiException(ApiErrorCode.INTERNAL_ERROR, "Failed to remove instance from load balancer rule"); |
| } |
| }catch (InvalidParameterValueException ex) { |
| throw new ServerApiException(ApiErrorCode.PARAM_ERROR, "Failed to remove instance from load balancer rule"); |
| } |
| } |
| |
| @Override |
| public String getSyncObjType() { |
| return BaseAsyncCmd.networkSyncObject; |
| } |
| |
| @Override |
| public Long getSyncObjId() { |
| LoadBalancer lb = _lbService.findById(id); |
| if (lb == null) { |
| throw new InvalidParameterValueException("Unable to find load balancer rule: " + id); |
| } |
| return lb.getNetworkId(); |
| } |
| } |