#79 Migrate to metricq_command

metricq_sink_nsca/check.py

@@ -21,7 +21,7 @@
 from asyncio import CancelledError, gather, sleep
 from typing import Dict, Iterable, NamedTuple, Optional, Set
 
-from metricq.types import Timedelta, Timestamp
+from metricq.timeseries import Timedelta, Timestamp
 
 from .logging import get_logger
 from .plugin import Plugin

metricq_sink_nsca/main.py

@@ -48,23 +48,21 @@ def _set_log_levels(_ctx, _param, value: str):
 CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"])
 
 
-@click.command(context_settings=CONTEXT_SETTINGS)
-@click.option("--metricq-server", "-s", default="amqp://localhost/")
-@click.option("--token", "-t", default="sink-nsca")
+@metricq_command(default_token="sink-nsca", client_version=None)
 @click.option("--dry-run", "-n", is_flag=True)
 @verbosity_option(root_logger)
-def main(metricq_server, token, dry_run):
-    try:
-        import uvloop
-
-        uvloop.install()
-        logging.info("Using uvloop-based event loop")
-    except ImportError:
-        logging.debug("Using default event loop")
-
-    reporter = ReporterSink(
-        dry_run=dry_run,
-        management_url=metricq_server,
-        token=token,
-    )
-    reporter.run(cancel_on_exception=True)
+def main(server, token, dry_run):
+	try:
+		import uvloop
+
+		uvloop.install()
+		logging.info("Using uvloop-based event loop")
+	except ImportError:
+		logging.debug("Using default event loop")
+
+	reporter = ReporterSink(
+		dry_run=dry_run,
+		management_url=server,
+		token=token,
+	)
+	reporter.run(cancel_on_exception=True)

metricq_sink_nsca/report_queue.py

@@ -22,7 +22,7 @@
 from dataclasses import dataclass
 from typing import AsyncIterator, Optional
 
-from metricq.types import Timedelta
+from metricq.timeseries import Timedelta
 
 from .state import State
 

metricq_sink_nsca/state_cache.py

@@ -25,7 +25,7 @@
 from dataclasses import replace as dataclass_replace
 from typing import Dict, Iterable, Iterator, List, Optional, Set, Tuple, Union
 
-from metricq.types import Timedelta, Timestamp
+from metricq.timeseries import Timedelta, Timestamp
 
 from .logging import get_logger
 from .state import State

metricq_sink_nsca/timeout_check.py

@@ -22,7 +22,7 @@
 from asyncio import CancelledError, Event
 from typing import Optional, Protocol
 
-from metricq.types import Timedelta, Timestamp
+from metricq.timeseries import Timedelta, Timestamp
 
 from .logging import get_logger
 from .subtask import subtask

setup.cfg

@@ -35,7 +35,7 @@ include_package_data = True
 setup_requires =
     setuptools_scm
 install_requires =
-    metricq ~= 4.0
+    metricq ~= 5.4
     click ~= 8.0
     click-log ~= 0.3