8
8
import click_log # type: ignore
9
9
from aiohttp import web
10
10
from metricq import get_logger
11
+ from metricq .cli import metricq_command
11
12
12
13
from .routes import setup_routes
13
14
from .sink import Sink
@@ -78,14 +79,12 @@ def create_app(token: str, url: str, port: int) -> web.Application:
78
79
return app
79
80
80
81
81
- @click .command ()
82
- @click .argument ("url" , default = "amqp://localhost/" )
83
- @click .option ("--token" , default = "metricq-sink-websocket" )
82
+ @metricq_command (default_token = "metricq-sink-websocket" )
84
83
@click .option ("--host" , default = "0.0.0.0" )
85
84
@click .option ("--port" , type = int , default = 3000 )
86
85
@click .version_option (client_version )
87
86
@click_log .simple_verbosity_option (logger ) # type: ignore
88
- def runserver_cmd (url : str , token : str , host : str , port : int ) -> None :
87
+ def runserver_cmd (server : str , token : str , host : str , port : int ) -> None :
89
88
try :
90
89
import uvloop # type: ignore
91
90
@@ -94,5 +93,5 @@ def runserver_cmd(url: str, token: str, host: str, port: int) -> None:
94
93
except ImportError :
95
94
logger .debug ("using default event loop" )
96
95
97
- app = create_app (token , url , port )
96
+ app = create_app (token , server , port )
98
97
web .run_app (app , host = host , port = port )
0 commit comments