def verify_slice_nodes(self, slice_urn, slice, rspec_nodes):
slivers = {}
for node in rspec_nodes:
hostname = node.get('component_name')
client_id = node.get('client_id')
#client_id = node.get('component_id')
component_id = node.get('component_id').strip()
if hostname:
hostname = hostname.strip()
elif component_id:
hostname = xrn_to_hostname(component_id)
if hostname:
slivers[hostname] = {'client_id': client_id, 'component_id': component_id}
all_nodes = self.driver.shell.GetNodes()
requested_slivers = []
for node in all_nodes:
if node['hostname'] in slivers.keys():
requested_slivers.append(node['node_id'])
if 'node_ids' not in slice.keys():
slice['node_ids']=[]
nodes = self.driver.shell.GetNodes({'node_ids': slice['node_ids']})
current_slivers = [node['node_id'] for node in nodes]
# remove nodes not in rspec
deleted_nodes = list(set(current_slivers).difference(requested_slivers))
# add nodes from rspec
added_nodes = list(set(requested_slivers).difference(current_slivers))
try:
self.driver.shell.AddSliceToNodes({'slice_id': slice['slice_id'], 'node_ids': added_nodes})
self.driver.shell.DeleteSliceFromNodes({'slice_id': slice['slice_id'], 'node_ids': deleted_nodes})
except:
logger.log_exc('Failed to add/remove slice from nodes')
slices = self.driver.shell.GetSlices({'slice_name': slice['slice_name']})
resulting_nodes = self.driver.shell.GetNodes({'node_ids': slices[0]['node_ids']})
# update sliver allocations
for node in resulting_nodes:
client_id = slivers[node['hostname']]['client_id']
component_id = slivers[node['hostname']]['component_id']
sliver_hrn = '%s.%s-%s' % (self.driver.hrn, slice['slice_id'], node['node_id'])
sliver_id = Xrn(sliver_hrn, type='sliver').urn
record = SliverAllocation(sliver_id=sliver_id, client_id=client_id,
component_id=component_id,
slice_urn = slice_urn,
allocation_state='geni_allocated')
record.sync(self.driver.api.dbsession())
return resulting_nodes
def verify_slice_nodes(self, slice_urn, slice, rspec_nodes):
slivers = {}
for node in rspec_nodes:
hostname = node.get('component_name')
client_id = node.get('client_id')
component_id = node.get('component_id').strip()
if hostname:
hostname = hostname.strip()
elif component_id:
hostname = xrn_to_hostname(component_id)
if hostname:
slivers[hostname] = {
'client_id': client_id,
'component_id': component_id
}
nodes = self.driver.shell.GetNodes(
slice['node_ids'], ['node_id', 'hostname', 'interface_ids'])
current_slivers = [node['hostname'] for node in nodes]
# remove nodes not in rspec
deleted_nodes = list(set(current_slivers).difference(slivers.keys()))
# add nodes from rspec
added_nodes = list(set(slivers.keys()).difference(current_slivers))
try:
self.driver.shell.AddSliceToNodes(slice['name'], added_nodes)
self.driver.shell.DeleteSliceFromNodes(slice['name'],
deleted_nodes)
except:
logger.log_exc('Failed to add/remove slice from nodes')
slices = self.driver.shell.GetSlices(slice['name'], ['node_ids'])
resulting_nodes = self.driver.shell.GetNodes(slices[0]['node_ids'])
# update sliver allocations
for node in resulting_nodes:
client_id = slivers[node['hostname']]['client_id']
component_id = slivers[node['hostname']]['component_id']
sliver_hrn = '%s.%s-%s' % (self.driver.hrn, slice['slice_id'],
node['node_id'])
sliver_id = Xrn(sliver_hrn, type='sliver').urn
record = SliverAllocation(sliver_id=sliver_id,
client_id=client_id,
component_id=component_id,
slice_urn=slice_urn,
allocation_state='geni_allocated')
record.sync(self.driver.api.dbsession())
return resulting_nodes
def allocate (self, urn, rspec_string, expiration, options=None):
if options is None: options={}
aggregate = OSAggregate(self)
rspec = RSpec(rspec_string)
xrn = Xrn(urn)
slice_hrn = xrn.get_hrn()
tenant_name = OSXrn(xrn=slice_hrn, type='slice').get_hrn()
instance_name = hrn_to_os_slicename(slice_hrn)
tenants = self.shell.auth_manager.tenants.findall()
# collect public keys & get the user name
users = options.get('geni_users', [])
pubkeys = []
key_name = None
if len(users) >= 1:
for user in users:
# TODO: We currently support one user name.
user_name = Xrn(user.get('urn')).get_hrn()
pubkeys.extend(user['keys'])
for tenant in tenants:
# Check if the tenant of the user exists in local OS or not
if tenant_name == tenant.name:
try:
self.shell.auth_manager.users.find(name=user_name)
except:
user = self.register_federation(user_hrn=user_name, \
slice_hrn=tenant_name, keys=pubkeys, email=None)
break
else:
user = self.register_federation(user_hrn=user_name, \
slice_hrn=tenant_name, keys=None, email=None)
# Update connection for the current client
self.shell.compute_manager.connect(username=user_name, tenant=tenant_name, password=user_name)
keypair_list = self.shell.compute_manager.keypairs.list()
keyname = OSXrn(xrn=user_name, type='user').get_slicename()
for keypair in keypair_list:
if keyname == keypair.name:
key_name = keypair.name
break
else:
raise SfaNotImplemented("No handle!")
# Update initial connection info
self.init_compute_manager_conn()
# key_name = aggregate.create_instance_key(slice_hrn, users[0])
# In case of federation or non-options
elif len(users) < 1:
if options.get('actual_caller_hrn') is None:
user_name = xrn.get_authority_hrn() + '.' + xrn.leaf.split('-')[0]
else:
user_name = options.get('actual_caller_hrn')
for tenant in tenants:
# Check if the tenant of the user in local OS or not
if tenant_name == tenant.name:
try:
self.shell.auth_manager.users.find(name=user_name)
except:
user = self.register_federation(user_hrn=user_name, \
slice_hrn=tenant_name, keys=pubkeys, email=None)
break
else:
user = self.register_federation(user_hrn=user_name, \
slice_hrn=tenant_name, keys=None, email=None)
# TODO: Wrapper for federation needs at least one pubkey of the user extracted by 'options'!!
# name = OSXrn(xrn=user_name, type='user').get_slicename()
# key_name = self.shell.compute_manager.keypairs.get(name).name
else:
raise SfaNotImplemented("No handle!")
slivers = aggregate.run_instances(tenant_name, user_name, rspec_string, key_name, pubkeys)
# Update sliver allocations
for sliver in slivers:
component_id = sliver.metadata.get('component_id')
sliver_id = OSXrn(name=('koren'+'.'+ sliver.name), id=sliver.id, type='node+openstack').get_urn()
record = SliverAllocation( sliver_id=sliver_id,
component_id=component_id,
allocation_state='geni_allocated')
record.sync(self.api.dbsession())
return aggregate.describe(urns=[urn], version=rspec.version)
def func(*args, **kwds):
if name == "list_resources":
(version, options) = args
slice_urn = slice_hrn = None
creds = []
rspec = getattr(self.driver,
"list_resources")(slice_urn, slice_hrn, [],
options)
result = rspec
elif name == "describe":
(urns, version, options) = args
slice_urn = urns[0]
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
rspec = getattr(self.driver,
"list_resources")(slice_urn, slice_hrn, creds,
options)
# SliverAllocation
if len(urns) == 1 and Xrn(xrn=urns[0]).type == 'slice':
constraint = SliverAllocation.slice_urn.in_(urns)
else:
constraint = SliverAllocation.sliver_id.in_(urns)
sliver_allocations = self.driver.api.dbsession().query(
SliverAllocation).filter(constraint)
sliver_status = getattr(self.driver,
"sliver_status")(slice_urn, slice_hrn)
if 'geni_expires' in sliver_status.keys():
geni_expires = sliver_status['geni_expires']
else:
geni_expires = ''
geni_slivers = []
for sliver_allocation in sliver_allocations:
geni_sliver = {}
geni_sliver['geni_expires'] = geni_expires
geni_sliver[
'geni_allocation'] = sliver_allocation.allocation_state
geni_sliver[
'geni_sliver_urn'] = sliver_allocation.sliver_id
geni_sliver['geni_error'] = ''
if geni_sliver['geni_allocation'] == 'geni_allocated':
geni_sliver[
'geni_operational_status'] = 'geni_pending_allocation'
else:
geni_sliver['geni_operational_status'] = 'geni_ready'
geni_slivers.append(geni_sliver)
result = {
'geni_urn': slice_urn,
'geni_rspec': rspec,
'geni_slivers': geni_slivers
}
elif name == "allocate":
(slice_urn, rspec_string, expiration, options) = args
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
users = options.get('sfa_users', [])
manifest_string = getattr(self.driver,
"create_sliver")(slice_urn,
slice_hrn, creds,
rspec_string, users,
options)
# slivers allocation
rspec = RSpec(manifest_string)
slivers = rspec.version.get_nodes_with_slivers()
##SliverAllocation
for sliver in slivers:
client_id = sliver['client_id']
component_id = sliver['component_id']
component_name = sliver['component_name']
slice_name = slice_hrn.replace('.', '-')
component_short_name = component_name.split('.')[0]
# self.driver.hrn
sliver_hrn = '%s.%s-%s' % (self.driver.hrn, slice_name,
component_short_name)
sliver_id = Xrn(sliver_hrn, type='sliver').urn
record = SliverAllocation(
sliver_id=sliver_id,
client_id=client_id,
component_id=component_id,
slice_urn=slice_urn,
allocation_state='geni_allocated')
record.sync()
# return manifest
rspec_version = RSpec(rspec_string).version
rspec_version_str = "%s" % rspec_version
options['geni_rspec_version'] = {
'version': rspec_version_str.split(' ')[1],
'type': rspec_version_str.lower().split(' ')[0]
}
result = self.describe([slice_urn], rspec_version, options)
elif name == "provision":
(urns, options) = args
if len(urns) == 1 and Xrn(xrn=urns[0]).type == 'slice':
constraint = SliverAllocation.slice_urn.in_(urns)
else:
constraint = SliverAllocation.sliver_id.in_(urns)
dbsession = self.driver.api.dbsession()
sliver_allocations = dbsession.query(SliverAllocation).filter(
constraint)
for sliver_allocation in sliver_allocations:
sliver_allocation.allocation_state = 'geni_provisioned'
dbsession.commit()
result = self.describe(urns, '', options)
elif name == "status":
urns = args
options = {}
options['geni_rspec_version'] = {
'version': '3',
'type': 'GENI'
}
descr = self.describe(urns[0], '', options)
result = {
'geni_urn': descr['geni_urn'],
'geni_slivers': descr['geni_slivers']
}
elif name == "delete":
(urns, options) = args
slice_urn = urns[0]
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
options['geni_rspec_version'] = {
'version': '3',
'type': 'GENI'
}
descr = self.describe(urns, '', options)
result = []
for sliver_allocation in descr['geni_slivers']:
geni_sliver = {
'geni_sliver_urn':
sliver_allocation['geni_sliver_urn'],
'geni_allocation_status': 'geni_unallocated',
'geni_expires': sliver_allocation['geni_expires'],
'geni_error': sliver_allocation['geni_error']
}
result.append(geni_sliver)
getattr(self.driver, "delete_sliver")(slice_urn, slice_hrn,
creds, options)
#SliverAllocation
constraints = SliverAllocation.slice_urn.in_(urns)
dbsession = self.driver.api.dbsession()
sliver_allocations = dbsession.query(SliverAllocation).filter(
constraints)
sliver_ids = [
sliver_allocation.sliver_id
for sliver_allocation in sliver_allocations
]
SliverAllocation.delete_allocations(sliver_ids, dbsession)
elif name == "renew":
(urns, expiration_time, options) = args
slice_urn = urns[0]
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
getattr(self.driver,
"renew_sliver")(slice_urn, slice_hrn, creds,
expiration_time, options)
options['geni_rspec_version'] = {
'version': '3',
'type': 'GENI'
}
descr = self.describe(urns, '', options)
result = descr['geni_slivers']
elif name == "perform_operational_action":
(urns, action, options) = args
options['geni_rspec_version'] = {
'version': '3',
'type': 'GENI'
}
result = self.describe(urns, '', options)['geni_slivers']
else:
# same as v2 ( registry methods)
result = getattr(self.driver, name)(*args, **kwds)
return result
def func(*args, **kwds):
if name == "list_resources":
(version, options) = args
slice_urn = slice_hrn = None
creds = []
rspec = getattr(self.driver, "list_resources")(slice_urn, slice_hrn, [], options)
result = rspec
elif name == "describe":
(urns, version, options) = args
slice_urn = urns[0]
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
rspec = getattr(self.driver, "list_resources")(slice_urn, slice_hrn, creds, options)
# SliverAllocation
if len(urns) == 1 and Xrn(xrn=urns[0]).type == 'slice':
constraint = SliverAllocation.slice_urn.in_(urns)
else:
constraint = SliverAllocation.sliver_id.in_(urns)
sliver_allocations = self.driver.api.dbsession().query (SliverAllocation).filter (constraint)
sliver_status = getattr(self.driver, "sliver_status")(slice_urn, slice_hrn)
if 'geni_expires' in sliver_status.keys():
geni_expires = sliver_status['geni_expires']
else:
geni_expires = ''
geni_slivers = []
for sliver_allocation in sliver_allocations:
geni_sliver = {}
geni_sliver['geni_expires'] = geni_expires
geni_sliver['geni_allocation'] = sliver_allocation.allocation_state
geni_sliver['geni_sliver_urn'] = sliver_allocation.sliver_id
geni_sliver['geni_error'] = ''
if geni_sliver['geni_allocation'] == 'geni_allocated':
geni_sliver['geni_operational_status'] = 'geni_pending_allocation'
else:
geni_sliver['geni_operational_status'] = 'geni_ready'
geni_slivers.append(geni_sliver)
result = {'geni_urn': slice_urn,
'geni_rspec': rspec,
'geni_slivers': geni_slivers}
elif name == "allocate":
(slice_urn, rspec_string, expiration, options) = args
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
users = options.get('sfa_users', [])
manifest_string = getattr(self.driver, "create_sliver")(slice_urn, slice_hrn, creds, rspec_string, users, options)
# slivers allocation
rspec = RSpec(manifest_string)
slivers = rspec.version.get_nodes_with_slivers()
##SliverAllocation
for sliver in slivers:
client_id = sliver['client_id']
component_id = sliver['component_id']
component_name = sliver['component_name']
slice_name = slice_hrn.replace('.','-')
component_short_name = component_name.split('.')[0]
# self.driver.hrn
sliver_hrn = '%s.%s-%s' % (self.driver.hrn, slice_name, component_short_name)
sliver_id = Xrn(sliver_hrn, type='sliver').urn
record = SliverAllocation(sliver_id=sliver_id,
client_id=client_id,
component_id=component_id,
slice_urn = slice_urn,
allocation_state='geni_allocated')
record.sync(self.driver.api.dbsession())
# return manifest
rspec_version = RSpec(rspec_string).version
rspec_version_str = "%s"%rspec_version
options['geni_rspec_version'] = {'version': rspec_version_str.split(' ')[1], 'type': rspec_version_str.lower().split(' ')[0]}
result = self.describe([slice_urn], rspec_version, options)
elif name == "provision":
(urns, options) = args
if len(urns) == 1 and Xrn(xrn=urns[0]).type == 'slice':
constraint = SliverAllocation.slice_urn.in_(urns)
else:
constraint = SliverAllocation.sliver_id.in_(urns)
dbsession = self.driver.api.dbsession()
sliver_allocations = dbsession.query (SliverAllocation).filter(constraint)
for sliver_allocation in sliver_allocations:
sliver_allocation.allocation_state = 'geni_provisioned'
dbsession.commit()
result = self.describe(urns, '', options)
elif name == "status":
urns = args
options = {}
options['geni_rspec_version'] = {'version': '3', 'type': 'GENI'}
descr = self.describe(urns[0], '', options)
result = {'geni_urn': descr['geni_urn'],
'geni_slivers': descr['geni_slivers']}
elif name == "delete":
(urns, options) = args
slice_urn = urns[0]
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
options['geni_rspec_version'] = {'version': '3', 'type': 'GENI'}
descr = self.describe(urns, '', options)
result = []
for sliver_allocation in descr['geni_slivers']:
geni_sliver = {'geni_sliver_urn': sliver_allocation['geni_sliver_urn'],
'geni_allocation_status': 'geni_unallocated',
'geni_expires': sliver_allocation['geni_expires'],
'geni_error': sliver_allocation['geni_error']}
result.append(geni_sliver)
getattr(self.driver, "delete_sliver")(slice_urn, slice_hrn, creds, options)
#SliverAllocation
constraints = SliverAllocation.slice_urn.in_(urns)
dbsession = self.driver.api.dbsession()
sliver_allocations = dbsession.query(SliverAllocation).filter(constraints)
sliver_ids = [sliver_allocation.sliver_id for sliver_allocation in sliver_allocations]
SliverAllocation.delete_allocations(sliver_ids, dbsession)
elif name == "renew":
(urns, expiration_time, options) = args
slice_urn = urns[0]
slice_hrn, type = urn_to_hrn(slice_urn)
creds = []
getattr(self.driver, "renew_sliver")(slice_urn, slice_hrn, creds, expiration_time, options)
options['geni_rspec_version'] = {'version': '3', 'type': 'GENI'}
descr = self.describe(urns, '', options)
result = descr['geni_slivers']
elif name == "perform_operational_action":
(urns, action, options) = args
options['geni_rspec_version'] = {'version': '3', 'type': 'GENI'}
result = self.describe(urns, '', options)['geni_slivers']
else:
# same as v2 ( registry methods)
result=getattr(self.driver, name)(*args, **kwds)
return result
