# 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 logging
import os
import signal
import sys
import threading
import time
from typing import Never
from oslo_config import cfg
import cotyledon
from cotyledon import _utils
from cotyledon import oslo_config_glue
logging.basicConfig(level=logging.DEBUG, stream=sys.stdout)
LOG = logging.getLogger("cotyledon.tests.examples")
# We don't want functional tests to wait for this:
cotyledon.ServiceManager._slowdown_respawn_if_needed = lambda *args: True
class FullService(cotyledon.Service):
name = "heavy"
def __init__(self, worker_id) -> None:
super().__init__(worker_id)
self._shutdown = threading.Event()
LOG.error("%s init", self.name)
def run(self) -> None:
LOG.error("%s run", self.name)
self._shutdown.wait()
def terminate(self) -> None:
LOG.error("%s terminate", self.name)
self._shutdown.set()
sys.exit(42)
def reload(self) -> None:
LOG.error("%s reload", self.name)
class LigthService(cotyledon.Service):
name = "light"
class BuggyService(cotyledon.Service):
name = "buggy"
graceful_shutdown_timeout = 1
def terminate(self) -> None: # noqa: PLR6301
time.sleep(60)
LOG.error("time.sleep done")
class BoomError(Exception):
pass
class BadlyCodedService(cotyledon.Service):
def run(self) -> Never: # noqa: PLR6301
msg = "so badly coded service"
raise BoomError(msg)
class OsloService(cotyledon.Service):
name = "oslo"
class WindowService(cotyledon.Service):
name = "window"
def on_terminate() -> None:
LOG.error("master terminate hook")
def on_terminate2() -> None:
LOG.error("master terminate2 hook")
def on_reload() -> None:
LOG.error("master reload hook")
def example_app() -> None:
p = cotyledon.ServiceManager()
p.add(FullService, 2)
service_id = p.add(LigthService, 5)
p.reconfigure(service_id, 1)
p.register_hooks(on_terminate, on_reload)
p.register_hooks(on_terminate2)
p.run()
def buggy_app() -> None:
p = cotyledon.ServiceManager()
p.add(BuggyService)
p.run()
def oslo_app() -> None:
conf = cfg.ConfigOpts()
conf([], project="openstack-app", validate_default_values=True, version="0.1")
p = cotyledon.ServiceManager()
oslo_config_glue.setup(p, conf)
p.add(OsloService)
p.run()
def window_sanity_check() -> None:
p = cotyledon.ServiceManager()
p.add(LigthService)
t = _utils.spawn(p.run)
time.sleep(10)
os.kill(os.getpid(), signal.SIGTERM)
t.join()
def badly_coded_app() -> None:
p = cotyledon.ServiceManager()
p.add(BadlyCodedService)
p.run()
def exit_on_special_child_app() -> None:
p = cotyledon.ServiceManager()
sid = p.add(LigthService, 1)
p.add(FullService, 2)
def on_dead_worker(service_id, worker_id, exit_code) -> None:
# Shutdown everybody if LigthService died
if service_id == sid:
p.shutdown()
p.register_hooks(on_dead_worker=on_dead_worker)
p.run()
def sigterm_during_init() -> None:
def kill() -> None:
os.kill(os.getpid(), signal.SIGTERM)
# Kill in 0.01 sec
threading.Timer(0.01, kill).start()
p = cotyledon.ServiceManager()
p.add(LigthService, 10)
p.run()
if __name__ == "__main__":
globals()[sys.argv[1]]()