forked from y-scope/clp
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
15d0e7f
commit 9347eee
Showing
5 changed files
with
207 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
109 changes: 109 additions & 0 deletions
109
components/job-orchestration/job_orchestration/reducer/reducer.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
| #!/usr/bin/env python3 | ||
|
|
||
| import argparse | ||
| import logging | ||
| import os | ||
| import sys | ||
| import subprocess | ||
| from pathlib import Path | ||
| from typing import Any, Dict, List, Optional | ||
|
|
||
| from clp_py_utils.clp_config import SEARCH_JOBS_TABLE_NAME | ||
|
|
||
| from clp_py_utils.clp_config import CLPConfig, Database, ResultsCache | ||
| from clp_py_utils.clp_logging import get_logging_level | ||
| from clp_py_utils.core import read_yaml_config_file | ||
|
|
||
| # Setup logging | ||
| # Create logger | ||
| logger = logging.getLogger("reducer") | ||
| logger.setLevel(logging.INFO) | ||
| # Setup console logging | ||
| logging_console_handler = logging.StreamHandler() | ||
| logging_formatter = logging.Formatter("%(asctime)s [%(levelname)s] %(message)s") | ||
| logging_console_handler.setFormatter(logging_formatter) | ||
| logger.addHandler(logging_console_handler) | ||
| # Prevents double logging from sub loggers for specific jobs | ||
| logger.propagate = False | ||
|
|
||
| def main(argv: List[str]) -> int: | ||
| # fmt: off | ||
| args_parser = argparse.ArgumentParser(description="Spin up reducers.") | ||
| args_parser.add_argument('--config', '-c', required=True, help='CLP configuration file.') | ||
| args_parser.add_argument('--host', required=True, help='Host ip this container is running on') | ||
| args_parser.add_argument('--concurrency', required=True, help='Number of reducer servers to run') | ||
|
|
||
| parsed_args = args_parser.parse_args(argv[1:]) | ||
|
|
||
| # Setup logging to file | ||
| logs_dir = Path(os.getenv("CLP_LOGS_DIR")) | ||
| log_file = Path(os.getenv("CLP_LOGS_DIR")) / "reducer.log" | ||
| logging_file_handler = logging.FileHandler(filename=log_file, encoding="utf-8") | ||
| logging_file_handler.setFormatter(logging_formatter) | ||
| logger.addHandler(logging_file_handler) | ||
|
|
||
| # Update logging level based on config | ||
| logging_level_str = os.getenv("CLP_LOGGING_LEVEL") | ||
| logging_level = get_logging_level(logging_level_str) | ||
| logger.setLevel(logging_level) | ||
| logger.info(f"Start logging level = {logging.getLevelName(logging_level)}") | ||
|
|
||
| # Load configuration | ||
| config_path = Path(parsed_args.config) | ||
| try: | ||
| clp_config = CLPConfig.parse_obj(read_yaml_config_file(config_path)) | ||
| except ValidationError as err: | ||
| logger.error(err) | ||
| return -1 | ||
| except Exception as ex: | ||
| logger.error(ex) | ||
| # read_yaml_config_file already logs the parsing error inside | ||
| return -1 | ||
|
|
||
| clp_home = Path(os.getenv("CLP_HOME")) | ||
| reducer_cmd = [ | ||
| str(clp_home / "bin" / "reducer_server"), | ||
| "--db-host", clp_config.database.host, | ||
| "--db-port", str(clp_config.database.port), | ||
| "--db-user", str(clp_config.database.username), | ||
| "--db-pass", str(clp_config.database.password), | ||
| "--database", clp_config.database.name, | ||
| "--mongodb-database", clp_config.results_cache.db_name, | ||
| "--mongodb-uri", f"mongodb://{clp_config.results_cache.host}:{clp_config.results_cache.port}/", | ||
| "--mongodb-collection", clp_config.results_cache.results_collection_name, | ||
| "--host", parsed_args.host, | ||
| "--port", | ||
| ] | ||
|
|
||
| reducers = [] | ||
| concurrency = max(int(parsed_args.concurrency), 1) | ||
| for i in range(concurrency): | ||
| reducer_instance_cmd = reducer_cmd + [str(clp_config.reducer.base_port + i)] | ||
|
|
||
| log_file_path = logs_dir / ("reducer-" + str(i) + ".log") | ||
| log_file = open(log_file_path, 'a') | ||
|
|
||
| reducers.append( | ||
| subprocess.Popen( | ||
| reducer_instance_cmd, | ||
| close_fds=True, | ||
| stdout=log_file, | ||
| stderr=log_file, | ||
| ) | ||
| ) | ||
|
|
||
| logger.info("reducers started.") | ||
| logger.info(f"Host={parsed_args.host} Base port={clp_config.reducer.base_port} Concurrency={concurrency}") | ||
| for r in reducers: | ||
| r.wait() | ||
|
|
||
| logger.error("all reducers terminated") | ||
|
|
||
| logger.removeHandler(logging_file_handler) | ||
| logging_file_handler.close() | ||
|
|
||
| return 0 | ||
|
|
||
|
|
||
| if "__main__" == __name__: | ||
| sys.exit(main(sys.argv)) |