Files
fn-serverless/vendor/git.apache.org/thrift.git/test/features/container_limit.py
Reed Allman 9eaf824398 add jaeger support, link hot container & req span (#840)
* 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
2018-03-13 15:57:12 -07:00

73 lines
1.9 KiB
Python

#!/usr/bin/env python
import argparse
import sys
from util import add_common_args, init_protocol
from local_thrift import thrift # noqa
from thrift.Thrift import TMessageType, TType
# TODO: generate from ThriftTest.thrift
def test_list(proto, value):
method_name = 'testList'
ttype = TType.LIST
etype = TType.I32
proto.writeMessageBegin(method_name, TMessageType.CALL, 3)
proto.writeStructBegin(method_name + '_args')
proto.writeFieldBegin('thing', ttype, 1)
proto.writeListBegin(etype, len(value))
for e in value:
proto.writeI32(e)
proto.writeListEnd()
proto.writeFieldEnd()
proto.writeFieldStop()
proto.writeStructEnd()
proto.writeMessageEnd()
proto.trans.flush()
_, mtype, _ = proto.readMessageBegin()
assert mtype == TMessageType.REPLY
proto.readStructBegin()
_, ftype, fid = proto.readFieldBegin()
assert fid == 0
assert ftype == ttype
etype2, len2 = proto.readListBegin()
assert etype == etype2
assert len2 == len(value)
for i in range(len2):
v = proto.readI32()
assert v == value[i]
proto.readListEnd()
proto.readFieldEnd()
_, ftype, _ = proto.readFieldBegin()
assert ftype == TType.STOP
proto.readStructEnd()
proto.readMessageEnd()
def main(argv):
p = argparse.ArgumentParser()
add_common_args(p)
p.add_argument('--limit', type=int)
args = p.parse_args()
proto = init_protocol(args)
# TODO: test set and map
test_list(proto, list(range(args.limit - 1)))
test_list(proto, list(range(args.limit - 1)))
print('[OK]: limit - 1')
test_list(proto, list(range(args.limit)))
test_list(proto, list(range(args.limit)))
print('[OK]: just limit')
try:
test_list(proto, list(range(args.limit + 1)))
except Exception:
print('[OK]: limit + 1')
else:
print('[ERROR]: limit + 1')
assert False
if __name__ == '__main__':
sys.exit(main(sys.argv[1:]))