mirror of
https://github.com/mozilla-services/syncstorage-rs.git
synced 2026-01-21 00:12:14 +01:00
8995db6268
Some checks are pending
Glean probe-scraper / glean-probe-scraper (push) Waiting to run
feat: ruff for python lint and format
96 lines
2.4 KiB
Python
96 lines
2.4 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
# You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
"""
|
|
|
|
Script to add a new node to the system.
|
|
|
|
"""
|
|
|
|
import logging
|
|
import optparse
|
|
|
|
from database import Database, SERVICE_NAME
|
|
import util
|
|
|
|
|
|
logger = logging.getLogger("tokenserver.scripts.add_node")
|
|
|
|
|
|
def add_node(node, capacity, **kwds):
|
|
"""Add the specific node to the system."""
|
|
logger.info("Adding node %s to service %s", node, SERVICE_NAME)
|
|
try:
|
|
database = Database()
|
|
database.add_node(node, capacity, **kwds)
|
|
except Exception:
|
|
logger.exception("Error while adding node")
|
|
return False
|
|
else:
|
|
logger.info("Finished adding node %s", node)
|
|
return True
|
|
|
|
|
|
def main(args=None):
|
|
"""Main entry-point for running this script.
|
|
|
|
This function parses command-line arguments and passes them on
|
|
to the add_node() function.
|
|
"""
|
|
usage = "usage: %prog [options] node_name capacity"
|
|
descr = "Add a new node to the tokenserver database"
|
|
parser = optparse.OptionParser(usage=usage, description=descr)
|
|
parser.add_option(
|
|
"", "--available", type="int", help="How many user slots the node has available"
|
|
)
|
|
parser.add_option(
|
|
"",
|
|
"--current-load",
|
|
type="int",
|
|
help="How many user slots the node has occupied",
|
|
)
|
|
parser.add_option(
|
|
"", "--downed", action="store_true", help="Mark the node as down in the db"
|
|
)
|
|
parser.add_option(
|
|
"",
|
|
"--backoff",
|
|
action="store_true",
|
|
help="Mark the node as backed-off in the db",
|
|
)
|
|
parser.add_option(
|
|
"-v",
|
|
"--verbose",
|
|
action="count",
|
|
dest="verbosity",
|
|
help="Control verbosity of log messages",
|
|
)
|
|
|
|
opts, args = parser.parse_args(args)
|
|
if len(args) != 2:
|
|
parser.print_usage()
|
|
return 1
|
|
|
|
util.configure_script_logging(opts)
|
|
|
|
node_name = args[0]
|
|
capacity = int(args[1])
|
|
|
|
kwds = {}
|
|
if opts.available is not None:
|
|
kwds["available"] = opts.available
|
|
if opts.current_load is not None:
|
|
kwds["current_load"] = opts.current_load
|
|
if opts.backoff is not None:
|
|
kwds["backoff"] = opts.backoff
|
|
if opts.downed is not None:
|
|
kwds["downed"] = opts.downed
|
|
|
|
add_node(node_name, capacity, **kwds)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
util.run_script(main)
|