在ApplicationSession的注册端点内访问RPC调用方的IP和HTTP连接头

Question

我使用的是Autobahn0.9.2和python3.4和asyncio。在

问题：使用WAMP，是否可以从RPC端点内部访问作为调用方IP和HTTP连接头的对等方？当建立连接时，这些信息是否保持不变？如果没有，我如何开始扩展一些工厂来支持这一点？在

我的目标很简单：我希望有一个RPC端点来对连接的对等方（调用者）的IP进行地理定位，并将扩充的数据中继到Redis。我已经读过资料来源，知道信息是从哪里传来的(高速公路.websocket.protocol.WebSocketServerProtocol->onConnect（request）），但无法从onJoin回调中定义的ApplicationSession的RPC终结点深入到它。我试图遍历传输/路由器/路由器会话链，但没有成功到达那里。我对对等方的IP和来自初始连接请求的HTTP头都感兴趣。在

这是蒸馏成分：

class IncomingComponent(ApplicationSession):

def __init__(self, **params):
    super().__init__()
    self.redis = StrictRedis(host=config["redis"]["host"], port=config["redis"]["port"], db=config["redis"]["databases"]["ailytics"])

def onConnect(self):
    self.join("abc")

@asyncio.coroutine
def onJoin(self, details):

    def geolocalize_and_store_event(event, detail):
        # Geolocalize here! Have access to caller ID through detail
        self.redis.rpush("abc:events", json.dumps(event))

    yield from self.register(
        geolocalize_and_store_event,
        "abc.geolocalize_and_store_event",
        options=RegisterOptions(details_arg='detail', discloseCaller = True)
    )

以及服务器的初始化：

^{pr2}$

Answer 1

您可以通过中的wamp.session.get元API访问额外的会话/传输信息横杆io至少：

@inlineCallbacks
def onJoin(self, ign):

    @inlineCallbacks
    def method(details):
        session = yield self.call('wamp.session.get', details.caller)
        peer = session['transport']['peer']
        print "peer's address", peer

        headers = session['transport']['http_headers_received']
        print "headers:"
        print '\n'.join(['{}: {}'.format(k, v) for (k, v) in headers.items()])

    yield self.register(
        method, 'some_method',
        types.RegisterOptions(details_arg='details'),
    )

Answer 2

功能还没有如@oberstet所说，但是由于使用了工厂模式的高速公路/WAMP工具，我能够在不更改库代码的情况下提出解决方案。在

子类有3个组件：

首先，我们向wamp.RouterSession公司的子类

class IncomingServerSession(wamp.RouterSession):

    def __init__(self, routerFactory):
        super().__init__(routerFactory)
        self.ipAddress = None

然后我们做wamp.RouterSessionFactory公司子类使用IncomingServerSession

^{pr2}$

最后，我们将子类websocket.WampWebSocketServer协议并设置ipAddress实例变量。因为我们在onOpen回调中，所以我们可以访问peer和HTTP头。我的服务器是反向代理的，所以我要在对等服务器上寻找一个自定义的HTTP报头。在

class IncomingServerProtocol(websocket.WampWebSocketServerProtocol):
    def onOpen(self):
        try:
            self._session = self.factory._factory()

            # Use your own header or just the peer if not reverse-proxied
            self._session.ipAddress = (self.http_headers.get('x-real-ip') or self.peer) 

            self._session.onOpen(self)

        except Exception as e:
            if self.factory.debug_wamp:
                traceback.print_exc()
                # # Exceptions raised in onOpen are fatal ..
                reason = "WAMP Internal Error ({})".format(e)
                self._bailout(protocol.WebSocketProtocol.CLOSE_STATUS_CODE_INTERNAL_ERROR, reason=reason)

下面是我们如何在RPC调用中访问对等方的IP:

@asyncio.coroutine
def onJoin(self, details):
    def event(e, details):
        caller_session_id = details.caller
        caller_session = self._transport._router._dealer._session_id_to_session[caller_session_id]

        print(caller_session.ipAddress)

    #discloseCaller needs to be True
    yield from self.register(event, "abc.event", options=RegisterOptions(details_arg='details', discloseCaller=True))

最后，我们需要更新初始化代码以使用我们的子类：

router_factory = wamp.RouterFactory()

session_factory = IncomingServerSessionFactory(router_factory)
session_factory.add(IncomingComponent())

transport_factory = websocket.WampWebSocketServerFactory(session_factory, debug=False, debug_wamp=False)
transport_factory.protocol = IncomingServerProtocol

loop = asyncio.get_event_loop()
coro = loop.create_server(transport_factory, '0.0.0.0', 7788)
server = loop.run_until_complete(coro)

try:
    loop.run_forever()
except KeyboardInterrupt:
    pass
finally:
    server.close()
    loop.close()

在官方支持之前你就是这样做的！