• Python如何实现函数调用追踪


    最近使用了分布式追踪系列:OpenTracing与Jaeger那些,日常的话可能很难使用,我实现了一种简单的调用追踪。

    import uuid
    import os
    import time
    
    
    l = []
    class Recorder(object):
        def __init__(self,servername,root_span = None):
            if root_span == None:
                self.__span = Span(servername)
            else:
                self.__span = Span(servername,root_span)
    
        #上下文管理器
        def __enter__(self):
            return self.__span
    
        # 退出方法中,用来实现善后处理工作
        def __exit__(self, exc_type, exc_val, exc_tb):
            self.__span.record()
            self.__span.record_save(self.__span.span)
    
    
    class Span(object):
        def __init__(self,servername,root_span = None):
            self.servername = servername
            self.span = self.newspan()
            if root_span != None:
                root_span.dic['child_span'] = self.span
                self.span["root_span_flag"] = False
    
        def newspan(self):
            self.dic = {
                "spanid": uuid.uuid4().int,
                "servername": self.servername,
                "location": "",
                "ip": "",
                "durationtime": 0,
                "starttime": time.time(),
                "endtime":0,
                "tag": "",
                "log": "",
                "root_span_flag":True,
                "child_span": ""
            }
            return self.dic
    
        def record_save(self, span):
            currenttracer = {"id": span["spanid"], "data": span}
            print(currenttracer)
            l.append(currenttracer)
    
        def record(self):
            self.span["servername"] = self.servername
            self.span["location"] = os.getcwd() + "." + self.servername
            self.span["endtime"] = time.time()
            self.span["durationtime"] = self.span["endtime"] - self.span["starttime"]
    
        def setspantag(self,tag):
            self.span["tag"] = tag
    
        def setspanlog(self,log):
            self.span["log"] = log
    
    # 连续调用
    with Recorder('server1') as span:
        time.sleep(1)
        span.setspantag("test")
        #调用server1方法
        print("server1")
        with Recorder('server2',span) as span1:
            time.sleep(2)
            # 调用server2方法
            print("server2")
            with Recorder('server3',span1) as span2:
                time.sleep(0.5)
                # 调用server3方法
                print("server3")
    
    #单独调用记录
    with Recorder('server4') as span_test1:
        time.sleep(1.5)
        print("server4")
    
    with Recorder('server5') as span_test2:
        time.sleep(2)
        print("server5")

    运行结果:

    server1
    server2
    server3
    {'id': 224716339449765695394515303164364012192, 'data': {'spanid': 224716339449765695394515303164364012192, 'servername': 'server3', 'location': 'D:\python_protest\protest\trace\trace_main.server3', 'ip': '', 'durationtime': 0.5004403591156006, 'starttime': 1598947338.0551107, 'endtime': 1598947338.555551, 'tag': '', 'log': '', 'root_span_flag': False, 'child_span': None}}
    {'id': 254736847532758359233387151739984206570, 'data': {'spanid': 254736847532758359233387151739984206570, 'servername': 'server2', 'location': 'D:\python_protest\protest\trace\trace_main.server2', 'ip': '', 'durationtime': 2.501264810562134, 'starttime': 1598947336.0542862, 'endtime': 1598947338.555551, 'tag': '', 'log': '', 'root_span_flag': False, 'child_span': {'spanid': 224716339449765695394515303164364012192, 'servername': 'server3', 'location': 'D:\python_protest\protest\trace\trace_main.server3', 'ip': '', 'durationtime': 0.5004403591156006, 'starttime': 1598947338.0551107, 'endtime': 1598947338.555551, 'tag': '', 'log': '', 'root_span_flag': False, 'child_span': None}}}
    {'id': 91028031631192607088457781914309166266, 'data': {'spanid': 91028031631192607088457781914309166266, 'servername': 'server1', 'location': 'D:\python_protest\protest\trace\trace_main.server1', 'ip': '', 'durationtime': 3.5021069049835205, 'starttime': 1598947335.0534441, 'endtime': 1598947338.555551, 'tag': 'test', 'log': '', 'root_span_flag': True, 'child_span': {'spanid': 254736847532758359233387151739984206570, 'servername': 'server2', 'location': 'D:\python_protest\protest\trace\trace_main.server2', 'ip': '', 'durationtime': 2.501264810562134, 'starttime': 1598947336.0542862, 'endtime': 1598947338.555551, 'tag': '', 'log': '', 'root_span_flag': False, 'child_span': {'spanid': 224716339449765695394515303164364012192, 'servername': 'server3', 'location': 'D:\python_protest\protest\trace\trace_main.server3', 'ip': '', 'durationtime': 0.5004403591156006, 'starttime': 1598947338.0551107, 'endtime': 1598947338.555551, 'tag': '', 'log': '', 'root_span_flag': False, 'child_span': None}}}}
    server4
    {'id': 103171729522922437998918618387133480096, 'data': {'spanid': 103171729522922437998918618387133480096, 'servername': 'server4', 'location': 'D:\python_protest\protest\trace\trace_main.server4', 'ip': '', 'durationtime': 1.5001769065856934, 'starttime': 1598947338.555551, 'endtime': 1598947340.055728, 'tag': '', 'log': '', 'root_span_flag': True, 'child_span': None}}
    server5
    {'id': 320091321623887285825256878422834254741, 'data': {'spanid': 320091321623887285825256878422834254741, 'servername': 'server5', 'location': 'D:\python_protest\protest\trace\trace_main.server5', 'ip': '', 'durationtime': 2.0005736351013184, 'starttime': 1598947340.055728, 'endtime': 1598947342.0563016, 'tag': '', 'log': '', 'root_span_flag': True, 'child_span': None}}

    关于下一步,会使用redis存储其结果,并进行相应的输出分析。

    修过不能追踪同级调用的问题

    import uuid
    import os
    import time
    
    
    l = []
    class Recorder(object):
        def __init__(self,servername,root_span = None):
            if root_span == None:
                self.__span = Span(servername)
            else:
                self.__span = Span(servername,root_span)
    
        #上下文管理器
        def __enter__(self):
            return self.__span
    
        # 退出方法中,用来实现善后处理工作
        def __exit__(self, exc_type, exc_val, exc_tb):
            self.__span.record()
            self.__span.record_save(self.__span.span)
    
    
    class Span(object):
        def __init__(self,servername,root_span = None):
            self.servername = servername
            self.span = self.newspan()
            if root_span != None:
                root_span.dic['child_span'].append(self.span)
                self.span["root_span_flag"] = False
    
        def newspan(self):
            self.dic = {
                "spanid": uuid.uuid4().int,
                "servername": self.servername,
                "location": "",
                "ip": "",
                "durationtime": 0,
                "starttime": time.time(),
                "endtime":0,
                "tag": "",
                "log": "",
                "root_span_flag":True,
                "child_span": []
            }
            return self.dic
    
        def record_save(self, span):
            currenttracer = {"id": span["spanid"], "data": span}
            print(currenttracer)
            l.append(currenttracer)
    
        def record(self):
            self.span["servername"] = self.servername
            self.span["location"] = os.getcwd() + "." + self.servername
            self.span["endtime"] = time.time()
            self.span["durationtime"] = self.span["endtime"] - self.span["starttime"]
    
        def setspantag(self,tag):
            self.span["tag"] = tag
    
        def setspanlog(self,log):
            self.span["log"] = log
    
    
    
    # 连续调用
    with Recorder('server1') as span:
        time.sleep(1)
        span.setspantag("test")
        #调用server1方法
        print("server1")
        with Recorder('server2',span) as span1:
            time.sleep(2)
            # 调用server2方法
            print("server2")
            with Recorder('server3',span1) as span2:
                time.sleep(0.5)
                # 调用server3方法
                print("server3")
    
    
    # 并级连续调用
    # with Recorder('server1') as span1:
    #     time.sleep(1)
    #     span1.setspantag("test")
    #     #调用server1方法
    #     print("server1")
    #     with Recorder('server2',span1) as span2:
    #         time.sleep(2)
    #         # 调用server2方法
    #         print("server2")
    #     with Recorder('server3',span1) as span3:
    #         time.sleep(0.5)
    #         # 调用server3方法
    #         print("server3")
    
    # #单独调用记录
    # with Recorder('server4') as span_test1:
    #     time.sleep(1.5)
    #     print("server4")
    #
    # with Recorder('server5') as span_test2:
    #     time.sleep(2)
    #     print("server5")
  • 相关阅读:
    DB2 9 基础根基(730 测验)认证指南,第 2 部门: 安适性(1)
    DB2 9 根底(730 考验)认证指南,第 3 部门: 访问 DB2 数据(7)
    IT人 软件工程师的务实职业生涯规划
    《Delphi 算法与数据结构》学习与感悟[8]: 单向链表的添加、删除与遍历
    《Delphi 算法与数据结构》简介及下载
    《Delphi 算法与数据结构》学习与感悟[1]: 通过 "顺序查找" 与 "二分查找" 说明算法的重要性
    《Delphi 算法与数据结构》学习与感悟[2]: 数据对齐
    学习 TTreeView [14] StateIndex(状态图标)、OverlayIndex(叠加图标)
    《Delphi 算法与数据结构》学习与感悟[3]: 获取一个字节中非空位的个数
    《Delphi 算法与数据结构》学习与感悟[6]: 一个简单的"单向链表"
  • 原文地址:https://www.cnblogs.com/-wenli/p/13596771.html
Copyright © 2020-2023  润新知