sdk_base

1 功能

支持多 IDC 配置，简化 SDK 编写

2 例子

from xlib.util import http_util
from xlib.middleware import sdk_base

SERVICE_NAME = "xxx"

@sdk_base.requested(service_name=SERVICE_NAME,
                    method="GET",
                    uri="/test/{order_id}"
                    )
def order_show(req, order_id, **kwargs):
    kwargs["url"] = kwargs["url"].format(order_id=order_id)
    headers = {"x_username": "x1_cli"}
    r = http_util.request(headers=headers, **kwargs)
    return r

3 代码

butterfly/xlib/middleware/sdk_base.py

from xlib.util import wrapt
from xlib.util import funcsigs
from xlib.util import config_util


def requested(service_name, method, uri):
    """
    填充 method & url
    """
    @wrapt.decorator
    def wrapper(wrapped, instance, args, kwargs):
        """
        wrapper
        """
        # 提取原 func 中的 req, 不论使用位置参数还是关键字参数
        sig = funcsigs.signature(wrapped)
        kargs = dict(zip([k for k in sig.parameters if k not in kwargs], args))
        req = kargs.get("req", kwargs.get("req"))

        host = config_util.get_config(req, service_name, "host")
        port = config_util.get_config(req, service_name, "port")
        url = "http://{}:{}{}".format(host, port, uri)

        kwargs["method"] = method
        kwargs["url"] = url
        return wrapped(*args, **kwargs)

    return wrapper