mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
9eaf824398
* add jaeger support, link hot container & req span * adds jaeger support now with FN_JAEGER_URL, there's a simple tutorial in the operating/metrics.md file now and it's pretty easy to get up and running. * links a hot request span to a hot container span. when we change this to sample at a lower ratio we'll need to finagle the hot container span to always sample or something, otherwise we'll hide that info. at least, since we're sampling at 100% for now if this is flipped on, can see freeze/unfreeze etc. if they hit. this is useful for debugging. note that zipkin's exporter does not follow the link at all, hence jaeger... and they're backed by the Cloud Empire now (CNCF) so we'll probably use it anyway. * vendor: add thrift for jaeger
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
import argparse
|
|
import socket
|
|
import sys
|
|
|
|
from util import add_common_args
|
|
from local_thrift import thrift # noqa
|
|
from thrift.Thrift import TMessageType, TType
|
|
from thrift.transport.TSocket import TSocket
|
|
from thrift.transport.TTransport import TBufferedTransport, TFramedTransport
|
|
from thrift.protocol.TBinaryProtocol import TBinaryProtocol
|
|
from thrift.protocol.TCompactProtocol import TCompactProtocol
|
|
|
|
|
|
def test_void(proto):
|
|
proto.writeMessageBegin('testVoid', TMessageType.CALL, 3)
|
|
proto.writeStructBegin('testVoid_args')
|
|
proto.writeFieldStop()
|
|
proto.writeStructEnd()
|
|
proto.writeMessageEnd()
|
|
proto.trans.flush()
|
|
|
|
_, mtype, _ = proto.readMessageBegin()
|
|
assert mtype == TMessageType.REPLY
|
|
proto.readStructBegin()
|
|
_, ftype, _ = proto.readFieldBegin()
|
|
assert ftype == TType.STOP
|
|
proto.readStructEnd()
|
|
proto.readMessageEnd()
|
|
|
|
|
|
# THeader stack should accept binary protocol with optionally framed transport
|
|
def main(argv):
|
|
p = argparse.ArgumentParser()
|
|
add_common_args(p)
|
|
# Since THeaderTransport acts as framed transport when detected frame, we
|
|
# cannot use --transport=framed as it would result in 2 layered frames.
|
|
p.add_argument('--override-transport')
|
|
p.add_argument('--override-protocol')
|
|
args = p.parse_args()
|
|
assert args.protocol == 'header'
|
|
assert args.transport == 'buffered'
|
|
assert not args.ssl
|
|
|
|
sock = TSocket(args.host, args.port, socket_family=socket.AF_INET)
|
|
if not args.override_transport or args.override_transport == 'buffered':
|
|
trans = TBufferedTransport(sock)
|
|
elif args.override_transport == 'framed':
|
|
print('TFRAMED')
|
|
trans = TFramedTransport(sock)
|
|
else:
|
|
raise ValueError('invalid transport')
|
|
trans.open()
|
|
|
|
if not args.override_protocol or args.override_protocol == 'binary':
|
|
proto = TBinaryProtocol(trans)
|
|
elif args.override_protocol == 'compact':
|
|
proto = TCompactProtocol(trans)
|
|
else:
|
|
raise ValueError('invalid transport')
|
|
|
|
test_void(proto)
|
|
test_void(proto)
|
|
|
|
trans.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main(sys.argv[1:]))
|