Skip to content

Commit e331976

Browse files
committed
Move all other saveDetails to use the UserVmDetailsDaoImpl
1 parent b76e698 commit e331976

File tree

8 files changed

+23
-44
lines changed

8 files changed

+23
-44
lines changed

engine/orchestration/src/main/java/com/cloud/vm/VirtualMachineManagerImpl.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1865,7 +1865,7 @@ protected boolean sendStop(final VirtualMachineGuru guru, final VirtualMachinePr
18651865
final UserVmVO userVm = _userVmDao.findById(vm.getId());
18661866
_userVmDao.loadDetails(userVm);
18671867
userVm.setDetail(VmDetailConstants.PLATFORM, platform);
1868-
_userVmDao.saveDetails(userVm);
1868+
userVmDetailsDao.saveDetails(userVm);
18691869
}
18701870
}
18711871

@@ -2190,7 +2190,7 @@ private void advanceStop(final VMInstanceVO vm, final boolean cleanUpEvenIfUnabl
21902190
final UserVmVO userVm = _userVmDao.findById(vm.getId());
21912191
_userVmDao.loadDetails(userVm);
21922192
userVm.setDetail(VmDetailConstants.PLATFORM, platform);
2193-
_userVmDao.saveDetails(userVm);
2193+
userVmDetailsDao.saveDetails(userVm);
21942194
}
21952195
}
21962196
}
@@ -3757,7 +3757,7 @@ private void updateVmMetaData(Long vmId, String platform) {
37573757
if (!userVm.details.containsKey(VmDetailConstants.HYPERVISOR_TOOLS_VERSION) || !userVm.details.get(VmDetailConstants.HYPERVISOR_TOOLS_VERSION).equals(pvdriver)) {
37583758
userVm.setDetail(VmDetailConstants.HYPERVISOR_TOOLS_VERSION, pvdriver);
37593759
}
3760-
_userVmDao.saveDetails(userVm);
3760+
userVmDetailsDao.saveDetails(userVm);
37613761
}
37623762

37633763
@Override

engine/schema/src/main/java/com/cloud/vm/dao/UserVmDao.java

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -87,10 +87,6 @@ void updateVM(long id, String displayName, boolean enable, Long osTypeId,
8787

8888
void loadDetails(UserVmVO vm);
8989

90-
void saveDetails(UserVmVO vm);
91-
92-
void saveDetails(UserVmVO vm, List<String> hiddenDetails);
93-
9490
List<Long> listPodIdsHavingVmsforAccount(long zoneId, long accountId);
9591

9692
public Long countAllocatedVMsForAccount(long accountId, boolean runningVMsonly);

engine/schema/src/main/java/com/cloud/vm/dao/UserVmDaoImpl.java

Lines changed: 0 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,6 @@
5757
import com.cloud.utils.db.TransactionLegacy;
5858
import com.cloud.utils.exception.CloudRuntimeException;
5959
import com.cloud.vm.NicVO;
60-
import com.cloud.vm.UserVmDetailVO;
6160
import com.cloud.vm.UserVmVO;
6261
import com.cloud.vm.VirtualMachine;
6362
import com.cloud.vm.VirtualMachine.State;
@@ -431,29 +430,6 @@ public void loadDetails(UserVmVO vm) {
431430
}
432431
}
433432

434-
@Override
435-
public void saveDetails(UserVmVO vm) {
436-
saveDetails(vm, new ArrayList<String>());
437-
}
438-
439-
@Override
440-
public void saveDetails(UserVmVO vm, List<String> hiddenDetails) {
441-
Map<String, String> detailsStr = vm.getDetails();
442-
if (detailsStr == null) {
443-
return;
444-
}
445-
446-
final Map<String, Boolean> visibilityMap = _detailsDao.listDetailsVisibility(vm.getId());
447-
448-
List<UserVmDetailVO> details = new ArrayList<UserVmDetailVO>();
449-
for (Map.Entry<String, String> entry : detailsStr.entrySet()) {
450-
boolean display = !hiddenDetails.contains(entry.getKey()) && visibilityMap.getOrDefault(entry.getKey(), true);
451-
details.add(new UserVmDetailVO(vm.getId(), entry.getKey(), entry.getValue(), display));
452-
}
453-
454-
_detailsDao.saveDetails(details);
455-
}
456-
457433
@Override
458434
public List<Long> listPodIdsHavingVmsforAccount(long zoneId, long accountId) {
459435
TransactionLegacy txn = TransactionLegacy.currentTxn();

server/src/main/java/com/cloud/capacity/CapacityManagerImpl.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -757,7 +757,7 @@ public void updateCapacityForHost(final Host host, final Map<Long, ServiceOfferi
757757
UserVmVO userVM = _userVMDao.findById(vm.getId());
758758
_userVMDao.loadDetails(userVM);
759759
userVM.setDetail(VmDetailConstants.MESSAGE_RESERVED_CAPACITY_FREED_FLAG, "true");
760-
_userVMDao.saveDetails(userVM);
760+
_userVmDetailsDao.saveDetails(userVM);
761761
}
762762
}
763763
}
@@ -985,7 +985,7 @@ public boolean postStateTransitionEvent(StateMachine2.Transition<State, Event> t
985985
_userVMDao.loadDetails(userVM);
986986
// free the message sent flag if it exists
987987
userVM.setDetail(VmDetailConstants.MESSAGE_RESERVED_CAPACITY_FREED_FLAG, "false");
988-
_userVMDao.saveDetails(userVM);
988+
_userVmDetailsDao.saveDetails(userVM);
989989
}
990990

