mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-08-31 08:39:34 +00:00
37fffa618f
- abstract_main only creates Config object using the passed configfile and instantiates the passed daemon class - common args parser is extracted into a separate func that is called from daemons' main funcs (specific args can be added afterwards) Co-authored-by: Alexander Kukushkin <[email protected]>
34 lines
882 B
Python
34 lines
882 B
Python
import logging
|
|
|
|
from .config import Config
|
|
from .daemon import AbstractPatroniDaemon, abstract_main, get_base_arg_parser
|
|
from .dcs.raft import KVStoreTTL
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RaftController(AbstractPatroniDaemon):
|
|
|
|
def __init__(self, config: Config) -> None:
|
|
super(RaftController, self).__init__(config)
|
|
|
|
kvstore_config = self.config.get('raft')
|
|
assert 'self_addr' in kvstore_config
|
|
self._raft = KVStoreTTL(None, None, None, **kvstore_config)
|
|
|
|
def _run_cycle(self) -> None:
|
|
try:
|
|
self._raft.doTick(self._raft.conf.autoTickPeriod)
|
|
except Exception:
|
|
logger.exception('doTick')
|
|
|
|
def _shutdown(self) -> None:
|
|
self._raft.destroy()
|
|
|
|
|
|
def main() -> None:
|
|
parser = get_base_arg_parser()
|
|
args = parser.parse_args()
|
|
|
|
abstract_main(RaftController, args.configfile)
|