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())
Esempio n. 2
0
 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
Esempio n. 5
0
    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
Esempio n. 6
0
 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)
Esempio n. 8
0
 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
Esempio n. 9
0
 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
Esempio n. 10
0
 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
Esempio n. 11
0
    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
Esempio n. 12
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)
                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)
Esempio n. 13
0
 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
Esempio n. 14
0
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)
Esempio n. 16
0
    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
Esempio n. 17
0
 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
Esempio n. 18
0
    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)
Esempio n. 19
0
    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)
Esempio n. 20
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()
Esempio n. 21
0
 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
Esempio n. 22
0
 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
Esempio n. 23
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.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)
Esempio n. 24
0
 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
Esempio n. 25
0
 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
Esempio n. 26
0
 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
Esempio n. 27
0
 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
Esempio n. 28
0
 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
Esempio n. 30
0
    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