Mirror of https://github.com/roostorg/osprey
github.com/roostorg/osprey
1from typing import Any, Sequence, Type
2
3from osprey.engine.udf.base import UDFBase
4from osprey.worker.adaptor.plugin_manager import hookimpl_osprey
5from osprey.worker.lib.config import Config
6from osprey.worker.sinks.sink.output_sink import BaseOutputSink, StdoutOutputSink
7from udfs.ban_user import BanUser
8from udfs.text_contains import TextContains
9
10
11@hookimpl_osprey
12def register_udfs() -> Sequence[Type[UDFBase[Any, Any]]]:
13 return [TextContains, BanUser]
14
15
16@hookimpl_osprey
17def register_output_sinks(config: Config) -> Sequence[BaseOutputSink]:
18 return [StdoutOutputSink(log_sampler=None)]