mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
57101500ea
* fixing pull statistics Redis flush worker with aggregation support --------- Co-authored-by: shudeshp <shudeshp@redhat.com>
149 lines
5.8 KiB
Python
149 lines
5.8 KiB
Python
import os
|
|
import os.path
|
|
import sys
|
|
from typing import List, Union
|
|
|
|
import jinja2
|
|
|
|
QUAYPATH = os.getenv("QUAYPATH", ".")
|
|
QUAYDIR = os.getenv("QUAYDIR", "/")
|
|
QUAYCONF_DIR = os.getenv("QUAYCONF", os.path.join(QUAYDIR, QUAYPATH, "conf"))
|
|
QUAYRUN_DIR = os.getenv("QUAYRUN", QUAYCONF_DIR)
|
|
|
|
QUAY_LOGGING = os.getenv("QUAY_LOGGING", "stdout") # or "syslog"
|
|
QUAY_HOTRELOAD: bool = os.getenv("QUAY_HOTRELOAD", "false") == "true"
|
|
|
|
QUAY_SERVICES: Union[List, str] = os.getenv("QUAY_SERVICES", [])
|
|
QUAY_OVERRIDE_SERVICES: Union[List, str] = os.getenv("QUAY_OVERRIDE_SERVICES", [])
|
|
|
|
|
|
def registry_services():
|
|
return {
|
|
"blobuploadcleanupworker": {"autostart": "true"},
|
|
"buildlogsarchiver": {"autostart": "true"},
|
|
"builder": {"autostart": "true"},
|
|
"chunkcleanupworker": {"autostart": "true"},
|
|
"expiredappspecifictokenworker": {"autostart": "true"},
|
|
"exportactionlogsworker": {"autostart": "true"},
|
|
"gcworker": {"autostart": "true"},
|
|
"globalpromstats": {"autostart": "true"},
|
|
"logrotateworker": {"autostart": "true"},
|
|
"namespacegcworker": {"autostart": "true"},
|
|
"repositorygcworker": {"autostart": "true"},
|
|
"notificationworker": {"autostart": "true"},
|
|
"queuecleanupworker": {"autostart": "true"},
|
|
"reconciliationworker": {"autostart": "true"},
|
|
"repositoryactioncounter": {"autostart": "true"},
|
|
"securityworker": {"autostart": "true"},
|
|
"storagereplication": {"autostart": "true"},
|
|
"teamsyncworker": {"autostart": "true"},
|
|
"dnsmasq": {"autostart": "true"},
|
|
"gunicorn-registry": {"autostart": "true"},
|
|
"gunicorn-secscan": {"autostart": "true"},
|
|
"gunicorn-web": {"autostart": "true"},
|
|
"ip-resolver-update-worker": {"autostart": "true"},
|
|
"memcache": {"autostart": "true"},
|
|
"nginx": {"autostart": "true"},
|
|
"pushgateway": {"autostart": "true"},
|
|
"servicekey": {"autostart": "true"},
|
|
"repomirrorworker": {"autostart": "false"},
|
|
"manifestbackfillworker": {"autostart": "true"},
|
|
"manifestsubjectbackfillworker": {"autostart": "true"},
|
|
"securityscanningnotificationworker": {"autostart": "true"},
|
|
"config-editor": {"autostart": "false"},
|
|
"quotatotalworker": {"autostart": "true"},
|
|
"quotaregistrysizeworker": {"autostart": "true"},
|
|
"autopruneworker": {"autostart": "true"},
|
|
"proxycacheblobworker": {"autostart": "true"},
|
|
"pullstatsredisflushworker": {"autostart": "true"},
|
|
}
|
|
|
|
|
|
def config_services():
|
|
return {
|
|
"blobuploadcleanupworker": {"autostart": "false"},
|
|
"buildlogsarchiver": {"autostart": "false"},
|
|
"builder": {"autostart": "false"},
|
|
"chunkcleanupworker": {"autostart": "false"},
|
|
"expiredappspecifictokenworker": {"autostart": "false"},
|
|
"exportactionlogsworker": {"autostart": "false"},
|
|
"gcworker": {"autostart": "false"},
|
|
"globalpromstats": {"autostart": "false"},
|
|
"logrotateworker": {"autostart": "false"},
|
|
"namespacegcworker": {"autostart": "false"},
|
|
"repositorygcworker": {"autostart": "false"},
|
|
"notificationworker": {"autostart": "false"},
|
|
"queuecleanupworker": {"autostart": "false"},
|
|
"repositoryactioncounter": {"autostart": "false"},
|
|
"reconciliationworker": {"autostart": "false"},
|
|
"securityworker": {"autostart": "false"},
|
|
"storagereplication": {"autostart": "false"},
|
|
"teamsyncworker": {"autostart": "false"},
|
|
"dnsmasq": {"autostart": "false"},
|
|
"gunicorn-registry": {"autostart": "false"},
|
|
"gunicorn-secscan": {"autostart": "false"},
|
|
"gunicorn-web": {"autostart": "false"},
|
|
"ip-resolver-update-worker": {"autostart": "false"},
|
|
"memcache": {"autostart": "false"},
|
|
"nginx": {"autostart": "false"},
|
|
"pushgateway": {"autostart": "false"},
|
|
"servicekey": {"autostart": "false"},
|
|
"repomirrorworker": {"autostart": "false"},
|
|
"manifestbackfillworker": {"autostart": "false"},
|
|
"manifestsubjectbackfillworker": {"autostart": "false"},
|
|
"securityscanningnotificationworker": {"autostart": "false"},
|
|
"config-editor": {"autostart": "true"},
|
|
"quotatotalworker": {"autostart": "false"},
|
|
"quotaregistrysizeworker": {"autostart": "false"},
|
|
"autopruneworker": {"autostart": "false"},
|
|
"proxycacheblobworker": {"autostart": "false"},
|
|
"pullstatsredisflushworker": {"autostart": "false"},
|
|
}
|
|
|
|
|
|
def generate_supervisord_config(filename, config, logdriver, hotreload):
|
|
with open(filename + ".jnj") as f:
|
|
template = jinja2.Template(f.read())
|
|
rendered = template.render(config=config, logdriver=logdriver, hotreload=hotreload)
|
|
|
|
with open(filename, "w") as f:
|
|
f.write(rendered)
|
|
|
|
|
|
def limit_services(config, enabled_services):
|
|
if enabled_services == []:
|
|
return
|
|
|
|
for service in list(config.keys()):
|
|
if service in enabled_services:
|
|
config[service]["autostart"] = "true"
|
|
else:
|
|
config[service]["autostart"] = "false"
|
|
|
|
|
|
def override_services(config, override_services):
|
|
if override_services == []:
|
|
return
|
|
|
|
for service in list(config.keys()):
|
|
if service + "=true" in override_services:
|
|
config[service]["autostart"] = "true"
|
|
elif service + "=false" in override_services:
|
|
config[service]["autostart"] = "false"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1 and sys.argv[1] == "config":
|
|
config = config_services()
|
|
else:
|
|
config = registry_services()
|
|
limit_services(config, QUAY_SERVICES)
|
|
override_services(config, QUAY_OVERRIDE_SERVICES)
|
|
|
|
generate_supervisord_config(
|
|
os.path.join(QUAYCONF_DIR, "supervisord.conf"),
|
|
config,
|
|
QUAY_LOGGING,
|
|
QUAY_HOTRELOAD,
|
|
)
|