@Test
public void test() throws ApiSenderException, InterruptedException {
long requiredSize = SizeUnit.GIGABYTE.toByte(10);
SimulatorPrimaryStorageDetails sp = new SimulatorPrimaryStorageDetails();
sp.setTotalCapacity(SizeUnit.TERABYTE.toByte(10));
sp.setAvailableCapacity(sp.getTotalCapacity());
sp.setUrl("nfs://simulator/primary/");
PrimaryStorageInventory pinv = api.createSimulatoPrimaryStorage(1, sp).get(0);
ZoneInventory zone = api.createZones(1).get(0);
ClusterInventory cluster = api.createClusters(1, zone.getUuid()).get(0);
HostInventory host = api.createHost(1, cluster.getUuid()).get(0);
api.attachPrimaryStorage(cluster.getUuid(), pinv.getUuid());
AllocatePrimaryStorageMsg msg = new AllocatePrimaryStorageMsg();
msg.setRequiredHostUuid(host.getUuid());
msg.setSize(requiredSize);
msg.setServiceId(bus.makeLocalServiceId(PrimaryStorageConstant.SERVICE_ID));
MessageReply reply = bus.call(msg);
Assert.assertEquals(AllocatePrimaryStorageReply.class, reply.getClass());
AllocatePrimaryStorageReply ar = (AllocatePrimaryStorageReply) reply;
Assert.assertEquals(pinv.getUuid(), ar.getPrimaryStorageInventory().getUuid());
ReturnPrimaryStorageCapacityMsg rmsg = new ReturnPrimaryStorageCapacityMsg();
rmsg.setDiskSize(requiredSize);
rmsg.setPrimaryStorageUuid(pinv.getUuid());
rmsg.setServiceId(bus.makeLocalServiceId(PrimaryStorageConstant.SERVICE_ID));
bus.send(rmsg);
Thread.sleep(2000);
PrimaryStorageVO pvo = dbf.findByUuid(pinv.getUuid(), PrimaryStorageVO.class);
Assert.assertEquals(
pvo.getCapacity().getTotalCapacity(), pvo.getCapacity().getAvailableCapacity());
}
@Override
public PrimaryStorageInventory createPrimaryStorage(
PrimaryStorageVO vo, APIAddPrimaryStorageMsg msg) {
vo.setMountPath(msg.getUrl());
vo = dbf.persistAndRefresh(vo);
return PrimaryStorageInventory.valueOf(vo);
}
@Override
public void rollback(FlowRollback trigger, Map data) {
Long size = (Long) data.get(LocalStorageAllocateCapacityForAttachingVolumeFlow.class);
if (size != null) {
PrimaryStorageInventory pri =
(PrimaryStorageInventory)
data.get(
VmInstanceConstant.Params.DestPrimaryStorageInventoryForAttachingVolume
.toString());
ReturnPrimaryStorageCapacityMsg rmsg = new ReturnPrimaryStorageCapacityMsg();
rmsg.setPrimaryStorageUuid(pri.getUuid());
rmsg.setDiskSize(size);
bus.makeTargetServiceIdByResourceUuid(rmsg, PrimaryStorageConstant.SERVICE_ID, pri.getUuid());
bus.send(rmsg);
}
trigger.rollback();
}
private void handle(PrimaryStorageDeletionMsg msg) {
PrimaryStorageInventory inv = PrimaryStorageInventory.valueOf(self);
extpEmitter.beforeDelete(inv);
deleteHook();
extpEmitter.afterDelete(inv);
PrimaryStorageDeletionReply reply = new PrimaryStorageDeletionReply();
bus.reply(msg, reply);
}
protected void handle(APIChangePrimaryStorageStateMsg msg) {
APIChangePrimaryStorageStateEvent evt = new APIChangePrimaryStorageStateEvent(msg.getId());
PrimaryStorageState currState = self.getState();
PrimaryStorageStateEvent event = PrimaryStorageStateEvent.valueOf(msg.getStateEvent());
PrimaryStorageState nextState = AbstractPrimaryStorage.getNextState(currState, event);
try {
extpEmitter.preChange(self, event);
} catch (PrimaryStorageException e) {
evt.setErrorCode(
errf.instantiateErrorCode(SysErrors.CHANGE_RESOURCE_STATE_ERROR, e.getMessage()));
bus.publish(evt);
return;
}
extpEmitter.beforeChange(self, event);
changeStateHook(event, nextState);
self.setState(nextState);
self = dbf.updateAndRefresh(self);
extpEmitter.afterChange(self, event, currState);
evt.setInventory(PrimaryStorageInventory.valueOf(self));
bus.publish(evt);
}
@Override
public PrimaryStorageInventory getInventory(String uuid) {
return PrimaryStorageInventory.valueOf(dbf.findByUuid(uuid, PrimaryStorageVO.class));
}
protected void handle(APIDeletePrimaryStorageMsg msg) {
final APIDeletePrimaryStorageEvent evt = new APIDeletePrimaryStorageEvent(msg.getId());
final String issuer = PrimaryStorageVO.class.getSimpleName();
final List<PrimaryStorageInventory> ctx = PrimaryStorageInventory.valueOf(Arrays.asList(self));
FlowChain chain = FlowChainBuilder.newSimpleFlowChain();
chain.setName(String.format("delete-primary-storage-%s", msg.getUuid()));
if (msg.getDeletionMode() == APIDeleteMessage.DeletionMode.Permissive) {
chain
.then(
new NoRollbackFlow() {
@Override
public void run(final FlowTrigger trigger, Map data) {
casf.asyncCascade(
CascadeConstant.DELETION_CHECK_CODE,
issuer,
ctx,
new Completion(trigger) {
@Override
public void success() {
trigger.next();
}
@Override
public void fail(ErrorCode errorCode) {
trigger.fail(errorCode);
}
});
}
})
.then(
new NoRollbackFlow() {
@Override
public void run(final FlowTrigger trigger, Map data) {
casf.asyncCascade(
CascadeConstant.DELETION_DELETE_CODE,
issuer,
ctx,
new Completion(trigger) {
@Override
public void success() {
trigger.next();
}
@Override
public void fail(ErrorCode errorCode) {
trigger.fail(errorCode);
}
});
}
});
} else {
chain.then(
new NoRollbackFlow() {
@Override
public void run(final FlowTrigger trigger, Map data) {
casf.asyncCascade(
CascadeConstant.DELETION_FORCE_DELETE_CODE,
issuer,
ctx,
new Completion(trigger) {
@Override
public void success() {
trigger.next();
}
@Override
public void fail(ErrorCode errorCode) {
trigger.fail(errorCode);
}
});
}
});
}
chain
.done(
new FlowDoneHandler(msg) {
@Override
public void handle(Map data) {
casf.asyncCascadeFull(
CascadeConstant.DELETION_CLEANUP_CODE, issuer, ctx, new NopeCompletion());
bus.publish(evt);
PrimaryStorageDeletedData d = new PrimaryStorageDeletedData();
d.setPrimaryStorageUuid(self.getUuid());
d.setInventory(PrimaryStorageInventory.valueOf(self));
evtf.fire(PrimaryStorageCanonicalEvent.PRIMARY_STORAGE_DELETED_PATH, d);
}
})
.error(
new FlowErrorHandler(msg) {
@Override
public void handle(ErrorCode errCode, Map data) {
evt.setErrorCode(
errf.instantiateErrorCode(SysErrors.DELETE_RESOURCE_ERROR, errCode));
bus.publish(evt);
}
})
.start();
}
protected PrimaryStorageInventory getSelfInventory() {
return PrimaryStorageInventory.valueOf(self);
}
