__init__.py 1.7 KB
Newer Older
P
peng.xu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
import logging
from mishards import settings
logger = logging.getLogger()

from mishards.db_base import DB
db = DB()

from mishards.server import Server
grpc_server = Server()


def create_app(testing_config=None):
    config = testing_config if testing_config else settings.DefaultConfig
赖龙 已提交
14 15 16
    db.init_db(uri=config.SQLALCHEMY_DATABASE_URI, echo=config.SQL_ECHO, pool_size=config.SQL_POOL_SIZE,
               pool_recycle=config.SQL_POOL_RECYCLE, pool_timeout=config.SQL_POOL_TIMEOUT,
               pool_pre_ping=config.SQL_POOL_PRE_PING, max_overflow=config.SQL_MAX_OVERFLOW)
P
peng.xu 已提交
17 18 19 20

    from mishards.connections import ConnectionMgr
    connect_mgr = ConnectionMgr()

P
peng.xu 已提交
21 22 23
    from discovery.factory import DiscoveryFactory
    discover = DiscoveryFactory(config.DISCOVERY_PLUGIN_PATH).create(config.DISCOVERY_CLASS_NAME,
                                                                     conn_mgr=connect_mgr)
P
peng.xu 已提交
24 25

    from mishards.grpc_utils import GrpcSpanDecorator
P
peng.xu 已提交
26
    from tracer.factory import TracerFactory
P
peng.xu 已提交
27
    tracer = TracerFactory(config.TRACER_PLUGIN_PATH).create(config.TRACER_CLASS_NAME,
P
peng.xu 已提交
28 29
                                                             plugin_config=settings.TracingConfig,
                                                             span_decorator=GrpcSpanDecorator())
P
peng.xu 已提交
30

P
peng.xu 已提交
31 32 33
    from mishards.router.factory import RouterFactory
    router = RouterFactory(config.ROUTER_PLUGIN_PATH).create(config.ROUTER_CLASS_NAME,
                                                             conn_mgr=connect_mgr)
P
peng.xu 已提交
34

P
peng.xu 已提交
35 36 37 38
    grpc_server.init_app(conn_mgr=connect_mgr,
                         tracer=tracer,
                         router=router,
                         discover=discover)
P
peng.xu 已提交
39 40 41 42

    from mishards import exception_handlers

    return grpc_server