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 onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_REMOTE_METHOD) arg = self.get_arg(*args, **kwargs) pinpointPy.add_clues(PP_ARGS, arg) ############################################################### return args, kwargs
def onEnd(self, ret): ############################################################### pinpointPy.add_clue(PP_NEXT_SPAN_ID, pinpointPy.get_context_key(PP_NEXT_SPAN_ID)) pinpointPy.add_clues(PP_HTTP_STATUS_CODE, str(ret.status_code)) pinpointPy.add_clues(PP_RETURN, str(ret)) ############################################################### super().onEnd(ret) return ret
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, args[0].get_view_name()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_METHOD_CALL) ############################################################### url = '{%s %r}' % (args[1].method, args[1].get_full_path()) pinpointPy.add_clues(PP_ARGS, url) return args, kwargs
def funcOnBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_CELERY) ############################################################### return args, kwargs
def onBefore(self,*args, **kwargs): args, kwargs = super(PinpointCommonPlugin,self).onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME,self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE,PP_METHOD_CALL) arg = self.get_arg(*args, **kwargs) pinpointPy.add_clues(PP_ARGS, arg) ############################################################### # print( threading.currentThread().ident) return args,kwargs
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)
def onBefore(self,*args, **kwargs): super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME,self.getFuncUniqueName(),self.node_id) pinpointPy.add_clue(PP_SERVER_TYPE,PP_METHOD_CALL,self.node_id) arg = self.get_arg(*args, **kwargs) pinpointPy.add_clues(PP_ARGS, arg,self.node_id) ############################################################### # print( threading.currentThread().ident) return args,kwargs
def onBefore(self, *args, **kwargs): args, kwargs = super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(FuncName, self.getFuncUniqueName()) pinpointPy.add_clue(ServerType, PYTHON_METHOD_CALL) arg = self.get_arg(*args, **kwargs) pinpointPy.add_clues(PY_ARGS, arg) ############################################################### # print( threading.currentThread().ident) return args, kwargs
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) self.url = args[0] generatePinpointHeader(self.url, kwargs['headers']) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_METHOD_CALL) pinpointPy.add_clues(PP_ARGS, self.url) ############################################################### return args, kwargs
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) self.url = args[0] self.target = urlparse(self.url).netloc generatePinpointHeader(self.target, kwargs['headers']) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_REMOTE_METHOD) pinpointPy.add_clues(PP_ARGS, self.url) ############################################################### return args, kwargs
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 onBefore(self, *args, **kwargs): args, kwargs = super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(FuncName, self.getFuncUniqueName()) pinpointPy.add_clue(ServerType, PYTHON_METHOD_CALL) arg = self.get_arg(*args, **kwargs) pinpointPy.add_clues(PY_ARGS, arg) if "headers" in kwargs: self.handleHttpHeader(args[0], kwargs["headers"]) else: kwargs["headers"] = {} self.handleHttpHeader(args[0], kwargs["headers"]) ############################################################### return args, kwargs
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_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.set_special_key('sid',id) self.assertEqual(pinpointPy.start_trace(),1) pinpointPy.add_clue("key","value3") pinpointPy.add_clues("key","value3") value = pinpointPy.get_special_key('sid') self.assertEqual(value,id) self.assertEqual(pinpointPy.end_trace(),0) pinpointPy.mark_as_error("fghjk","fghjkl",234234)
def onEnd(self, ret): ############################################################### if ret: response = ret pinpointPy.add_clues(PP_HTTP_STATUS_CODE, str(response.status_code)) if response.status_code > 400: pinpointPy.add_clue( PP_ADD_EXCEPTION, "status_code is %d" % (response.status_code)) ############################################################### super(CinderRequestPlugin, self).onEnd(ret) return ret
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_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_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)
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()
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_MYSQL) pinpointPy.add_clue(PP_SQL_FORMAT, args[1]) ############################################################### dst = self.get_dst(args[0]) pinpointPy.add_clue(PP_DESTINATION, dst) return args, kwargs
def onBefore(self, *args, **kwargs): super(PyMysqlPlugin, self).onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_MYSQL) pinpointPy.add_clue(PP_SQL_FORMAT, args[1]) ############################################################### cursor = args[0] dst = cursor.connection.get_host_info() pinpointPy.add_clue(PP_DESTINATION, dst) return args, kwargs
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): super().onBefore(*args, **kwargs) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME,self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_REDIS) ############################################################### pinpointPy.add_clue(PP_DESTINATION, str(args[0])) return args,kwargs
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) pinpointPy.add_clue(PP_SERVER_TYPE, PP_REMOTE_METHOD) pinpointPy.add_clue(PP_DESTINATION, args[0].name) nsid = generateSid() pinpointPy.set_context_key(PP_NEXT_SPAN_ID, nsid) pinpointPy.add_clue(PP_NEXT_SPAN_ID, nsid) return args, kwargs
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) connection = args[0] self.dst = connection.get_host_info() ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_MYSQL) query = args[1] ## copy from MySQLdb def query(self, query): if isinstance(query, bytearray): query = bytes(query) pinpointPy.add_clue(PP_SQL_FORMAT, query.decode('utf-8')) ############################################################### pinpointPy.add_clue(PP_DESTINATION, self.dst) return args, kwargs
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) self.dst = str(args[0].database._client.address) # ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName(), self.traceId) pinpointPy.add_clue(PP_SERVER_TYPE, PP_MONGDB_EXE_QUERY, self.traceId) # ############################################################### pinpointPy.add_clue(PP_DESTINATION, self.dst, self.traceId) return args, kwargs
def onEnd(self,ret): ############################################################### pinpointPy.add_clue(PP_DESTINATION, urlparse(self.url)['netloc']) pinpointPy.add_clue(PP_SERVER_TYPE, PP_REMOTE_METHOD) pinpointPy.add_clue(PP_NEXT_SPAN_ID, pinpointPy.get_context_key(PP_NEXT_SPAN_ID)) pinpointPy.add_clues(PP_HTTP_URL, self.url) pinpointPy.add_clues(PP_HTTP_STATUS_CODE, str(ret.status_code)) pinpointPy.add_clues(PP_RETURN, str(ret)) ############################################################### super(UrlOpenPlugin,self).onEnd(ret) return ret
def onBefore(self, *args, **kwargs): super().onBefore(*args, **kwargs) collection = args[0] self.dst = str(collection.__database.address) ############################################################### pinpointPy.add_clue(PP_INTERCEPTOR_NAME, self.getFuncUniqueName()) pinpointPy.add_clue(PP_SERVER_TYPE, PP_MONGDB_EXE_QUERY) # pinpointPy.add_clues(PP_ARGS, ) ############################################################### pinpointPy.add_clue(PP_DESTINATION, self.dst) return args, kwargs
def onEnd(self, ret): ############################################################### pinpointPy.add_clue("dst", self.getHostFromURL(self.url)) pinpointPy.add_clue("stp", PYTHON_REMOTE_METHOD) pinpointPy.add_clue('nsid', self.get_nsid()) pinpointPy.add_clues(HTTP_URL, self.url) pinpointPy.add_clues(HTTP_STATUS_CODE, str(ret.status_code)) pinpointPy.add_clues(PY_RETURN, str(ret)) ############################################################### super().onEnd(ret) return ret