Allow to set the number of cpus for virtualbox and kvm deployments.

2 files changed, 9 insertions, 3 deletions

diff --git a/kvm.write b/kvm.write
index 67ac40e7..9a6be135 100755
--- a/kvm.write
+++ b/kvm.write
@@ -113,8 +113,11 @@ class KvmPlusSshWriteExtension(morphlib.writeexts.WriteExtension):
 
         ram_mebibytes = str(self.get_ram_size() / (1024**2))
 
-        cmdline = ['virt-install', '--connect', 'qemu:///system', '--import',
-                   '--name', vm_name, '--vnc', '--ram=%s' % ram_mebibytes,
+        vcpu_count = str(self.get_vcpu_count())
+
+        cmdline = ['virt-install', '--connect', 'qemu:///system',
+                   '--import', '--name', vm_name, '--vnc',
+                   '--ram', ram_mebibytes, '--vcpus', vcpu_count,
                    '--disk', 'path=%s,bus=ide' % vm_path] + attach_opts
         if not autostart:
             cmdline += ['--noreboot']
diff --git a/virtualbox-ssh.write b/virtualbox-ssh.write
index 3ee2eae0..1abe233e 100755
--- a/virtualbox-ssh.write
+++ b/virtualbox-ssh.write
@@ -113,12 +113,15 @@ class VirtualBoxPlusSshWriteExtension(morphlib.writeexts.WriteExtension):
 
         ram_mebibytes = str(self.get_ram_size() / (1024**2))
 
+        vcpu_count = str(self.get_vcpu_count())
+
         hostonly_iface = self.get_host_interface(ssh_host)
 
         commands = [
             ['createvm', '--name', vm_name, '--ostype', 'Linux26_64',
              '--register'],
-            ['modifyvm', vm_name, '--ioapic', 'on', '--memory', ram_mebibytes,
+            ['modifyvm', vm_name, '--ioapic', 'on',
+             '--memory', ram_mebibytes, '--cpus', vcpu_count,
              '--nic1', 'hostonly', '--hostonlyadapter1', hostonly_iface,
              '--nic2', 'nat', '--natnet2', 'default'],
             ['storagectl', vm_name, '--name', '"SATA Controller"',