def before_cursor_execute(conn, cursor, statement, parameters, context,
executemany):
pinpointPy.start_trace()
pinpointPy.add_clue(PP_INTERCEPTOR_NAME, 'before_cursor_execute')
pinpointPy.add_clue(PP_SQL_FORMAT, statement)
pinpointPy.add_clues(PP_ARGS, 'user not cared')
DBUrl = urlparse(str(conn.engine.url))
if 'mysql' in DBUrl.scheme:
pinpointPy.add_clue(PP_SERVER_TYPE, PP_MYSQL)
if 'postgresql' in DBUrl.scheme:
pinpointPy.add_clue(PP_SERVER_TYPE, PP_POSTGRESQL)
pinpointPy.add_clue(PP_DESTINATION, DBUrl.hostname)
def test_trace_life(self):
self.assertFalse(pinpointPy.trace_has_root())
pinpointPy.start_trace()
self.assertTrue(pinpointPy.trace_has_root())
pinpointPy.start_trace()
self.assertTrue(pinpointPy.trace_has_root())
pinpointPy.start_trace()
pinpointPy.start_trace()
# self.assertRaises(Exception,lambda: pinpointPy.trace_has_root(1025))
# self.assertRaises(Exception,lambda: pinpointPy.trace_has_root(-1025))
pinpointPy.add_clue("key", "value")
pinpointPy.add_clue("key", "value3")
pinpointPy.set_context_key('sid', '12345')
value = pinpointPy.get_context_key('sid')
self.assertEqual(value, '12345')
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.force_flush_trace(10)
self.assertFalse(pinpointPy.trace_has_root())
def _test_api_flow(self):
self.assertTrue(pinpointPy.set_agent(collector_host='unix:/tmp/unexist.sock'))
# self.assertTrue(pinpointPy.enable_debug(None))
while self.thread_running:
pinpointPy.start_trace()
pinpointPy.set_special_key('sid','12345678')
pinpointPy.add_clue("key","value3")
pinpointPy.add_clues("key","value3")
value = pinpointPy.get_special_key('sid')
self.assertEqual(value,'12345678')
pinpointPy.mark_as_error("fghjk","fghjkl",234234)
pinpointPy.end_trace()
value = pinpointPy.get_special_key('sid')
self.assertFalse(value)
def _test_api_flow(self):
self.assertTrue(
pinpointPy.set_agent(collector_host='unix:/tmp/unexist.sock'))
# self.assertTrue(pinpointPy.enable_debug(None))
while True:
id = str(random.randint(1, 10000000))
pinpointPy.start_trace()
pinpointPy.set_special_key('sid', id)
pinpointPy.add_clue("key", "value3")
pinpointPy.add_clues("key", "value3")
value = pinpointPy.get_special_key('sid')
self.assertEqual(value, id)
pinpointPy.mark_as_error("fghjk", "fghjkl", 234234)
pinpointPy.end_trace()
def onBefore(self, *args, **kwargs):
global _Id_
id = _Id_.get()
traceId = pinpointPy.start_trace(id)
_Id_.set(traceId)
self.traceId = traceId
return (args, kwargs)
def test_trace_life(self):
self.assertEqual(pinpointPy.start_trace(), 1)
self.assertEqual(pinpointPy.start_trace(), 2)
self.assertEqual(pinpointPy.start_trace(), 3)
self.assertEqual(pinpointPy.start_trace(), 4)
pinpointPy.add_clue("key", "value")
pinpointPy.add_clue("key", "value3")
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
self.assertEqual(pinpointPy.end_trace(), 3)
self.assertEqual(pinpointPy.end_trace(), 2)
self.assertEqual(pinpointPy.end_trace(), 1)
self.assertEqual(pinpointPy.end_trace(), 0)
async def pinpoint_trace(*args, **kwargs):
id = pinpointId.get()
id = pinpointPy.start_trace(id)
pinpointId.set(id)
func_name = func.__name__
pinpointPy.add_clue('name', func_name, id, 0)
pinpointPy.add_clues('start', '3434', id)
ret = await func(*args, **kwargs)
pinpointPy.add_clue('end', '3434', id)
id = pinpointPy.end_trace(id)
async def pinpoint_trace(*args, **kwargs):
id = pinpointId.get()
id = pinpointPy.start_trace(id)
pinpointId.set(id)
func_name = func.__name__
pinpointPy.add_clue('name', func_name, id, 0)
pinpointPy.add_clues('start', '3434', id)
pinpointPy.set_context_key('sid', '12345', id)
value = pinpointPy.get_context_key('sid', id)
self.assertEqual(value, '12345')
ret = await func(*args, **kwargs)
pinpointPy.add_clue('end', '3434', id)
id = pinpointPy.end_trace(id)
def test_trace_life(self):
pinpointPy.start_trace()
pinpointPy.start_trace()
pinpointPy.start_trace()
pinpointPy.start_trace()
pinpointPy.add_clue("key", "value")
pinpointPy.add_clue("key", "value3")
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.end_trace()
async def pinpoint_trace(*args, **kwargs):
start_size = pinpointPy.start_trace()
func_name = func.__name__
pinpointPy.add_clue('name', func_name)
pinpointPy.add_clues('sdf', '3434')
ret = await func(*args, **kwargs)
pinpointPy.add_clue('sdf', '3434')
end_size = pinpointPy.end_trace()
if func_name == 'unit':
self.assertEqual(start_size, 1)
self.assertEqual(end_size, 0)
elif func_name == 'unit2':
self.assertEqual(start_size, 2)
self.assertEqual(end_size, 1)
return ret
def test_trace_life(self):
pinpointPy.start_trace()
pinpointPy.start_trace()
pinpointPy.start_trace()
pinpointPy.start_trace()
pinpointPy.add_clue("key", "value")
pinpointPy.add_clue("key", "value3")
pinpointPy.set_context_key('sid', '12345')
value = pinpointPy.get_context_key('sid')
self.assertEqual(value, '12345')
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.add_clues("key", "values")
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.end_trace()
pinpointPy.force_flush_trace(10)
def onBefore(self, *args, **kwargs):
pinpointPy.start_trace()
return (args, kwargs)
def onBefore(self, *args, **kwargs):
id = pinpointId.get()
new_id = pinpointPy.start_trace(id)
pinpointId.set(new_id)
self.node_id = new_id
return (args, kwargs)
def onBefore(self, *args, **kwargs):
id = context.get('_pinpoint_id_', default=0)
traceId = pinpointPy.start_trace(id)
context['_pinpoint_id_'] = traceId
self.traceId = traceId
return (args, kwargs)