991991
return true;

server/src/main/java/com/cloud/network/element/VirtualRouterElement.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424

2525
import javax.inject.Inject;
2626

27+
import com.cloud.vm.dao.UserVmDetailsDao;
2728
import org.apache.cloudstack.network.BgpPeer;
2829
import org.apache.commons.collections.CollectionUtils;
2930
import org.apache.commons.lang3.ObjectUtils;
@@ -176,6 +177,9 @@ public class VirtualRouterElement extends AdapterBase implements VirtualRouterEl
176177
@Inject
177178
NetworkDetailsDao _networkDetailsDao;
178179

180+
@Inject
181+
private UserVmDetailsDao userVmDetailsDao;
182+
179183
@Inject
180184
protected RouterDeploymentDefinitionBuilder routerDeploymentDefinitionBuilder;
181185

@@ -737,7 +741,7 @@ public boolean savePassword(final Network network, final NicProfile nic, final V
737741

738742
_userVmDao.loadDetails(userVmVO);
739743
userVmVO.setDetail(VmDetailConstants.PASSWORD, password_encrypted);
740-
_userVmDao.saveDetails(userVmVO);
744+
userVmDetailsDao.saveDetails(userVmVO);
741745

742746
userVmVO.setUpdateParameters(true);
743747
_userVmDao.update(userVmVO.getId(), userVmVO);

server/src/main/java/com/cloud/vm/UserVmManagerImpl.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1087,7 +1087,7 @@ private boolean resetVMSSHKeyInternal(Long vmId, String sshPublicKeys, String ke
10871087
_vmDao.loadDetails(userVm);
10881088
userVm.setDetail(VmDetailConstants.SSH_PUBLIC_KEY, sshPublicKeys);
10891089
userVm.setDetail(VmDetailConstants.SSH_KEY_PAIR_NAMES, keypairnames);
1090-
_vmDao.saveDetails(userVm);
1090+
userVmDetailsDao.saveDetails(userVm);
10911091

10921092
if (vmInstance.getState() == State.Stopped) {
10931093
logger.debug("Vm " + vmInstance + " is stopped, not rebooting it as a part of SSH Key reset");
@@ -2910,7 +2910,7 @@ public UserVm updateVirtualMachine(UpdateVMCmd cmd) throws ResourceUnavailableEx
29102910

29112911
verifyVmLimits(vmInstance, details);
29122912
vmInstance.setDetails(details);
2913-
_vmDao.saveDetails(vmInstance);
2913+
userVmDetailsDao.saveDetails(vmInstance);
29142914
}
29152915
if (StringUtils.isNotBlank(extraConfig)) {
29162916
if (EnableAdditionalVmConfig.valueIn(accountId)) {
@@ -4710,7 +4710,7 @@ private UserVmVO commitUserVm(final boolean isImport, final DataCenter zone, fin
47104710
if (customParameters.containsKey(VmDetailConstants.NAME_ON_HYPERVISOR)) {
47114711
hiddenDetails.add(VmDetailConstants.NAME_ON_HYPERVISOR);
47124712
}
4713-
_vmDao.saveDetails(vm, hiddenDetails);
4713+
userVmDetailsDao.saveDetails(vm, hiddenDetails);
47144714
if (!isImport) {
47154715
logger.debug("Allocating in the DB for vm");
47164716
DataCenterDeployment plan = new DataCenterDeployment(zone.getId());
@@ -8510,7 +8510,7 @@ private void encryptAndStorePassword(UserVmVO vm, String password) {
85108510
}
85118511

85128512
vm.setDetail(VmDetailConstants.ENCRYPTED_PASSWORD, encryptedPasswd);
8513-
_vmDao.saveDetails(vm);
8513+
userVmDetailsDao.saveDetails(vm);
85148514
}
85158515
}
85168516

@@ -8519,7 +8519,7 @@ public void persistDeviceBusInfo(UserVmVO vm, String rootDiskController) {
85198519
String existingVmRootDiskController = vm.getDetail(VmDetailConstants.ROOT_DISK_CONTROLLER);
85208520
if (StringUtils.isEmpty(existingVmRootDiskController) && StringUtils.isNotEmpty(rootDiskController)) {
85218521
vm.setDetail(VmDetailConstants.ROOT_DISK_CONTROLLER, rootDiskController);
8522-
_vmDao.saveDetails(vm);
8522+
userVmDetailsDao.saveDetails(vm);
85238523
if (logger.isDebugEnabled()) {
85248524
logger.debug("Persisted device bus information rootDiskController=" + rootDiskController + " for vm: " + vm.getDisplayName());
85258525
}

server/src/test/java/com/cloud/vm/UserVmManagerImplTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -471,14 +471,14 @@ private void prepareAndExecuteMethodDealingWithDetails(boolean cleanUpDetails, b
471471
Mockito.verify(userVmVoMock, Mockito.times(cleanUpDetails || isDetailsEmpty ? 0 : 1)).setDetails(details);
472472
Mockito.verify(userVmDetailsDao, Mockito.times(cleanUpDetails ? 1 : 0)).removeDetail(vmId, "existingdetail");
473473
Mockito.verify(userVmDetailsDao, Mockito.times(0)).removeDetail(vmId, "systemdetail");
474-
Mockito.verify(userVmDao, Mockito.times(cleanUpDetails || isDetailsEmpty ? 0 : 1)).saveDetails(userVmVoMock);
474+
Mockito.verify(userVmDetailsDao, Mockito.times(cleanUpDetails || isDetailsEmpty ? 0 : 1)).saveDetails(userVmVoMock);
475475
Mockito.verify(userVmManagerImpl, Mockito.times(0)).updateDisplayVmFlag(false, vmId, userVmVoMock);
476476
}
477477

478478
private void configureDoNothingForDetailsMethod() {
479479
Mockito.lenient().doNothing().when(userVmManagerImpl).updateDisplayVmFlag(false, vmId, userVmVoMock);
480480
Mockito.doNothing().when(userVmDetailsDao).removeDetail(anyLong(), anyString());
481-
Mockito.doNothing().when(userVmDao).saveDetails(userVmVoMock);
481+
Mockito.doNothing().when(userVmDetailsDao).saveDetails(userVmVoMock);
482482
}
483483

484484
@SuppressWarnings("unchecked")

server/src/test/java/com/cloud/vm/UserVmManagerTest.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@
6363
import com.cloud.vm.VirtualMachine.State;
6464
import com.cloud.vm.dao.NicDao;
6565
import com.cloud.vm.dao.UserVmDao;
66+
import com.cloud.vm.dao.UserVmDetailsDao;
6667
import com.cloud.vm.dao.VMInstanceDao;
6768
import com.cloud.vm.snapshot.VMSnapshotVO;
6869
import com.cloud.vm.snapshot.dao.VMSnapshotDao;
@@ -145,6 +146,8 @@ public class UserVmManagerTest {
145146
@Mock
146147
private UserVmDao _vmDao;
147148
@Mock
149+
private UserVmDetailsDao userVmDetailsDao;
150+
@Mock
148151
private VMInstanceDao _vmInstanceDao;
149152
@Mock
150153
private VMTemplateDao _templateDao;
@@ -821,20 +824,20 @@ public void testApplyUserDataSuccessful() throws Exception {
821824
public void testPersistDeviceBusInfoWithNullController() {
822825
when(_vmMock.getDetail(any(String.class))).thenReturn(null);
823826
_userVmMgr.persistDeviceBusInfo(_vmMock, null);
824-
verify(_vmDao, times(0)).saveDetails(any(UserVmVO.class));
827+
verify(userVmDetailsDao, times(0)).saveDetails(any(UserVmVO.class));
825828
}
826829

827830
@Test
828831
public void testPersistDeviceBusInfoWithEmptyController() {
829832
when(_vmMock.getDetail(any(String.class))).thenReturn("");
830833
_userVmMgr.persistDeviceBusInfo(_vmMock, "");
831-
verify(_vmDao, times(0)).saveDetails(any(UserVmVO.class));
834+
verify(userVmDetailsDao, times(0)).saveDetails(any(UserVmVO.class));
832835
}
833836

834837
@Test
835838
public void testPersistDeviceBusInfo() {
836839
when(_vmMock.getDetail(any(String.class))).thenReturn(null);
837840
_userVmMgr.persistDeviceBusInfo(_vmMock, "lsilogic");
838-
verify(_vmDao, times(1)).saveDetails(any(UserVmVO.class));
841+
verify(userVmDetailsDao, times(1)).saveDetails(any(UserVmVO.class));
839842
}
840843
}

0 commit comments

Comments
 (0)