mirror of
https://github.com/quay/quay.git
synced 2026-01-27 18:42:52 +03:00
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
import logging
|
|
import os
|
|
import time
|
|
import socket
|
|
|
|
import features
|
|
|
|
from app import app, userfiles as user_files, build_logs, dockerfile_build_queue
|
|
from util.log import logfile_path
|
|
|
|
from buildman.manager.enterprise import EnterpriseManager
|
|
from buildman.manager.ephemeral import EphemeralBuilderManager
|
|
from buildman.server import BuilderServer
|
|
|
|
from trollius import SSLContext
|
|
from raven.handlers.logging import SentryHandler
|
|
from raven.conf import setup_logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
BUILD_MANAGERS = {
|
|
"enterprise": EnterpriseManager,
|
|
"ephemeral": EphemeralBuilderManager,
|
|
}
|
|
|
|
EXTERNALLY_MANAGED = "external"
|
|
|
|
DEFAULT_WEBSOCKET_PORT = 8787
|
|
DEFAULT_CONTROLLER_PORT = 8686
|
|
|
|
LOG_FORMAT = "%(asctime)s [%(process)d] [%(levelname)s] [%(name)s] %(message)s"
|
|
|
|
|
|
def run_build_manager():
|
|
if not features.BUILD_SUPPORT:
|
|
logger.debug("Building is disabled. Please enable the feature flag")
|
|
while True:
|
|
time.sleep(1000)
|
|
return
|
|
|
|
if app.config.get("REGISTRY_STATE", "normal") == "readonly":
|
|
logger.debug("Building is disabled while in read-only mode.")
|
|
while True:
|
|
time.sleep(1000)
|
|
return
|
|
|
|
build_manager_config = app.config.get("BUILD_MANAGER")
|
|
if build_manager_config is None:
|
|
return
|
|
|
|
# If the build system is externally managed, then we just sleep this process.
|
|
if build_manager_config[0] == EXTERNALLY_MANAGED:
|
|
logger.debug("Builds are externally managed.")
|
|
while True:
|
|
time.sleep(1000)
|
|
return
|
|
|
|
logger.debug('Asking to start build manager with lifecycle "%s"', build_manager_config[0])
|
|
manager_klass = BUILD_MANAGERS.get(build_manager_config[0])
|
|
if manager_klass is None:
|
|
return
|
|
|
|
manager_hostname = os.environ.get(
|
|
"BUILDMAN_HOSTNAME", app.config.get("BUILDMAN_HOSTNAME", app.config["SERVER_HOSTNAME"])
|
|
)
|
|
websocket_port = int(
|
|
os.environ.get(
|
|
"BUILDMAN_WEBSOCKET_PORT",
|
|
app.config.get("BUILDMAN_WEBSOCKET_PORT", DEFAULT_WEBSOCKET_PORT),
|
|
)
|
|
)
|
|
controller_port = int(
|
|
os.environ.get(
|
|
"BUILDMAN_CONTROLLER_PORT",
|
|
app.config.get("BUILDMAN_CONTROLLER_PORT", DEFAULT_CONTROLLER_PORT),
|
|
)
|
|
)
|
|
|
|
logger.debug(
|
|
"Will pass buildman hostname %s to builders for websocket connection", manager_hostname
|
|
)
|
|
|
|
logger.debug('Starting build manager with lifecycle "%s"', build_manager_config[0])
|
|
ssl_context = None
|
|
if os.environ.get("SSL_CONFIG"):
|
|
logger.debug("Loading SSL cert and key")
|
|
ssl_context = SSLContext()
|
|
ssl_context.load_cert_chain(
|
|
os.path.join(os.environ.get("SSL_CONFIG"), "ssl.cert"),
|
|
os.path.join(os.environ.get("SSL_CONFIG"), "ssl.key"),
|
|
)
|
|
|
|
server = BuilderServer(
|
|
app.config["SERVER_HOSTNAME"],
|
|
dockerfile_build_queue,
|
|
build_logs,
|
|
user_files,
|
|
manager_klass,
|
|
build_manager_config[1],
|
|
manager_hostname,
|
|
)
|
|
server.run("0.0.0.0", websocket_port, controller_port, ssl=ssl_context)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.config.fileConfig(logfile_path(debug=True), disable_existing_loggers=False)
|
|
logging.getLogger("peewee").setLevel(logging.WARN)
|
|
logging.getLogger("boto").setLevel(logging.WARN)
|
|
|
|
if app.config.get("EXCEPTION_LOG_TYPE", "FakeSentry") == "Sentry":
|
|
buildman_name = "%s:buildman" % socket.gethostname()
|
|
setup_logging(
|
|
SentryHandler(app.config.get("SENTRY_DSN", ""), name=buildman_name, level=logging.ERROR)
|
|
)
|
|
|
|
run_build_manager()
|