gcpa0124-002.py

# create a thread-local global context
_trace_context = threading.local()
 
##===============================================================
 
class TraceContext(object):
	
	def __enter__(self):
		self.start()
		return self
 
	def start(self):
		_trace_context.spans = []
 
	def __exit__(self, exc_type, exc_value, traceback):
		self.finish()
 
	def finish(self):
		self.write()
		_trace_context.spans = []
 
	def span(self,name=""):
		spn = Span(name)
		_trace_context.spans.append(spn)
		return spn