def start_shudder():
logging.info('Started shudder.')
uncatchable = ['SIG_DFL', 'SIGSTOP', 'SIGKILL']
for i in [x for x in dir(signal) if x.startswith("SIG")]:
if not i in uncatchable:
signum = getattr(signal, i)
signal.signal(signum, receive_signal)
queue_url = queue.create_queue()
sns_connection, subscription_arn = queue.subscribe_sns(queue_url)
while True:
message = queue.poll_queue(queue_url)
if message or metadata.poll_instance_metadata():
queue.clean_up_sns(sns_connection, subscription_arn, queue_url)
if 'endpoint' in CONFIG:
call_endpoint(CONFIG["endpoint"])
if 'endpoints' in CONFIG:
for endpoint in CONFIG["endpoints"]:
call_endpoint(endpoint)
if 'commands' in CONFIG:
for command in CONFIG["commands"]:
run_command(message, command)
logging.info('Sending a COMPLETE lifecycle action.')
queue.complete_lifecycle_action(message)
logging.info('Finished successfully. Exiting now.')
sys.exit(0)
logging.info('Waiting for TERMINATION trigger.')
time.sleep(10)
def receive_signal(signum, stack):
if signum in [1, 2, 3, 15]:
print 'Caught signal %s, exiting.' % (str(signum))
queue.clean_up_sns(sns_connection, subscription_arn, sqs_queue)
sys.exit()
else:
print 'Caught signal %s, ignoring.' % (str(signum))
if __name__ == '__main__':
uncatchable = ['SIG_DFL','SIGSTOP','SIGKILL']
for i in [x for x in dir(signal) if x.startswith("SIG")]:
if not i in uncatchable:
signum = getattr(signal,i)
signal.signal(signum,receive_signal)
sqs_connection, sqs_queue = queue.create_queue()
sns_connection, subscription_arn = queue.subscribe_sns(sqs_queue)
running = True
while running:
try:
message = queue.poll_queue(sqs_connection, sqs_queue)
if message or metadata.poll_instance_metadata():
queue.clean_up_sns(sns_connection, subscription_arn, sqs_queue)
if 'endpoint' in CONFIG:
requests.get(CONFIG["endpoint"])
if 'endpoints' in CONFIG:
for endpoint in CONFIG["endpoints"]:
requests.get(endpoint)
if 'commands' in CONFIG:
for command in CONFIG["commands"]:
print 'Running command: %s' % command
process = subprocess.Popen(command)
while process.poll() is None:
time.sleep(30)
"""Send a heart beat to aws"""
try:
queue.record_lifecycle_action_heartbeat(message)
except:
# 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.
"""Start polling of SQS and metadata."""
import shudder.queue as queue
import shudder.metadata as metadata
from shudder.config import CONFIG
import time
import requests
if __name__ == '__main__':
sqs_connection, sqs_queue = queue.create_queue()
sns_connection, subscription_arn = queue.subscribe_sns(sqs_queue)
while True:
if queue.poll_queue(sqs_connection, sqs_queue) \
or metadata.poll_instance_metadata():
queue.clean_up_sns(sns_connection, subscription_arn, sqs_queue)
requests.get(CONFIG["endpoint"])
break
time.sleep(5)
