Ejemplo n.º 1
0
Archivo: server.py Proyecto: ziq211/hue
    def call_downstream_http(self, span, downstream):
        url = 'http://%s:%s/join_trace' % (downstream.host, downstream.port)
        body = serializer.join_trace_request_to_json(downstream.downstream, downstream.serverRole)

        req = tornado.httpclient.HTTPRequest(url=url, method='POST',
                                             headers={'Content-Type': 'application/json'},
                                             body=body)
        http_client.before_http_request(tornado_http.TornadoRequestWrapper(request=req),
                                        lambda: span)
        client = tornado.httpclient.AsyncHTTPClient()
        http_result = yield client.fetch(req)

        raise tornado.gen.Return(serializer.traceresponse_from_json(http_result.body))
 def do_open(self, req, conn):
     request_wrapper = Urllib2RequestWrapper(request=req)
     span = before_http_request(
         request=request_wrapper,
         current_span_extractor=get_current_span)
     with span:
         resp = urllib2.AbstractHTTPHandler.do_open(self, conn, req)
         if resp.code is not None:
             span.set_tag('status_code', resp.code)
     return resp
 def do_open(self, req, conn):
     request_wrapper = Urllib2RequestWrapper(request=req)
     span = before_http_request(
         request=request_wrapper,
         current_span_extractor=get_current_span)
     with span:
         resp = urllib2.AbstractHTTPHandler.do_open(self, conn, req)
         if resp.code is not None:
             span.set_tag('http.status_code', resp.code)
     return resp
Ejemplo n.º 4
0
    def call_downstream_http(self, span, trace_request, trace_response, response_writer):

        def handle_response(response):
            tr = serializer.traceresponse_from_json(response.body)
            if tr.notImplementedError:
                response_writer.write(serializer.obj_to_json(tr))
            else:
                trace_response.downstream = tr
                response_writer.write(serializer.obj_to_json(trace_response))
            response_writer.finish()

        downstream = trace_request.downstream
        url = "http://%s:%s/join_trace" % (downstream.host, downstream.port)
        body = serializer.downstream_to_json(downstream.downstream)

        req = tornado.httpclient.HTTPRequest(url=url, method="POST",
                                             headers={"Content-Type": "application/json"},
                                             body=body)
        http_client.before_http_request(tornado_http.TornadoRequestWrapper(request=req),
                                        lambda: span)
        client = tornado.httpclient.AsyncHTTPClient()
        client.fetch(req, handle_response)
 def do_open(self, req, conn):
     request_wrapper = Urllib2RequestWrapper(request=req)
     span = before_http_request(
         request=request_wrapper,
         current_span_extractor=get_current_span)
     with span:
         if base_cls:
             # urllib2.AbstractHTTPHandler doesn't support super()
             resp = base_cls.do_open(self, conn, req)
         else:
             resp = super(DerivedHandler, self).do_open(conn, req)
         if resp.code is not None:
             span.set_tag('http.status_code', resp.code)
     return resp
    def new_fetch_impl(self, request, callback):
        request_wrapper = TornadoRequestWrapper(request=request)
        span = before_http_request(request=request_wrapper,
                                   current_span_extractor=get_current_span)

        def new_callback(response):
            if hasattr(response, 'code') and response.code:
                span.set_tag('http.status_code', '%s' % response.code)
            if hasattr(response, 'error') and response.error:
                span.log(event='error', payload='%s' % response.error)
            span.finish()
            return callback(response)

        real_fetch_impl(self, request, new_callback)
Ejemplo n.º 7
0
    def new_fetch_impl(self, request, callback):
        request_wrapper = TornadoRequestWrapper(request=request)
        span = before_http_request(request=request_wrapper,
                                   current_span_extractor=get_current_span)

        def new_callback(response):
            if hasattr(response, 'code') and response.code:
                span.set_tag('http.status_code', '%s' % response.code)
            if hasattr(response, 'error') and response.error:
                span.log(event='error', payload='%s' % response.error)
            span.finish()
            return callback(response)

        real_fetch_impl(self, request, new_callback)