kjg/ungleich-tools
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Some optimizations to improve throughput

Browse source
This commit is contained in:
ahmadbilalkhalid 2020-01-28 23:28:00 +05:00
parent e34abc449d
commit 55a6868006
1 changed files with 29 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
opennebula-vm-etcd/put-vm-info-into-etcd.py
View file

@ -2,6 +2,7 @@ import pyone
from enum import IntEnum from enum import IntEnum
from config import config, etcd_client from config import config, etcd_client
from functools import reduce
# How to get client secrets? # How to get client secrets?
# 1. Login to OpenNebula # 1. Login to OpenNebula
@ -14,24 +15,24 @@ one_client = pyone.OneServer(
session=config['oca']['client_secrets'] session=config['oca']['client_secrets']
) )
host_pool = {
host.NAME: {
'name': host.NAME,
'id': host.ID,
'cluster': {
'name': host.CLUSTER,
'id': host.CLUSTER_ID
},
'vms': host.VMS.ID
}
for host in one_client.hostpool.info().HOST
}
def get_hostname_of_vm(vm_id): def get_hostname_of_vm(vm_id):
host_pool = {
host.NAME: {
'name': host.NAME,
'id': host.ID,
'cluster': {
'name': host.CLUSTER,
'id': host.CLUSTER_ID
},
'vms': host.VMS.ID
}
for host in one_client.hostpool.info().HOST
}
for hostname, host in host_pool.items(): for hostname, host in host_pool.items():
if vm_id in host['vms']: if vm_id in host['vms']:
return host return host
return None return None
@ -115,11 +116,21 @@ def main():
START_ID = -1 # First id whatever it is START_ID = -1 # First id whatever it is
END_ID = -1 # Last id whatever it is END_ID = -1 # Last id whatever it is
for VM_STATE in VM_STATES: # Get VMs in all kind of states
vm_pool = one_client.vmpool.infoextended(VmFilterFlag.AllResources.value, START_ID, END_ID, VM_STATE)
for i, vm in enumerate(vm_pool.VM): # vms is a list of lists
vm = VM(vm) vms = [
etcd_client.put('/opennebula/vm/{}'.format(vm.id), vm.get_data()) one_client.vmpool.infoextended(VmFilterFlag.AllResources.value, START_ID, END_ID, vm_state).VM
for vm_state in VM_STATES
]
# Take out elements from nested lists and put them into the original list
# forming a nice flat list
vms = list(reduce(lambda n, n_1: n + n_1, vms))
print('Total VMs:', len(vms))
for i, _vm in enumerate(vms):
vm = VM(_vm)
etcd_client.put('/opennebula/vm/{}'.format(vm.id), vm.get_data())
print(i, end=' ')
if __name__ == "__main__": if __name__ == "__main__":