zuul/zuul/cmd/web.py

123 lines
4.2 KiB
Python
Executable File

#!/usr/bin/env python
# Copyright 2017 Red Hat, Inc.
#
# 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 asyncio
import logging
import signal
import sys
import threading
import zuul.cmd
import zuul.web
from zuul.driver.sql import sqlconnection
from zuul.lib.config import get_default
class WebServer(zuul.cmd.ZuulDaemonApp):
app_name = 'web'
app_description = 'A standalone Zuul web server.'
def exit_handler(self, signum, frame):
self.web.stop()
def _run(self):
params = dict()
params['listen_address'] = get_default(self.config,
'web', 'listen_address',
'127.0.0.1')
params['listen_port'] = get_default(self.config, 'web', 'port', 9000)
params['static_cache_expiry'] = get_default(self.config, 'web',
'static_cache_expiry',
3600)
params['gear_server'] = get_default(self.config, 'gearman', 'server')
params['gear_port'] = get_default(self.config, 'gearman', 'port', 4730)
params['ssl_key'] = get_default(self.config, 'gearman', 'ssl_key')
params['ssl_cert'] = get_default(self.config, 'gearman', 'ssl_cert')
params['ssl_ca'] = get_default(self.config, 'gearman', 'ssl_ca')
sql_conn_name = get_default(self.config, 'web',
'sql_connection_name')
sql_conn = None
if sql_conn_name:
# we want a specific sql connection
sql_conn = self.connections.connections.get(sql_conn_name)
if not sql_conn:
self.log.error("Couldn't find sql connection '%s'" %
sql_conn_name)
sys.exit(1)
else:
# look for any sql connection
connections = [c for c in self.connections.connections.values()
if isinstance(c, sqlconnection.SQLConnection)]
if len(connections) > 1:
self.log.error("Multiple sql connection found, "
"set the sql_connection_name option "
"in zuul.conf [web] section")
sys.exit(1)
if connections:
# use this sql connection by default
sql_conn = connections[0]
params['sql_connection'] = sql_conn
try:
self.web = zuul.web.ZuulWeb(**params)
except Exception as e:
self.log.exception("Error creating ZuulWeb:")
sys.exit(1)
loop = asyncio.get_event_loop()
signal.signal(signal.SIGUSR1, self.exit_handler)
signal.signal(signal.SIGTERM, self.exit_handler)
self.log.info('Zuul Web Server starting')
self.thread = threading.Thread(target=self.web.run,
args=(loop,),
name='web')
self.thread.start()
try:
signal.pause()
except KeyboardInterrupt:
print("Ctrl + C: asking web server to exit nicely...\n")
self.exit_handler(signal.SIGINT, None)
self.thread.join()
loop.stop()
loop.close()
self.log.info("Zuul Web Server stopped")
def run(self):
self.setup_logging('web', 'log_config')
self.log = logging.getLogger("zuul.WebServer")
self.configure_connections()
signal.signal(signal.SIGUSR2, zuul.cmd.stack_dump_handler)
try:
self._run()
except Exception:
self.log.exception("Exception from WebServer:")
def main():
WebServer().main()
if __name__ == "__main__":
main()