Fix a bug in updating template_store_ref entry.

This commit is contained in:
Min Chen 2013-05-07 16:45:37 -07:00
parent e8c69632b6
commit 70f866156f
3 changed files with 42 additions and 33 deletions

View File

@ -175,6 +175,8 @@ public class CloudStackImageStoreDriverImpl implements ImageStoreDriver {
DataObject obj = context.data;
DataStore store = obj.getDataStore();
TemplateDataStoreVO tmpltStoreVO = _templateStoreDao.findByStoreTemplate(store.getId(),obj.getId());
if (tmpltStoreVO != null) {
TemplateDataStoreVO updateBuilder = _templateStoreDao.createForUpdate();
updateBuilder.setDownloadPercent(answer.getDownloadPct());
updateBuilder.setDownloadState(answer.getDownloadStatus());
@ -185,7 +187,8 @@ public class CloudStackImageStoreDriverImpl implements ImageStoreDriver {
updateBuilder.setInstallPath(answer.getInstallPath());
updateBuilder.setSize(answer.getTemplateSize());
updateBuilder.setPhysicalSize(answer.getTemplatePhySicalSize());
_templateStoreDao.update(store.getId(), updateBuilder);
_templateStoreDao.update(tmpltStoreVO.getId(), updateBuilder);
}
AsyncCompletionCallback<CreateCmdResult> caller = context.getParentCallback();

View File

@ -225,6 +225,8 @@ public class S3ImageStoreDriverImpl implements ImageStoreDriver {
DataObject obj = context.data;
DataStore store = obj.getDataStore();
TemplateDataStoreVO tmpltStoreVO = _templateStoreDao.findByStoreTemplate(store.getId(),obj.getId());
if (tmpltStoreVO != null) {
TemplateDataStoreVO updateBuilder = _templateStoreDao.createForUpdate();
updateBuilder.setDownloadPercent(answer.getDownloadPct());
updateBuilder.setDownloadState(answer.getDownloadStatus());
@ -235,7 +237,8 @@ public class S3ImageStoreDriverImpl implements ImageStoreDriver {
updateBuilder.setInstallPath(answer.getInstallPath());
updateBuilder.setSize(answer.getTemplateSize());
updateBuilder.setPhysicalSize(answer.getTemplatePhySicalSize());
_templateStoreDao.update(store.getId(), updateBuilder);
_templateStoreDao.update(tmpltStoreVO.getId(), updateBuilder);
}
AsyncCompletionCallback<CreateCmdResult> caller = context.getParentCallback();

View File

@ -178,6 +178,8 @@ public class SwiftImageStoreDriverImpl implements ImageStoreDriver {
DataObject obj = context.data;
DataStore store = obj.getDataStore();
TemplateDataStoreVO tmpltStoreVO = _templateStoreDao.findByStoreTemplate(store.getId(),obj.getId());
if (tmpltStoreVO != null) {
TemplateDataStoreVO updateBuilder = _templateStoreDao.createForUpdate();
updateBuilder.setDownloadPercent(answer.getDownloadPct());
updateBuilder.setDownloadState(answer.getDownloadStatus());
@ -188,7 +190,8 @@ public class SwiftImageStoreDriverImpl implements ImageStoreDriver {
updateBuilder.setInstallPath(answer.getInstallPath());
updateBuilder.setSize(answer.getTemplateSize());
updateBuilder.setPhysicalSize(answer.getTemplatePhySicalSize());
_templateStoreDao.update(store.getId(), updateBuilder);
_templateStoreDao.update(tmpltStoreVO.getId(), updateBuilder);
}
AsyncCompletionCallback<CreateCmdResult> caller = context.getParentCallback();