mirror of
https://github.com/valitydev/redash.git
synced 2024-11-06 09:05:17 +00:00
67 lines
2.0 KiB
Python
Executable File
67 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
CLI to manage redash.
|
|
"""
|
|
import atfork
|
|
atfork.monkeypatch_os_fork_functions()
|
|
import atfork.stdlib_fixer
|
|
atfork.stdlib_fixer.fix_logging_module()
|
|
|
|
import argparse
|
|
import logging
|
|
import time
|
|
from redash import settings, app, data_manager
|
|
|
|
|
|
def start_workers():
|
|
try:
|
|
old_workers = data_manager.redis_connection.smembers('workers')
|
|
data_manager.redis_connection.delete('workers')
|
|
|
|
logging.info("Cleaning old workers: %s", old_workers)
|
|
|
|
data_manager.start_workers(settings.WORKERS_COUNT, settings.CONNECTION_STRING)
|
|
logging.info("Workers started.")
|
|
|
|
while True:
|
|
try:
|
|
data_manager.refresh_queries()
|
|
except Exception as e:
|
|
logging.error("Something went wrong with refreshing queries...")
|
|
logging.exception(e)
|
|
time.sleep(60)
|
|
except KeyboardInterrupt:
|
|
logging.warning("Exiting; waiting for threads")
|
|
data_manager.stop_workers()
|
|
|
|
|
|
def start_server(port, debug):
|
|
app.run(debug=debug, port=port)
|
|
|
|
if __name__ == '__main__':
|
|
channel = logging.StreamHandler()
|
|
logging.getLogger().addHandler(channel)
|
|
logging.getLogger().setLevel(settings.LOG_LEVEL)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers(title='command', dest='command')
|
|
|
|
subparsers.add_parser('worker', help='start query execution workers')
|
|
server_parser = subparsers.add_parser('server', help='start api server')
|
|
server_parser.add_argument('--debug',
|
|
action='store_true',
|
|
help='start in debug mode (code reload)')
|
|
server_parser.add_argument('--port',
|
|
default=8888,
|
|
help='port to bind to')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.command == "worker":
|
|
start_workers()
|
|
elif args.command == 'server':
|
|
start_server(args.port, args.debug)
|
|
else:
|
|
print "Unknown command"
|
|
|