mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
Add store_role to template_store_ref, and add search function to find
template_store_ref entry based on DataStoreRole and ZoneId.
This commit is contained in:
parent
545d2e4eb6
commit
fa11575212
@ -18,9 +18,12 @@
|
||||
*/
|
||||
package org.apache.cloudstack.engine.subsystem.api.storage;
|
||||
|
||||
import com.cloud.storage.DataStoreRole;
|
||||
|
||||
|
||||
public interface TemplateDataFactory {
|
||||
TemplateInfo getTemplate(long templateId, DataStore store);
|
||||
TemplateInfo getTemplate(DataObject obj, DataStore store);
|
||||
TemplateInfo getTemplate(long templateId);
|
||||
TemplateInfo getTemplate(long templateId, DataStoreRole storeRole);
|
||||
TemplateInfo getTemplate(long templateId, DataStoreRole storeRole, Long zoneId);
|
||||
}
|
||||
|
||||
@ -23,6 +23,7 @@ import org.apache.cloudstack.engine.subsystem.api.storage.DataObjectInStore;
|
||||
import org.apache.cloudstack.engine.subsystem.api.storage.ObjectInDataStoreStateMachine;
|
||||
|
||||
|
||||
import com.cloud.storage.DataStoreRole;
|
||||
import com.cloud.storage.VMTemplateStorageResourceAssoc;
|
||||
import com.cloud.utils.db.GenericDao;
|
||||
import com.cloud.utils.fsm.StateDao;
|
||||
@ -50,7 +51,9 @@ public interface TemplateDataStoreDao extends GenericDao<TemplateDataStoreVO, Lo
|
||||
|
||||
TemplateDataStoreVO findByStoreTemplate(long storeId, long templateId, boolean lock);
|
||||
|
||||
TemplateDataStoreVO findByTemplate(long templateId);
|
||||
TemplateDataStoreVO findByTemplate(long templateId, DataStoreRole role);
|
||||
|
||||
TemplateDataStoreVO findByTemplateZone(long templateId, Long zoneId, DataStoreRole role);
|
||||
|
||||
List<TemplateDataStoreVO> listByTemplate(long templateId);
|
||||
}
|
||||
|
||||
@ -33,6 +33,7 @@ import org.apache.cloudstack.engine.subsystem.api.storage.DataObjectInStore;
|
||||
import org.apache.cloudstack.engine.subsystem.api.storage.ObjectInDataStoreStateMachine;
|
||||
import org.apache.cloudstack.engine.subsystem.api.storage.ObjectInDataStoreStateMachine.State;
|
||||
|
||||
import com.cloud.storage.DataStoreRole;
|
||||
import com.cloud.storage.VMTemplateStorageResourceAssoc.Status;
|
||||
|
||||
import com.cloud.utils.db.GenericDaoBase;
|
||||
@ -55,6 +56,10 @@ public class TemplateDataStoreVO implements StateObject<ObjectInDataStoreStateMa
|
||||
@Column(name="template_id")
|
||||
private long templateId;
|
||||
|
||||
@Column(name="store_role")
|
||||
@Enumerated(EnumType.STRING)
|
||||
private DataStoreRole dataStoreRole;
|
||||
|
||||
@Column(name=GenericDaoBase.CREATED_COLUMN)
|
||||
private Date created = null;
|
||||
|
||||
@ -108,6 +113,56 @@ public class TemplateDataStoreVO implements StateObject<ObjectInDataStoreStateMa
|
||||
ObjectInDataStoreStateMachine.State state;
|
||||
|
||||
|
||||
|
||||
|
||||
public TemplateDataStoreVO(Long hostId, long templateId) {
|
||||
super();
|
||||
this.dataStoreId = hostId;
|
||||
this.templateId = templateId;
|
||||
this.state = ObjectInDataStoreStateMachine.State.Allocated;
|
||||
}
|
||||
|
||||
public TemplateDataStoreVO(Long hostId, long templateId, Date lastUpdated,
|
||||
int downloadPercent, Status downloadState,
|
||||
String localDownloadPath, String errorString, String jobId,
|
||||
String installPath, String downloadUrl) {
|
||||
super();
|
||||
this.dataStoreId = hostId;
|
||||
this.templateId = templateId;
|
||||
this.lastUpdated = lastUpdated;
|
||||
this.downloadPercent = downloadPercent;
|
||||
this.downloadState = downloadState;
|
||||
this.localDownloadPath = localDownloadPath;
|
||||
this.errorString = errorString;
|
||||
this.jobId = jobId;
|
||||
this.installPath = installPath;
|
||||
this.setDownloadUrl(downloadUrl);
|
||||
switch (downloadState) {
|
||||
case DOWNLOADED:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Ready;
|
||||
break;
|
||||
case CREATING:
|
||||
case DOWNLOAD_IN_PROGRESS:
|
||||
case UPLOAD_IN_PROGRESS:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Creating2;
|
||||
break;
|
||||
case DOWNLOAD_ERROR:
|
||||
case UPLOAD_ERROR:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Failed;
|
||||
break;
|
||||
case ABANDONED:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Destroyed;
|
||||
break;
|
||||
default:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Allocated;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
public TemplateDataStoreVO() {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getInstallPath() {
|
||||
return installPath;
|
||||
@ -168,55 +223,6 @@ public class TemplateDataStoreVO implements StateObject<ObjectInDataStoreStateMa
|
||||
return downloadState;
|
||||
}
|
||||
|
||||
public TemplateDataStoreVO(Long hostId, long templateId) {
|
||||
super();
|
||||
this.dataStoreId = hostId;
|
||||
this.templateId = templateId;
|
||||
this.state = ObjectInDataStoreStateMachine.State.Allocated;
|
||||
}
|
||||
|
||||
public TemplateDataStoreVO(Long hostId, long templateId, Date lastUpdated,
|
||||
int downloadPercent, Status downloadState,
|
||||
String localDownloadPath, String errorString, String jobId,
|
||||
String installPath, String downloadUrl) {
|
||||
super();
|
||||
this.dataStoreId = hostId;
|
||||
this.templateId = templateId;
|
||||
this.lastUpdated = lastUpdated;
|
||||
this.downloadPercent = downloadPercent;
|
||||
this.downloadState = downloadState;
|
||||
this.localDownloadPath = localDownloadPath;
|
||||
this.errorString = errorString;
|
||||
this.jobId = jobId;
|
||||
this.installPath = installPath;
|
||||
this.setDownloadUrl(downloadUrl);
|
||||
switch (downloadState) {
|
||||
case DOWNLOADED:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Ready;
|
||||
break;
|
||||
case CREATING:
|
||||
case DOWNLOAD_IN_PROGRESS:
|
||||
case UPLOAD_IN_PROGRESS:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Creating2;
|
||||
break;
|
||||
case DOWNLOAD_ERROR:
|
||||
case UPLOAD_ERROR:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Failed;
|
||||
break;
|
||||
case ABANDONED:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Destroyed;
|
||||
break;
|
||||
default:
|
||||
this.state = ObjectInDataStoreStateMachine.State.Allocated;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
public TemplateDataStoreVO() {
|
||||
|
||||
}
|
||||
|
||||
|
||||
public void setLocalDownloadPath(String localPath) {
|
||||
this.localDownloadPath = localPath;
|
||||
}
|
||||
@ -345,4 +351,15 @@ public class TemplateDataStoreVO implements StateObject<ObjectInDataStoreStateMa
|
||||
public State getObjectInStoreState() {
|
||||
return this.state;
|
||||
}
|
||||
|
||||
public DataStoreRole getDataStoreRole() {
|
||||
return dataStoreRole;
|
||||
}
|
||||
|
||||
public void setDataStoreRole(DataStoreRole dataStoreRole) {
|
||||
this.dataStoreRole = dataStoreRole;
|
||||
}
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
@ -78,20 +78,31 @@ public class TemplateDataFactoryImpl implements TemplateDataFactory {
|
||||
return tmpl;
|
||||
}
|
||||
|
||||
// NOTE that this method can only be used for get template information stored in secondary storage
|
||||
//TODO: this method is problematic, since one template can be stored in multiple image stores.
|
||||
// need to see if we can get rid of this method or change to plural format, or restrict to 1:1 mapping
|
||||
|
||||
@Override
|
||||
public TemplateInfo getTemplate(long templateId) {
|
||||
public TemplateInfo getTemplate(long templateId, DataStoreRole storeRole) {
|
||||
VMTemplateVO templ = imageDataDao.findById(templateId);
|
||||
TemplateDataStoreVO tmplStore = templateStoreDao.findByTemplate(templateId);
|
||||
TemplateDataStoreVO tmplStore = templateStoreDao.findByTemplate(templateId, storeRole);
|
||||
DataStore store = null;
|
||||
if ( tmplStore != null ){
|
||||
store = this.storeMgr.getDataStore(tmplStore.getDataStoreId(), DataStoreRole.Image);
|
||||
store = this.storeMgr.getDataStore(tmplStore.getDataStoreId(), storeRole);
|
||||
}
|
||||
return this.getTemplate(templateId, store);
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public TemplateInfo getTemplate(long templateId, DataStoreRole storeRole, Long zoneId) {
|
||||
VMTemplateVO templ = imageDataDao.findById(templateId);
|
||||
TemplateDataStoreVO tmplStore = templateStoreDao.findByTemplateZone(templateId, zoneId, storeRole);
|
||||
DataStore store = null;
|
||||
if ( tmplStore != null ){
|
||||
store = this.storeMgr.getDataStore(tmplStore.getDataStoreId(), storeRole);
|
||||
}
|
||||
return this.getTemplate(templateId, store);
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public TemplateInfo getTemplate(DataObject obj, DataStore store) {
|
||||
return this.getTemplate(obj.getId(), store);
|
||||
|
||||
@ -73,6 +73,7 @@ import com.cloud.exception.ResourceAllocationException;
|
||||
import com.cloud.hypervisor.Hypervisor.HypervisorType;
|
||||
import com.cloud.storage.StoragePool;
|
||||
import com.cloud.storage.VMTemplateStorageResourceAssoc.Status;
|
||||
import com.cloud.storage.DataStoreRole;
|
||||
import com.cloud.storage.VMTemplateVO;
|
||||
import com.cloud.storage.VMTemplateZoneVO;
|
||||
import com.cloud.storage.dao.VMTemplateDao;
|
||||
@ -189,7 +190,7 @@ public class TemplateServiceImpl implements TemplateService {
|
||||
for (VMTemplateVO template : toBeDownloaded) {
|
||||
TemplateDataStoreVO tmpltHost = _vmTemplateStoreDao.findByStoreTemplate(store.getId(), template.getId());
|
||||
if (tmpltHost == null || tmpltHost.getState() != ObjectInDataStoreStateMachine.State.Ready) {
|
||||
TemplateInfo tmplt = this._templateFactory.getTemplate(template.getId());
|
||||
TemplateInfo tmplt = this._templateFactory.getTemplate(template.getId(), DataStoreRole.Image);
|
||||
this.createTemplateAsync(tmplt, store, null);
|
||||
}
|
||||
}
|
||||
@ -224,7 +225,7 @@ public class TemplateServiceImpl implements TemplateService {
|
||||
for (VMTemplateVO template: toBeDownloaded) {
|
||||
TemplateDataStoreVO tmpltHost = _vmTemplateStoreDao.findByStoreTemplate(store.getId(), template.getId());
|
||||
if (tmpltHost == null || tmpltHost.getState() != ObjectInDataStoreStateMachine.State.Ready) {
|
||||
TemplateInfo tmplt = this._templateFactory.getTemplate(template.getId());
|
||||
TemplateInfo tmplt = this._templateFactory.getTemplate(template.getId(), DataStoreRole.Image);
|
||||
this.createTemplateAsync(tmplt, store, null);
|
||||
}
|
||||
}
|
||||
@ -328,6 +329,7 @@ public class TemplateServiceImpl implements TemplateService {
|
||||
tmpltStore = new TemplateDataStoreVO(storeId, tmplt.getId(), new Date(), 100, Status.DOWNLOADED, null, null, null, tmpltInfo.getInstallPath(), tmplt.getUrl());
|
||||
tmpltStore.setSize(tmpltInfo.getSize());
|
||||
tmpltStore.setPhysicalSize(tmpltInfo.getPhysicalSize());
|
||||
tmpltStore.setDataStoreRole(store.getRole());
|
||||
_vmTemplateStoreDao.persist(tmpltStore);
|
||||
this.associateTemplateToZone(tmplt.getId(), zoneId);
|
||||
}
|
||||
@ -381,7 +383,7 @@ public class TemplateServiceImpl implements TemplateService {
|
||||
}
|
||||
s_logger.debug("Template " + tmplt.getName() + " needs to be downloaded to " + store.getName());
|
||||
//TODO: we should pass a callback here
|
||||
TemplateInfo tmpl = this._templateFactory.getTemplate(tmplt.getId());
|
||||
TemplateInfo tmpl = this._templateFactory.getTemplate(tmplt.getId(), DataStoreRole.Image);
|
||||
this.createTemplateAsync(tmpl, store, null);
|
||||
}
|
||||
}
|
||||
@ -580,6 +582,7 @@ public class TemplateServiceImpl implements TemplateService {
|
||||
tmpltStore = new TemplateDataStoreVO(storeId, tmplt.getId(), new Date(), 100, Status.DOWNLOADED, null, null, null, TemplateConstants.DEFAULT_SYSTEM_VM_TEMPLATE_PATH + tmplt.getId() + File.separator, tmplt.getUrl());
|
||||
tmpltStore.setSize(0);
|
||||
tmpltStore.setPhysicalSize(0); // no size information for pre-seeded system vm templates
|
||||
tmpltStore.setDataStoreRole(store.getRole());
|
||||
_vmTemplateStoreDao.persist(tmpltStore);
|
||||
}
|
||||
}
|
||||
|
||||
@ -137,7 +137,7 @@ public class S3TemplateTest extends CloudStackTestNGBase {
|
||||
|
||||
@Test(priority = 1)
|
||||
public void registerTemplate() {
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId);
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId, DataStoreRole.Image);
|
||||
DataStore store = dataStoreMgr.getImageStore(dcId);
|
||||
AsyncCallFuture<TemplateApiResult> future = new AsyncCallFuture<TemplateApiResult>();
|
||||
templateSvr.createTemplateAsync(template, store, future);
|
||||
@ -157,7 +157,7 @@ public class S3TemplateTest extends CloudStackTestNGBase {
|
||||
|
||||
@Test(priority = 2)
|
||||
public void copyTemplateToCache() {
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId);
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId, DataStoreRole.Image);
|
||||
DataObject cacheObj = this.cacheMgr.createCacheObject(template, new ZoneScope(dcId));
|
||||
assertNotNull(cacheObj, "failed to create cache object");
|
||||
}
|
||||
|
||||
@ -240,7 +240,7 @@ public class SnapshotTest extends CloudStackTestNGBase {
|
||||
|
||||
|
||||
DataStore store = this.dataStoreMgr.getDataStore(imageStore.getId(), DataStoreRole.Image);
|
||||
TemplateInfo template = templateFactory.getTemplate(image.getId());
|
||||
TemplateInfo template = templateFactory.getTemplate(image.getId(), DataStoreRole.Image);
|
||||
DataObject templateOnStore = store.create(template);
|
||||
TemplateObjectTO to = new TemplateObjectTO();
|
||||
to.setPath(this.getImageInstallPath());
|
||||
@ -336,7 +336,7 @@ public class SnapshotTest extends CloudStackTestNGBase {
|
||||
primaryStore = this.dataStoreMgr.getPrimaryDataStore(primaryStoreId);
|
||||
VolumeVO volume = createVolume(image.getId(), primaryStore.getId());
|
||||
VolumeInfo volInfo = this.volFactory.getVolume(volume.getId());
|
||||
AsyncCallFuture<VolumeApiResult> future = this.volumeService.createVolumeFromTemplateAsync(volInfo, this.primaryStoreId, this.templateFactory.getTemplate(this.image.getId()));
|
||||
AsyncCallFuture<VolumeApiResult> future = this.volumeService.createVolumeFromTemplateAsync(volInfo, this.primaryStoreId, this.templateFactory.getTemplate(this.image.getId(), DataStoreRole.Image));
|
||||
|
||||
VolumeApiResult result;
|
||||
try {
|
||||
@ -404,7 +404,7 @@ public class SnapshotTest extends CloudStackTestNGBase {
|
||||
ep.setResource(new MockLocalNfsSecondaryStorageResource());
|
||||
Mockito.when(epSelector.select(Mockito.any(DataObject.class), Mockito.any(DataObject.class))).thenReturn(ep);
|
||||
VMTemplateVO templateVO = createTemplateInDb();
|
||||
TemplateInfo tmpl = this.templateFactory.getTemplate(templateVO.getId());
|
||||
TemplateInfo tmpl = this.templateFactory.getTemplate(templateVO.getId(), DataStoreRole.Image);
|
||||
DataStore imageStore = this.dataStoreMgr.getImageStore(this.dcId);
|
||||
this.imageService.createTemplateFromSnapshotAsync(snapshot, tmpl, imageStore);
|
||||
}
|
||||
|
||||
@ -117,7 +117,7 @@ public class TemplateTest extends CloudStackTestNGBase {
|
||||
|
||||
@Test
|
||||
public void registerTemplate() {
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId);
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId, DataStoreRole.Image);
|
||||
DataStore store = dataStoreMgr.getImageStore(dcId);
|
||||
AsyncCallFuture<TemplateApiResult> future = new AsyncCallFuture<TemplateApiResult>();
|
||||
templateSvr.createTemplateAsync(template, store, future);
|
||||
@ -137,7 +137,7 @@ public class TemplateTest extends CloudStackTestNGBase {
|
||||
|
||||
// @Test
|
||||
public void deleteTemplate() {
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId);
|
||||
TemplateInfo template = templateFactory.getTemplate(templateId, DataStoreRole.Image);
|
||||
DataStore store = dataStoreMgr.getImageStore(dcId);
|
||||
AsyncCallFuture<TemplateApiResult> future = new AsyncCallFuture<TemplateApiResult>();
|
||||
templateSvr.deleteTemplateAsync(template);
|
||||
|
||||
@ -232,7 +232,7 @@ public class VolumeTest extends CloudStackTestNGBase {
|
||||
|
||||
|
||||
DataStore store = this.dataStoreMgr.getDataStore(imageStore.getId(), DataStoreRole.Image);
|
||||
TemplateInfo template = templateFactory.getTemplate(image.getId());
|
||||
TemplateInfo template = templateFactory.getTemplate(image.getId(), DataStoreRole.Image);
|
||||
DataObject templateOnStore = store.create(template);
|
||||
TemplateObjectTO to = new TemplateObjectTO();
|
||||
to.setPath(this.getImageInstallPath());
|
||||
@ -320,7 +320,7 @@ public class VolumeTest extends CloudStackTestNGBase {
|
||||
primaryStore = this.dataStoreMgr.getPrimaryDataStore(primaryStoreId);
|
||||
VolumeVO volume = createVolume(image.getId(), primaryStore.getId());
|
||||
VolumeInfo volInfo = this.volFactory.getVolume(volume.getId());
|
||||
AsyncCallFuture<VolumeApiResult> future = this.volumeService.createVolumeFromTemplateAsync(volInfo, this.primaryStoreId, this.templateFactory.getTemplate(this.image.getId()));
|
||||
AsyncCallFuture<VolumeApiResult> future = this.volumeService.createVolumeFromTemplateAsync(volInfo, this.primaryStoreId, this.templateFactory.getTemplate(this.image.getId(), DataStoreRole.Image));
|
||||
try {
|
||||
VolumeApiResult result = future.get();
|
||||
|
||||
@ -416,7 +416,7 @@ public class VolumeTest extends CloudStackTestNGBase {
|
||||
AssertJUnit.assertTrue(result.isSuccess());
|
||||
volInfo = result.getVolume();
|
||||
VMTemplateVO templateVO = createTemplateInDb();
|
||||
TemplateInfo tmpl = this.templateFactory.getTemplate(templateVO.getId());
|
||||
TemplateInfo tmpl = this.templateFactory.getTemplate(templateVO.getId(), DataStoreRole.Image);
|
||||
DataStore imageStore = this.dataStoreMgr.getImageStore(this.dcId);
|
||||
|
||||
this.imageService.createTemplateFromVolumeAsync(volInfo, tmpl, imageStore);
|
||||
|
||||
@ -303,7 +303,7 @@ public class ObjectInDataStoreManagerImpl implements ObjectInDataStoreManager {
|
||||
DataObjectInStore vo = null;
|
||||
switch (type){
|
||||
case TEMPLATE:
|
||||
vo = templateDataStoreDao.findByTemplate(objId);
|
||||
vo = templateDataStoreDao.findByTemplate(objId, role);
|
||||
break;
|
||||
case SNAPSHOT:
|
||||
vo = snapshotDataStoreDao.findBySnapshot(objId, role);
|
||||
@ -313,7 +313,7 @@ public class ObjectInDataStoreManagerImpl implements ObjectInDataStoreManager {
|
||||
break;
|
||||
}
|
||||
if (vo != null) {
|
||||
store = this.storeMgr.getDataStore(vo.getDataStoreId(), DataStoreRole.Image);
|
||||
store = this.storeMgr.getDataStore(vo.getDataStoreId(), role);
|
||||
}
|
||||
}
|
||||
return store;
|
||||
|
||||
@ -35,6 +35,7 @@ import org.apache.log4j.Logger;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
|
||||
import com.cloud.storage.DataStoreRole;
|
||||
import com.cloud.storage.VMTemplateStorageResourceAssoc.Status;
|
||||
import com.cloud.utils.db.GenericDaoBase;
|
||||
import com.cloud.utils.db.SearchBuilder;
|
||||
@ -51,6 +52,7 @@ public class TemplateDataStoreDaoImpl extends GenericDaoBase<TemplateDataStoreVO
|
||||
private SearchBuilder<TemplateDataStoreVO> updateStateSearch;
|
||||
private SearchBuilder<TemplateDataStoreVO> storeSearch;
|
||||
private SearchBuilder<TemplateDataStoreVO> templateSearch;
|
||||
private SearchBuilder<TemplateDataStoreVO> templateRoleSearch;
|
||||
private SearchBuilder<TemplateDataStoreVO> storeTemplateSearch;
|
||||
private SearchBuilder<TemplateDataStoreVO> storeTemplateStateSearch;
|
||||
private SearchBuilder<TemplateDataStoreVO> storeTemplateDownloadStatusSearch;
|
||||
@ -73,6 +75,12 @@ public class TemplateDataStoreDaoImpl extends GenericDaoBase<TemplateDataStoreVO
|
||||
templateSearch.and("destroyed", templateSearch.entity().getDestroyed(), SearchCriteria.Op.EQ);
|
||||
templateSearch.done();
|
||||
|
||||
templateRoleSearch = createSearchBuilder();
|
||||
templateRoleSearch.and("template_id", templateRoleSearch.entity().getTemplateId(), SearchCriteria.Op.EQ);
|
||||
templateRoleSearch.and("store_role", templateRoleSearch.entity().getDataStoreRole(), SearchCriteria.Op.EQ);
|
||||
templateRoleSearch.and("destroyed", templateRoleSearch.entity().getDestroyed(), SearchCriteria.Op.EQ);
|
||||
templateRoleSearch.done();
|
||||
|
||||
|
||||
updateStateSearch = this.createSearchBuilder();
|
||||
updateStateSearch.and("id", updateStateSearch.entity().getId(), Op.EQ);
|
||||
@ -263,10 +271,12 @@ public class TemplateDataStoreDaoImpl extends GenericDaoBase<TemplateDataStoreVO
|
||||
return lockOneRandomRow(sc, true);
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public TemplateDataStoreVO findByTemplate(long templateId) {
|
||||
SearchCriteria<TemplateDataStoreVO> sc = templateSearch.create();
|
||||
public TemplateDataStoreVO findByTemplate(long templateId, DataStoreRole role) {
|
||||
SearchCriteria<TemplateDataStoreVO> sc = templateRoleSearch.create();
|
||||
sc.setParameters("template_id", templateId);
|
||||
sc.setParameters("store_role", role);
|
||||
sc.setParameters("destroyed", false);
|
||||
return findOneIncludingRemovedBy(sc);
|
||||
}
|
||||
@ -279,6 +289,24 @@ public class TemplateDataStoreDaoImpl extends GenericDaoBase<TemplateDataStoreVO
|
||||
return search(sc, null);
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public TemplateDataStoreVO findByTemplateZone(long templateId, Long zoneId, DataStoreRole role) {
|
||||
// get all elgible image stores
|
||||
List<DataStore> imgStores = null;
|
||||
if ( role == DataStoreRole.Image){
|
||||
imgStores = this._storeMgr.getImageStoresByScope(new ZoneScope(zoneId));
|
||||
} else if (role == DataStoreRole.ImageCache){
|
||||
imgStores = this._storeMgr.getImageCacheStores(new ZoneScope(zoneId));
|
||||
}
|
||||
if ( imgStores != null ){
|
||||
for (DataStore store : imgStores){
|
||||
List<TemplateDataStoreVO> sRes = this.listByTemplateStore(templateId, store.getId());
|
||||
if ( sRes != null && sRes.size() > 0){
|
||||
return sRes.get(0);
|
||||
}
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@ -671,7 +671,7 @@ public class VolumeManagerImpl extends ManagerBase implements VolumeManager {
|
||||
if (isNotCreatedFromTemplate) {
|
||||
future = this.volService.createVolumeAsync(volume, store);
|
||||
} else {
|
||||
TemplateInfo templ = this.tmplFactory.getTemplate(template.getId());
|
||||
TemplateInfo templ = this.tmplFactory.getTemplate(template.getId(), DataStoreRole.Image);
|
||||
future = this.volService.createVolumeFromTemplateAsync(volume, store.getId(), templ);
|
||||
}
|
||||
try {
|
||||
@ -2266,7 +2266,7 @@ public class VolumeManagerImpl extends ManagerBase implements VolumeManager {
|
||||
if (templateId == null) {
|
||||
future = this.volService.createVolumeAsync(volume, destPool);
|
||||
} else {
|
||||
TemplateInfo templ = this.tmplFactory.getTemplate(templateId);
|
||||
TemplateInfo templ = this.tmplFactory.getTemplate(templateId, DataStoreRole.Image);
|
||||
future = this.volService.createVolumeFromTemplateAsync(volume, destPool.getId(), templ);
|
||||
}
|
||||
VolumeApiResult result = null;
|
||||
|
||||
@ -174,6 +174,7 @@ public class DownloadMonitorImpl extends ManagerBase implements DownloadMonitor
|
||||
if (vmTemplateStore == null) {
|
||||
vmTemplateStore = new TemplateDataStoreVO(store.getId(), template.getId(), new Date(), 0,
|
||||
VMTemplateStorageResourceAssoc.Status.NOT_DOWNLOADED, null, null, "jobid0000", null, template.getUri());
|
||||
vmTemplateStore.setDataStoreRole(store.getRole());
|
||||
_vmTemplateStoreDao.persist(vmTemplateStore);
|
||||
} else if ((vmTemplateStore.getJobId() != null) && (vmTemplateStore.getJobId().length() > 2)) {
|
||||
downloadJobExists = true;
|
||||
|
||||
@ -1348,7 +1348,7 @@ public class TemplateManagerImpl extends ManagerBase implements TemplateManager,
|
||||
VolumeVO volume = null;
|
||||
|
||||
try {
|
||||
TemplateInfo tmplInfo = this._tmplFactory.getTemplate(templateId);
|
||||
TemplateInfo tmplInfo = this._tmplFactory.getTemplate(templateId, DataStoreRole.Image);
|
||||
Long zoneId = null;
|
||||
if (snapshotId != null) {
|
||||
snapshot = _snapshotDao.findById(snapshotId);
|
||||
|
||||
@ -128,6 +128,7 @@ CREATE TABLE `cloud`.`template_store_ref` (
|
||||
`job_id` varchar(255),
|
||||
`download_pct` int(10) unsigned,
|
||||
`size` bigint unsigned,
|
||||
`store_role` varchar(255),
|
||||
`physical_size` bigint unsigned DEFAULT 0,
|
||||
`download_state` varchar(255),
|
||||
`error_str` varchar(255),
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user