Skip to content

Commit 0f5f3da

Browse files
committed
more tests and refactor
Signed-off-by: Abhishek Kumar <[email protected]>
1 parent 480fa9d commit 0f5f3da

File tree

6 files changed

+353
-28
lines changed

6 files changed

+353
-28
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@
1919

2020
import org.apache.cloudstack.extension.Extension;
2121

22-
public class DownloadExtensionFilesCommand extends ExtensionServerActionBaseCommand {
22+
public class PrepareDownloadExtensionFilesCommand extends ExtensionServerActionBaseCommand {
2323

24-
public DownloadExtensionFilesCommand(long msId, Extension extension) {
24+
public PrepareDownloadExtensionFilesCommand(long msId, Extension extension) {
2525
super(msId, extension);
2626
}
2727
}

framework/extensions/src/main/java/org/apache/cloudstack/framework/extensions/manager/ExtensionsManagerImpl.java

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -78,10 +78,10 @@
7878
import org.apache.cloudstack.framework.extensions.api.response.DownloadExtensionResponse;
7979
import org.apache.cloudstack.framework.extensions.command.CleanupExtensionFilesCommand;
8080
import org.apache.cloudstack.framework.extensions.command.DownloadAndSyncExtensionFilesCommand;
81-
import org.apache.cloudstack.framework.extensions.command.DownloadExtensionFilesCommand;
8281
import org.apache.cloudstack.framework.extensions.command.ExtensionRoutingUpdateCommand;
8382
import org.apache.cloudstack.framework.extensions.command.ExtensionServerActionBaseCommand;
8483
import org.apache.cloudstack.framework.extensions.command.GetExtensionPathChecksumCommand;
84+
import org.apache.cloudstack.framework.extensions.command.PrepareDownloadExtensionFilesCommand;
8585
import org.apache.cloudstack.framework.extensions.command.PrepareExtensionPathCommand;
8686
import org.apache.cloudstack.framework.extensions.command.StartSyncExtensionFilesCommand;
8787
import org.apache.cloudstack.framework.extensions.dao.ExtensionCustomActionDao;
@@ -1575,9 +1575,9 @@ public String handleExtensionServerCommands(ExtensionServerActionBaseCommand com
15751575
final DownloadAndSyncExtensionFilesCommand cmd = (DownloadAndSyncExtensionFilesCommand)command;
15761576
Pair<Boolean, String> result = downloadAndSyncExtensionFiles(cmd);
15771577
answer = new Answer(cmd, result.first(), result.second());
1578-
} else if (command instanceof DownloadExtensionFilesCommand) {
1579-
final DownloadExtensionFilesCommand cmd = (DownloadExtensionFilesCommand)command;
1580-
Pair<Boolean, String> result = downloadExtensionFiles(cmd);
1578+
} else if (command instanceof PrepareDownloadExtensionFilesCommand) {
1579+
final PrepareDownloadExtensionFilesCommand cmd = (PrepareDownloadExtensionFilesCommand)command;
1580+
Pair<Boolean, String> result = prepareDownloadExtensionFiles(cmd);
15811581
answer = new Answer(cmd, result.first(), result.second());
15821582
}
15831583
final Answer[] answers = new Answer[1];
@@ -1767,21 +1767,21 @@ protected Pair<Boolean, String> downloadExtensionUsingMSPeer(ExtensionVO extensi
17671767
logger.debug("Initiating download for {} using {}", extension, managementServer);
17681768
final String msPeer = Long.toString(managementServer.getMsid());
17691769
final Command[] cmds = new Command[1];
1770-
cmds[0] = new DownloadExtensionFilesCommand(
1770+
cmds[0] = new PrepareDownloadExtensionFilesCommand(
17711771
ManagementServerNode.getManagementServerId(), extension);
17721772
String answersStr = clusterManager.execute(msPeer, 0L, GsonHelper.getGson().toJson(cmds), true);
17731773
return getResultFromAnswersString(answersStr, extension, managementServer, "download");
17741774
}
17751775

1776-
protected Pair<Boolean, String> downloadExtensionFiles(DownloadExtensionFilesCommand cmd) {
1776+
protected Pair<Boolean, String> prepareDownloadExtensionFiles(PrepareDownloadExtensionFilesCommand cmd) {
17771777
final long extensionId = cmd.getExtensionId();
17781778
final ExtensionVO extension = extensionDao.findById(extensionId);
17791779
if (extension == null) {
17801780
String msg = String.format("Unable to find extension with id: %d for starting sync", extensionId);
17811781
logger.error(msg);
17821782
return new Pair<>(false, msg);
17831783
}
1784-
return extensionsShareManager.downloadExtension(extension,
1784+
return extensionsShareManager.prepareExtensionDownload(extension,
17851785
managementServerHostDao.findByMsid(ManagementServerNode.getManagementServerId()));
17861786
}
17871787

@@ -1803,7 +1803,7 @@ public DownloadExtensionResponse downloadExtension(DownloadExtensionCmd cmd) {
18031803
if (ManagementServerNode.getManagementServerId() != managementServer.getMsid()) {
18041804
result = downloadExtensionUsingMSPeer(extension, managementServer);
18051805
} else {
1806-
result = extensionsShareManager.downloadExtension(extension, managementServer);
1806+
result = extensionsShareManager.prepareExtensionDownload(extension, managementServer);
18071807
}
18081808
if (result == null || !result.first()) {
18091809
String msg = result == null ? "Null result received for download operation" : result.second();

framework/extensions/src/main/java/org/apache/cloudstack/framework/extensions/manager/ExtensionsShareManager.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,5 +32,5 @@ Pair<Boolean, String> syncExtension(Extension extension, ManagementServerHost so
3232

3333
Pair<Boolean, String> downloadAndApplyExtensionSync(Extension extension, DownloadAndSyncExtensionFilesCommand cmd);
3434

35-
Pair<Boolean, String> downloadExtension(Extension extension, ManagementServerHost managementServer);
35+
Pair<Boolean, String> prepareExtensionDownload(Extension extension, ManagementServerHost managementServer);
3636
}

framework/extensions/src/main/java/org/apache/cloudstack/framework/extensions/manager/ExtensionsShareManagerImpl.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -789,7 +789,7 @@ public Pair<Boolean, String> downloadAndApplyExtensionSync(Extension extension,
789789
}
790790

791791
@Override
792-
public Pair<Boolean, String> downloadExtension(Extension extension, ManagementServerHost managementServer) {
792+
public Pair<Boolean, String> prepareExtensionDownload(Extension extension, ManagementServerHost managementServer) {
793793
ArchiveInfo archiveInfo;
794794
try {
795795
archiveInfo = createArchiveForDownload(extension);

0 commit comments

Comments
 (0)