from sfa.planetlab.plxrn import PlXrn, hostname_to_urn, hrn_to_pl_slicename, slicename_to_hrn
from sfa.planetlab.vlink import get_tc_rate
from sfa.planetlab.topology import Topology
+from sfa.storage.alchemy import dbsession
+from sfa.storage.model import SliverAllocation
+
import time
pl_initscripts[initscript['initscript_id']] = initscript
return pl_initscripts
- def get_slivers(self, urns, options):
+ def get_slivers(self, urns, options={}):
names = set()
slice_ids = set()
node_ids = []
node.update(slices[0])
node['tags'] = tags_dict[node['node_id']]
sliver_hrn = '%s.%s-%s' % (self.driver.hrn, slice['slice_id'], node['node_id'])
- node['urn'] = PlXrn(xrn=sliver_hrn, type='sliver').get_urn()
+ node['sliver_id'] = Xrn(sliver_hrn, type='sliver').urn
+ node['urn'] = node['sliver_id']
slivers.append(node)
return slivers
nodes_dict[node['node_id']] = node
return nodes_dict
- def rspec_node_to_geni_sliver(self, rspec_node, allocation_status=None):
+ def rspec_node_to_geni_sliver(self, rspec_node):
op_status = "geni_unknown"
state = rspec_node['boot_state'].lower()
if state == 'boot':
else:
op_status =' geni_failed'
- if not allocation_status:
- allocation_status = 'geni_provisioned'
# required fields
geni_sliver = {'geni_sliver_urn': rspec_node['sliver_id'],
'geni_expires': rspec_node['expires'],
- 'geni_allocation_status': allocation_status,
'geni_operational_status': op_status,
'geni_error': None,
}
rspec.version.add_links(links)
return rspec.toxml()
- def describe(self, urns, version=None, options={}, allocation_status=None):
+ def describe(self, urns, version=None, options={}):
version_manager = VersionManager()
version = version_manager.get_version(version)
rspec_version = version_manager._get_version(version.type, version.version, 'manifest')
if len(slivers) == 0:
raise SliverDoesNotExist("You have not allocated any slivers here for %s" % str(urns))
rspec.xml.set('expires', datetime_to_string(utcparse(slivers[0]['expires'])))
+
+ # lookup the sliver allocations
+ sliver_ids = [sliver['sliver_id'] for sliver in slivers]
+ constraint = SliverAllocation.sliver_id.in_(sliver_ids)
+ sliver_allocations = dbsession.query(SliverAllocation).filter(constraint)
+ sliver_allocation_dict = {}
+ for sliver_allocation in sliver_allocations:
+ sliver_allocation_dict[sliver_allocation.sliver_id] = sliver_allocation
if not options.get('list_leases') or options['list_leases'] != 'leases':
# add slivers
if sliver['slice_ids_whitelist'] and sliver['slice_id'] not in sliver['slice_ids_whitelist']:
continue
rspec_node = self.sliver_to_rspec_node(sliver, sites, interfaces, node_tags, pl_initscripts)
- geni_sliver = self.rspec_node_to_geni_sliver(rspec_node, allocation_status=allocation_status)
+ geni_sliver = self.rspec_node_to_geni_sliver(rspec_node)
+ sliver_allocation = sliver_allocation_dict[sliver['sliver_id']]
+ geni_sliver['geni_allocation_status'] = sliver_allocation.allocation_state
rspec_nodes.append(rspec_node)
geni_slivers.append(geni_sliver)
rspec.version.add_nodes(rspec_nodes)
# one would think the driver should not need to mess with the SFA db, but..
from sfa.storage.alchemy import dbsession
-from sfa.storage.model import RegRecord
+from sfa.storage.model import RegRecord, SliverAllocation
# used to be used in get_ticket
#from sfa.trust.sfaticket import SfaTicket
desc = aggregate.describe(urns)
return desc['geni_slivers']
- # find out where this slice is currently running
- slicename = hrn_to_pl_slicename(slice_hrn)
-
- slices = self.shell.GetSlices([slicename], ['slice_id', 'node_ids','person_ids','name','expires'])
- if len(slices) == 0:
- raise SliverDoesNotExist("%s (used %s as slicename internally)" % (slice_hrn, slicename))
- slice = slices[0]
-
- # report about the local nodes only
- nodes = self.shell.GetNodes({'node_id':slice['node_ids'],'peer_id':None},
- ['node_id', 'hostname', 'site_id', 'boot_state', 'last_contact'])
-
- if len(nodes) == 0:
- raise SliverDoesNotExist("You have not allocated any slivers here")
-
- # get login info
- user = {}
- if slice['person_ids']:
- persons = self.shell.GetPersons(slice['person_ids'], ['key_ids'])
- key_ids = [key_id for person in persons for key_id in person['key_ids']]
- person_keys = self.shell.GetKeys(key_ids)
- keys = [key['key'] for key in person_keys]
-
- user.update({'urn': slice_urn,
- 'login': slice['name'],
- 'protocol': ['ssh'],
- 'port': ['22'],
- 'keys': keys})
-
- site_ids = [node['site_id'] for node in nodes]
-
- result = {}
- result['geni_urn'] = slice_urn
- result['pl_login'] = slice['name']
- result['pl_expires'] = datetime_to_string(utcparse(slice['expires']))
- result['geni_expires'] = datetime_to_string(utcparse(slice['expires']))
-
- resources = []
- for node in nodes:
- res = {}
- res['pl_hostname'] = node['hostname']
- res['pl_boot_state'] = node['boot_state']
- res['pl_last_contact'] = node['last_contact']
- res['geni_expires'] = datetime_to_string(utcparse(slice['expires']))
- if node['last_contact'] is not None:
-
- res['pl_last_contact'] = datetime_to_string(utcparse(node['last_contact']))
- sliver_id = "%s:%s" % (slice['slice_id'], node['node_id'])
- sliver_xrn = Xrn(slice_urn, id = sliver_id)
- sliver_xrn.set_authority(self.hrn)
- res['geni_urn'] = sliver_xrn.get_urn()
- if node['boot_state'] == 'boot':
- res['geni_status'] = 'ready'
- else:
- res['geni_status'] = 'failed'
- res['geni_allocation_status'] = 'geni_provisioned'
-
- res['geni_error'] = ''
- res['users'] = [user]
- resources.append(res)
-
- result['geni_resources'] = resources
- return result
-
def allocate (self, urn, rspec_string, options={}):
xrn = Xrn(urn)
aggregate = PlAggregate(self)
hostname = xrn_to_hostname(node.get('component_id').strip())
if hostname:
requested_slivers.append(hostname)
- nodes = slices.verify_slice_nodes(slice, requested_slivers, peer)
+ nodes = slices.verify_slice_nodes(slice, requested_slivers, peer)
+
+ # update all sliver allocation states setting then to geni_allocated
+ sliver_state_updated = {}
+ for node in nodes:
+ sliver_hrn = '%s.%s-%s' % (self.hrn, slice['slice_id'], node['node_id'])
+ sliver_id = Xrn(sliver_hrn, type='sliver').urn
+ sliver_state_updated[sliver_id] = False
+
+ constraint = SliverAllocation.sliver_id.in_(sliver_state_updated.keys())
+ cur_sliver_allocations = dbsession.query(SliverAllocation).filter(constraint)
+ for sliver_allocation in cur_sliver_allocations:
+ sliver_allocation.allocation_state = 'geni_allocated'
+ sliver_state_updated[sliver_allocation.sliver_id] = True
+ dbsession.commit()
+
+ # Some states may not have been updated becuase no sliver allocation state record
+ # exists for the sliver. Insert new allocation records for these slivers and set
+ # it to geni_allocated.
+ for (sliver_id, state_updated) in sliver_state_updated.items():
+ if state_updated == False:
+ record = SliverAllocation(sliver_id=sliver_id, allocation_state='geni_allocated')
+ dbsession.add(record)
+ dbsession.commit()
# add/remove links links
slices.verify_slice_links(slice, rspec.version.get_link_requests(), nodes)
requested_leases.append(requested_lease)
leases = slices.verify_slice_leases(slice, requested_leases, kept_leases, peer)
-
# handle MyPLC peer association.
# only used by plc and ple.
slices.handle_peer(site, slice, persons, peer)
- return aggregate.describe([xrn.get_urn()], version=rspec.version, allocation_status='geni_allocated')
+ return aggregate.describe([xrn.get_urn()], version=rspec.version)
def provision(self, urns, options={}):
- return self.describe(urns, None, options=options, allocation_status='geni_provisioned')
+ # update sliver allocation states and set them to geni_provisioned
+ aggregate = PlAggregate(self)
+ slivers = aggregate.get_slivers(urns)
+ sliver_ids = [sliver['sliver_id'] for sliver in slivers]
+ constraint = SliverAllocation.sliver_id.in_(sliver_ids)
+ cur_sliver_allocations = dbsession.query(SliverAllocation).filter(constraint)
+ for sliver_allocation in cur_sliver_allocations:
+ sliver_allocation.allocation_state = 'geni_provisioned'
+ dbsession.commit()
+
+ return self.describe(urns, None, options=options)
def delete(self, urns, options={}):
- # urns argument may contain slice or sliver urns. Slice urns contain the
- # a slice's name whereas sliver urns contain a slice_id, node_id tupple.
- # We will collect names/ids specified in the urns provided by the caller.
- slice_names = []
- slice_ids = []
+
+ aggregate = PlAggregate(self)
+ slivers = aggregate.get_slivers(urns)
+ slice_id = slivers[0]['slice_id']
node_ids = []
+ sliver_ids = []
+ for sliver in slivers:
+ node_ids.append(sliver['node_id'])
+ sliver_ids.append(sliver['sliver_id'])
- for urn in urns:
- xrn = PlXrn(xrn=urn)
- if xrn.type == 'slice' and xrn.pl_slicename() not in slice_names:
- slice_names.append(xrn.pl_slicename())
- elif xrn.type == 'sliver':
- leaf = xrn.leaf
- leaf_split = leaf.split('-')
- if len(leaf_split) > 1:
- slice_ids.append(leaf_split[0])
- node_ids.append(leaf_split[1])
-
- # look up the requested urns
- filter = {}
- if slice_names:
- filter['name'] = slice_names
- if slice_ids:
- filter['id'] = slice_ids
- slices = self.shell.GetSlices(filter)
- if not slices:
- raise SearchFailed(urns)
- slice = slices[0]
- if not node_ids:
- node_ids = slice['node_ids']
# determine if this is a peer slice
# xxx I wonder if this would not need to use PlSlices.get_peer instead
# in which case plc.peers could be deprecated as this here
# is the only/last call to this last method in plc.peers
- slice_hrn = PlXrn(auth=self.hrn, slicename=slice['name']).get_hrn()
+ slice_hrn = PlXrn(auth=self.hrn, slicename=slivers[0]['name']).get_hrn()
peer = peers.get_peer(self, slice_hrn)
try:
if peer:
- self.shell.UnBindObjectFromPeer('slice', slice['slice_id'], peer)
-
- self.shell.DeleteSliceFromNodes(slice['slice_id'], node_ids)
+ self.shell.UnBindObjectFromPeer('slice', slice_id, peer)
+
+ self.shell.DeleteSliceFromNodes(slice_id, node_ids)
+
+ # update slivera allocation states
+ constraint = SliverAllocation.sliver_id.in_(sliver_ids)
+ cur_sliver_allocations = dbsession.query(SliverAllocation).filter(constraint)
+ for sliver_allocation in cur_sliver_allocations:
+ sliver_allocation.allocation_state = 'geni_unallocated'
+ dbsession.commit()
finally:
if peer:
- self.shell.BindObjectToPeer('slice', slice['slice_id'], peer, slice['peer_slice_id'])
+ self.shell.BindObjectToPeer('slice', slice_id, peer, slice['peer_slice_id'])
+
+
# prepare return struct
geni_slivers = []
for node_id in node_ids:
- sliver_hrn = '%s.%s-%s' % (self.hrn, slice['slice_id'], node_id)
+ sliver_hrn = '%s.%s-%s' % (self.hrn, slice_id, node_id)
geni_slivers.append(
{'geni_sliver_urn': Xrn(sliver_hrn, type='sliver').urn,
'geni_allocation_status': 'geni_unallocated',
- 'geni_expires': datetime_to_string(utcparse(slice['expires']))})
+ 'geni_expires': datetime_to_string(utcparse(slivers[0]['expires']))})
return geni_slivers
def renew (self, urns, expiration_time, options={}):