140 lines
5.0 KiB
Python
140 lines
5.0 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import eventlet
|
|
import logging
|
|
import json
|
|
|
|
from heat.common import exception
|
|
from heat.db import api as db_api
|
|
from heat.engine import resources
|
|
|
|
logger = logging.getLogger('heat.engine.wait_condition')
|
|
|
|
|
|
class WaitConditionHandle(resources.Resource):
|
|
'''
|
|
the main point of this class is to :
|
|
have no dependancies (so the instance can reference it)
|
|
generate a unique url (to be returned in the refernce)
|
|
then the cfn-signal will use this url to post to and
|
|
WaitCondition will poll it to see if has been written to.
|
|
'''
|
|
properties_schema = {}
|
|
|
|
def __init__(self, name, json_snippet, stack):
|
|
super(WaitConditionHandle, self).__init__(name, json_snippet, stack)
|
|
|
|
def handle_create(self):
|
|
self.instance_id = '%s/stacks/%s/resources/%s' % \
|
|
(resources.metadata_server(),
|
|
self.stack.id,
|
|
self.name)
|
|
|
|
|
|
class WaitCondition(resources.Resource):
|
|
properties_schema = {'Handle': {'Type': 'String',
|
|
'Required': True},
|
|
'Timeout': {'Type': 'Number',
|
|
'Required': True,
|
|
'MinValue': '1'},
|
|
'Count': {'Type': 'Number',
|
|
'MinValue': '1'}}
|
|
|
|
def __init__(self, name, json_snippet, stack):
|
|
super(WaitCondition, self).__init__(name, json_snippet, stack)
|
|
self.resource_id = None
|
|
|
|
self.timeout = int(self.t['Properties']['Timeout'])
|
|
self.count = int(self.t['Properties'].get('Count', '1'))
|
|
|
|
def _get_handle_resource_id(self):
|
|
if self.resource_id is None:
|
|
self.calculate_properties()
|
|
handle_url = self.properties['Handle']
|
|
self.resource_id = handle_url.split('/')[-1]
|
|
return self.resource_id
|
|
|
|
def handle_create(self):
|
|
self._get_handle_resource_id()
|
|
res_name = self.resource_id
|
|
cntx = self.stack.context
|
|
|
|
# keep polling our Metadata to see if the cfn-signal has written
|
|
# it yet. The execution here is limited by timeout.
|
|
tmo = eventlet.Timeout(self.timeout)
|
|
status = 'WAITING'
|
|
reason = ''
|
|
res = None
|
|
sleep_time = 1
|
|
try:
|
|
while status == 'WAITING':
|
|
try:
|
|
res = db_api.resource_get_by_name_and_stack(cntx,
|
|
res_name,
|
|
self.stack.id)
|
|
except Exception as ex:
|
|
logger.exception('resource %s not found' % res_name)
|
|
if 'not found' in ex:
|
|
# it has been deleted
|
|
status = 'DELETED'
|
|
else:
|
|
pass
|
|
|
|
if res:
|
|
if res.rsrc_metadata:
|
|
meta = res.rsrc_metadata
|
|
status = meta.get('Status',
|
|
'WAITING')
|
|
reason = meta.get('Reason',
|
|
'Reason not provided')
|
|
logger.debug('got %s' % json.dumps(res.rsrc_metadata))
|
|
if status == 'WAITING':
|
|
logger.debug('Waiting some more for the Metadata[Status]')
|
|
eventlet.sleep(sleep_time)
|
|
sleep_time = min(sleep_time * 2, self.timeout / 4)
|
|
if res:
|
|
res.expire()
|
|
|
|
except eventlet.Timeout, t:
|
|
if t is not tmo:
|
|
# not my timeout
|
|
raise
|
|
else:
|
|
status = 'TIMEDOUT'
|
|
reason = 'Timed out waiting for instance'
|
|
finally:
|
|
tmo.cancel()
|
|
|
|
if status != 'SUCCESS':
|
|
raise exception.Error(reason)
|
|
|
|
def FnGetAtt(self, key):
|
|
res = None
|
|
if key == 'Data':
|
|
try:
|
|
r = db_api.resource_get(self.stack.context, self.id)
|
|
if r.rsrc_metadata and 'Data' in r.rsrc_metadata:
|
|
res = r.rsrc_metadata['Data']
|
|
except Exception as ex:
|
|
pass
|
|
|
|
else:
|
|
raise exception.InvalidTemplateAttribute(resource=self.name,
|
|
key=key)
|
|
|
|
logger.debug('%s.GetAtt(%s) == %s' % (self.name, key, res))
|
|
return unicode(res)
|