| // 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 com.cloud.storage.swift; |
| |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.Date; |
| import java.util.List; |
| import java.util.Map; |
| |
| import javax.ejb.Local; |
| import javax.inject.Inject; |
| import javax.naming.ConfigurationException; |
| |
| import org.apache.cloudstack.api.command.admin.swift.ListSwiftsCmd; |
| import org.apache.cloudstack.api.command.user.iso.DeleteIsoCmd; |
| import org.apache.cloudstack.api.command.user.template.DeleteTemplateCmd; |
| import org.apache.log4j.Logger; |
| import org.springframework.stereotype.Component; |
| |
| import com.cloud.agent.AgentManager; |
| import com.cloud.agent.api.Answer; |
| import com.cloud.agent.api.DeleteObjectFromSwiftCommand; |
| import com.cloud.agent.api.to.SwiftTO; |
| import org.apache.cloudstack.api.command.admin.swift.AddSwiftCmd; |
| import com.cloud.configuration.Config; |
| import com.cloud.configuration.dao.ConfigurationDao; |
| import com.cloud.dc.DataCenterVO; |
| import com.cloud.dc.dao.DataCenterDao; |
| import com.cloud.exception.DiscoveryException; |
| import com.cloud.host.HostVO; |
| import com.cloud.host.dao.HostDao; |
| import com.cloud.storage.SwiftVO; |
| import com.cloud.storage.VMTemplateHostVO; |
| import com.cloud.storage.VMTemplateSwiftVO; |
| import com.cloud.storage.VMTemplateZoneVO; |
| import com.cloud.storage.dao.SwiftDao; |
| import com.cloud.storage.dao.VMTemplateHostDao; |
| import com.cloud.storage.dao.VMTemplateSwiftDao; |
| import com.cloud.storage.dao.VMTemplateZoneDao; |
| import com.cloud.utils.Pair; |
| import com.cloud.utils.component.ManagerBase; |
| import com.cloud.utils.db.Filter; |
| import com.cloud.utils.db.SearchCriteria; |
| import com.cloud.utils.db.SearchCriteria.Op; |
| import com.cloud.utils.db.SearchCriteria2; |
| import com.cloud.utils.db.SearchCriteriaService; |
| import com.cloud.utils.exception.CloudRuntimeException; |
| |
| @Component |
| @Local(value = { SwiftManager.class }) |
| public class SwiftManagerImpl extends ManagerBase implements SwiftManager { |
| private static final Logger s_logger = Logger.getLogger(SwiftManagerImpl.class); |
| |
| @Inject |
| private SwiftDao _swiftDao; |
| @Inject |
| VMTemplateSwiftDao _vmTmpltSwiftlDao; |
| @Inject |
| private ConfigurationDao _configDao; |
| @Inject |
| private AgentManager _agentMgr; |
| @Inject |
| private DataCenterDao _dcDao; |
| @Inject |
| private VMTemplateZoneDao _vmTmpltZoneDao; |
| @Inject |
| private VMTemplateHostDao _vmTmpltHostDao; |
| @Inject |
| private HostDao _hostDao; |
| |
| @Override |
| public SwiftTO getSwiftTO(Long swiftId) { |
| return _swiftDao.getSwiftTO(swiftId); |
| } |
| |
| @Override |
| public SwiftTO getSwiftTO() { |
| return _swiftDao.getSwiftTO(null); |
| } |
| |
| @Override |
| public boolean isSwiftEnabled() { |
| Boolean swiftEnable = Boolean.valueOf(_configDao.getValue(Config.SwiftEnable.key())); |
| if (swiftEnable) { |
| return true; |
| } |
| return false; |
| } |
| |
| @Override |
| public boolean isTemplateInstalled(Long templateId) { |
| |
| SearchCriteriaService<VMTemplateSwiftVO, VMTemplateSwiftVO> sc = SearchCriteria2.create(VMTemplateSwiftVO.class); |
| sc.addAnd(sc.getEntity().getTemplateId(), Op.EQ, templateId); |
| return !sc.list().isEmpty(); |
| } |
| |
| @Override |
| public SwiftVO addSwift(AddSwiftCmd cmd) throws DiscoveryException { |
| if (!isSwiftEnabled()) { |
| throw new DiscoveryException("Swift is not enabled"); |
| } |
| SwiftVO swift = new SwiftVO(cmd.getUrl(), cmd.getAccount(), cmd.getUsername(), cmd.getKey()); |
| swift = _swiftDao.persist(swift); |
| return swift; |
| } |
| |
| @Override |
| public boolean start() { |
| if (s_logger.isInfoEnabled()) { |
| s_logger.info("Start Swift Manager"); |
| } |
| |
| return true; |
| } |
| |
| @Override |
| public void deleteIso(DeleteIsoCmd cmd) { |
| String msg; |
| SwiftTO swift = getSwiftTO(); |
| if (swift == null) { |
| msg = "There is no Swift in this setup"; |
| s_logger.warn(msg); |
| throw new CloudRuntimeException(msg); |
| } |
| VMTemplateSwiftVO tmpltSwiftRef = _vmTmpltSwiftlDao.findBySwiftTemplate(swift.getId(), cmd.getId()); |
| if ( tmpltSwiftRef == null ) { |
| msg = "Delete ISO failed due to cannot find ISO " + cmd.getId() + " in Swift "; |
| s_logger.warn(msg); |
| throw new CloudRuntimeException(msg); |
| } |
| Answer answer = _agentMgr.sendToSSVM(null, new DeleteObjectFromSwiftCommand(swift, "T-" + cmd.getId(), null)); |
| if (answer == null || !answer.getResult()) { |
| msg = "Failed to delete " + tmpltSwiftRef + " due to " + ((answer == null) ? "answer is null" : answer.getDetails()); |
| s_logger.warn(msg); |
| throw new CloudRuntimeException(msg); |
| } else { |
| _vmTmpltSwiftlDao.remove(tmpltSwiftRef.getId()); |
| s_logger.debug("Deleted template " + cmd.getId() + " in Swift"); |
| } |
| } |
| |
| @Override |
| public void deleteTemplate(DeleteTemplateCmd cmd) { |
| String msg; |
| SwiftTO swift = getSwiftTO(); |
| if (swift == null) { |
| msg = "There is no Swift in this setup"; |
| s_logger.warn(msg); |
| throw new CloudRuntimeException(msg); |
| } |
| VMTemplateSwiftVO tmpltSwiftRef = _vmTmpltSwiftlDao.findBySwiftTemplate(swift.getId(), cmd.getId()); |
| if (tmpltSwiftRef == null) { |
| msg = "Delete Template failed due to cannot find Template" + cmd.getId() + " in Swift "; |
| s_logger.warn(msg); |
| throw new CloudRuntimeException(msg); |
| } |
| Answer answer = _agentMgr.sendToSSVM(null, new DeleteObjectFromSwiftCommand(swift, "T-" + cmd.getId(), null)); |
| if (answer == null || !answer.getResult()) { |
| msg = "Failed to delete " + tmpltSwiftRef + " due to " + ((answer == null) ? "answer is null" : answer.getDetails()); |
| s_logger.warn(msg); |
| throw new CloudRuntimeException(msg); |
| } else { |
| _vmTmpltSwiftlDao.remove(tmpltSwiftRef.getId()); |
| s_logger.debug("Deleted template " + cmd.getId() + " in Swift"); |
| } |
| } |
| |
| @Override |
| public void propagateTemplateOnAllZones(Long tmpltId) { |
| String msg; |
| SwiftTO swift = getSwiftTO(); |
| if (swift == null) { |
| msg = "There is no Swift in this setup"; |
| s_logger.trace(msg); |
| return; |
| } |
| VMTemplateSwiftVO tmpltSwiftRef = _vmTmpltSwiftlDao.findOneByTemplateId(tmpltId); |
| if (tmpltSwiftRef != null) { |
| List<DataCenterVO> dcs = _dcDao.listAll(); |
| for (DataCenterVO dc : dcs) { |
| VMTemplateZoneVO tmpltZoneVO = new VMTemplateZoneVO(dc.getId(), tmpltId, new Date()); |
| try { |
| _vmTmpltZoneDao.persist(tmpltZoneVO); |
| } catch (Exception e) { |
| } |
| } |
| } |
| } |
| |
| @Override |
| public void propagateSwiftTmplteOnZone(Long zoneId) { |
| String msg; |
| SwiftTO swift = getSwiftTO(); |
| if (swift == null) { |
| msg = "There is no Swift in this setup"; |
| s_logger.trace(msg); |
| return; |
| } |
| List<Long> tmpltIds = new ArrayList<Long>(); |
| List<VMTemplateSwiftVO> tmpltSwiftRefs = _vmTmpltSwiftlDao.listAll(); |
| if (tmpltSwiftRefs == null) { |
| return; |
| } |
| for (VMTemplateSwiftVO tmpltSwiftRef : tmpltSwiftRefs) { |
| Long tmpltId = tmpltSwiftRef.getTemplateId(); |
| if (!tmpltIds.contains(tmpltId)) { |
| tmpltIds.add(tmpltId); |
| VMTemplateZoneVO tmpltZoneVO = new VMTemplateZoneVO(zoneId, tmpltId, new Date()); |
| try { |
| _vmTmpltZoneDao.persist(tmpltZoneVO); |
| } catch (Exception e) { |
| } |
| } |
| } |
| } |
| |
| @Override |
| public Long chooseZoneForTmpltExtract(Long tmpltId) { |
| SwiftTO swift = getSwiftTO(); |
| if (swift == null) { |
| return null; |
| } |
| |
| List<VMTemplateHostVO> tmpltHosts = _vmTmpltHostDao.listByOnlyTemplateId(tmpltId); |
| if (tmpltHosts != null) { |
| Collections.shuffle(tmpltHosts); |
| for (VMTemplateHostVO tHost : tmpltHosts) { |
| HostVO host = _hostDao.findById(tHost.getHostId()); |
| if (host != null) { |
| return host.getDataCenterId(); |
| } |
| throw new CloudRuntimeException("can not find secondary storage host"); |
| } |
| } |
| List<DataCenterVO> dcs = _dcDao.listAll(); |
| Collections.shuffle(dcs); |
| return dcs.get(0).getId(); |
| } |
| |
| @Override |
| public Pair<List<SwiftVO>, Integer> listSwifts(ListSwiftsCmd cmd) { |
| Filter searchFilter = new Filter(SwiftVO.class, "id", Boolean.TRUE, cmd.getStartIndex(), cmd.getPageSizeVal()); |
| SearchCriteria<SwiftVO> sc = _swiftDao.createSearchCriteria(); |
| if (cmd.getId() != null) { |
| sc.addAnd("id", SearchCriteria.Op.EQ, cmd.getId()); |
| } |
| return _swiftDao.searchAndCount(sc, searchFilter); |
| |
| } |
| |
| @Override |
| public VMTemplateSwiftVO findByTmpltId(Long tmpltId) { |
| return _vmTmpltSwiftlDao.findOneByTemplateId(tmpltId); |
| } |
| |
| @Override |
| public boolean stop() { |
| if (s_logger.isInfoEnabled()) { |
| s_logger.info("Stop Swift Manager"); |
| } |
| return true; |
| } |
| |
| @Override |
| public boolean configure(String name, Map<String, Object> params) throws ConfigurationException { |
| if (s_logger.isInfoEnabled()) { |
| s_logger.info("Start configuring Swift Manager : " + name); |
| } |
| |
| return true; |
| } |
| |
| protected SwiftManagerImpl() { |
| } |
| } |
