feat: add new type all

This commit is contained in:
BryantHe 2023-07-20 13:12:17 +08:00
parent 08e818e1e1
commit 8cf674f0df
8 changed files with 185 additions and 8 deletions

View File

@ -40,10 +40,10 @@ def cli():
@click.option('-d', '--directory', @click.option('-d', '--directory',
required=True, required=True,
help='The directory name of nameko services') help='The directory name of nameko services')
@click.option('-f', '--type', '_type', @click.option('-t', '--type', '_type',
default='rpc', default='all',
show_default=True, show_default=True,
type=click.Choice(['rpc', 'event', 'http', 'timer'], case_sensitive=False), type=click.Choice(['all', 'rpc', 'event', 'http', 'timer'], case_sensitive=False),
help='The template type of nameko service') help='The template type of nameko service')
def init(directory, _type): def init(directory, _type):
""" """

View File

View File

@ -0,0 +1,144 @@
import json
from nameko.events import EventDispatcher, event_handler
from nameko.rpc import rpc, ServiceRpc
from nameko.timer import timer
from nameko.web.handlers import http
from werkzeug.wrappers import Response
from nameko_tracer import Tracer
from namekoplus import init_statsd, init_sentry
class HttpDemoService:
name = "http_demo_service"
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@http("GET", "/broken")
@statsd.timer('broken')
def broken(self, request):
raise ConnectionRefusedError()
@http('GET', '/books/<string:uuid>')
@statsd.timer('demo_get')
def demo_get(self, request, uuid):
data = {'id': uuid, 'title': 'The unbearable lightness of being',
'author': 'Milan Kundera'}
return Response(json.dumps({'book': data}),
mimetype='application/json')
@http('POST', '/books')
@statsd.timer('demo_post')
def demo_post(self, request):
return Response(json.dumps({'book': request.data.decode()}),
mimetype='application/json')
class RpcResponderDemoService:
name = "rpc_responder_demo_service"
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@rpc
@statsd.timer('hello')
def hello(self, name):
return "Hello, {}!".format(name)
class RpcCallerDemoService:
name = "rpc_caller_demo_service"
remote = ServiceRpc("rpc_responder_demo_service")
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@rpc
@statsd.timer('remote_hello')
def remote_hello(self, value="John Doe"):
res = u"{}".format(value)
return self.remote.hello(res)
class EventPublisherService:
name = "publisher_service"
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
dispatch = EventDispatcher()
@rpc
@statsd.timer('publish')
def publish(self, event_type, payload):
self.dispatch(event_type, payload)
class AnEventListenerService:
name = "an_event_listener_service"
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@event_handler("publisher_service", "an_event")
@statsd.timer('consume_an_event')
def consume_an_event(self, payload):
print("service {} received:".format(self.name), payload)
class AnotherEventListenerService:
name = "another_event_listener_service"
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@event_handler("publisher_service", "another_event")
@statsd.timer('consume_another_event')
def consume_another_event(self, payload):
print("service {} received:".format(self.name), payload)
class ListenBothEventsService:
name = "listen_both_events_service"
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@event_handler("publisher_service", "an_event")
@statsd.timer('consume_an_event')
def consume_an_event(self, payload):
print("service {} received:".format(self.name), payload)
@event_handler("publisher_service", "another_event")
@statsd.timer('consume_another_event')
def consume_another_event(self, payload):
print("service {} received:".format(self.name), payload)
class Timer:
name = 'timer'
tracer = Tracer()
sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@timer(interval=1)
@statsd.timer('ping')
def ping(self):
# method executed every second
print("pong")

View File

@ -0,0 +1,24 @@
AMQP_URI: pyamqp://${RABBIT_USER:guest}:${RABBIT_PASSWORD:guest}@${RABBIT_HOST:localhost}:${RABBIT_PORT:5672}/
RPC_EXCHANGE: 'nameko-rpc'
max_workers: 10
parent_calls_tracked: 20
LOGGING:
version: 1
formatters:
tracer:
(): nameko_tracer.formatters.PrettyJSONFormatter
handlers:
tracer:
class: logging.StreamHandler
formatter: tracer
loggers:
nameko_tracer:
level: INFO
handlers: [tracer]
SENTRY:
DSN: ${SENTRY_DSN}
CLIENT_CONFIG:
site: ${SENTRY_SITE}

View File

@ -5,14 +5,15 @@ from namekoplus import init_statsd, init_sentry
class EventPublisherService: class EventPublisherService:
name = "publisher_service"
dispatch = EventDispatcher() name = "publisher_service"
tracer = Tracer() tracer = Tracer()
sentry = init_sentry() sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port') statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
dispatch = EventDispatcher()
@rpc @rpc
@statsd.timer('publish') @statsd.timer('publish')
def publish(self, event_type, payload): def publish(self, event_type, payload):
@ -20,8 +21,10 @@ class EventPublisherService:
class AnEventListenerService: class AnEventListenerService:
name = "an_event_listener_service" name = "an_event_listener_service"
tracer = Tracer()
sentry = init_sentry() sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port') statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@ -32,8 +35,10 @@ class AnEventListenerService:
class AnotherEventListenerService: class AnotherEventListenerService:
name = "another_event_listener_service" name = "another_event_listener_service"
tracer = Tracer()
sentry = init_sentry() sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port') statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')
@ -44,8 +49,10 @@ class AnotherEventListenerService:
class ListenBothEventsService: class ListenBothEventsService:
name = "listen_both_events_service" name = "listen_both_events_service"
tracer = Tracer()
sentry = init_sentry() sentry = init_sentry()
statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port') statsd = init_statsd('statsd_prefix', 'statsd_host', 'statsd_port')

View File

@ -6,6 +6,7 @@ from namekoplus import init_statsd, init_sentry
class HttpDemoService: class HttpDemoService:
name = "http_demo_service" name = "http_demo_service"
tracer = Tracer() tracer = Tracer()

View File

@ -3,8 +3,9 @@ from nameko_tracer import Tracer
from namekoplus import init_statsd, init_sentry from namekoplus import init_statsd, init_sentry
class Service: class Timer:
name = "service"
name = 'timer'
tracer = Tracer() tracer = Tracer()
sentry = init_sentry() sentry = init_sentry()

View File

@ -9,7 +9,7 @@ with open(path.join(here, 'README.md'), encoding='utf-8') as f:
setup( setup(
name='namekoplus', name='namekoplus',
version='0.1.1', version='0.1.2',
description='A lightweight Python distributed microservice solution', description='A lightweight Python distributed microservice solution',
long_description=long_description, long_description=long_description,
long_description_content_type='text/markdown', long_description_content_type='text/markdown',