借助二分法匹配时间戳实现快速查找日志内容

前言

　　借助二分法能够快速匹配时间戳，根据时间戳来找出需要查找的日志内容。

　　查找前提条件：

　　　　1 因为是二分法特性，所以日志的时间必须是按从小到大或者从大到小的顺序排列。

　　　　2 根据查找需要，修改匹配时间戳。

　　　　3 根据不同的时间戳格式，修改对应的代码。

import os
import time
import sys

class SearchLog(object):

    def __init__(self, file_name='./log.txt'):　　　　　　
        self.fp = open(file_name)　　　　　　　　　　　　　
        self.fp.seek(0, os.SEEK_END)　　　　　　#把指针移动到结尾
        self.size = self.fp.tell()　　　　　　　#tell()方法能够告诉我们文件的大小
        self.fp.seek(0, os.SEEK_SET)　　　　　　#吧指针移动到文件内容的开头

    def Search_cmp(self, timestamp, line):　　　　　　#定义一个匹配时间戳的方法
        tmp = line.split(' ')　　　　　　　　　　
        riqi=tmp[0]
        shijian=tmp[1]
        realtime=riqi+' '+shijian                  #在这里可以修改要匹配的时间戳格式
        #print realtime
        return cmp(timestamp, realtime)　　　　　　　　　　#cmp函数用于比较二者是否相同，如果返回为-1，那么就说明timestamp>realtime,如果等于0，那么两者相等。
            
            
    def Search_LineHead(self):　　　　　　　　　　　　　　#定义一个匹配每行开头的方法

        while self.fp.tell() > 0:　　　　　　　　　　　　　
            self.fp.seek(-1, os.SEEK_CUR)　　　　　　　　#把文件内容的指针在当前位置往前挪一位
            val = self.fp.read(1)　　　　　　　　　　　　　# 读取该指针位置的内容
            if val == '
':　　　　　　　　　　　　　　　　　#匹配该指针内容的值是否等于换行符，等于就跳出循环
                break　　
            self.fp.seek(-1, os.SEEK_CUR)　　　　　　　　#不等于就就继续往前挪一位查找
            
    def Search_TimeStamp(self, timestamp, start_p = 0):

        fp_start = start_p
        fp_end = self.size

        while fp_start < fp_end:　　　　　　　　　　　　　　#判断指针开始位置是否小于结束位置
            mid = fp_start + (fp_end-fp_start)/2　　　　　#找出中间值（利用二分法）
            self.fp.seek(mid, os.SEEK_SET)　　　　　　　　#通过seek定位到中间值
            self.Search_LineHead()　　　　　　　　　　　　
            line = self.fp.readline()
            val = self.Search_cmp(timestamp, line)　　　　　　#调用该方法能够匹配时间戳，返回值为0那么就说明找到了时间戳
            if val == 0:　　　　　　　　　　
                print "find timestamp:%s" % line
                return True
            elif val == 1:
                fp_start = self.fp.tell()　　　　　　　　　　#如果等于1，说明当前内容的时间戳比输入的时间戳大
            else:
                fp_end = mid　　　　　　　　　　　　　　　　　　#如果等于其他的，说明当前内容的时间戳比输入的时间戳小，
        return False
            
    def Search_FirstStamp(self, timestamp, start_p = 0):

        first_point = -1
        val = self.Search_TimeStamp(timestamp, start_p)

        if val == True:
            point = self.fp.tell()　　　　　　　　　　　　　　　　#这里的逻辑是这样的==>
            while point > 0:　　　　　　　　　　　　　　　　　　　　#如果search_timestamp这个方法匹配到了时间戳，并且指针恒大于0的话，指针就在当前位置往前挪，直到匹配到换行符，再把这行通过readlie()方法整行独取出来。
                self.fp.seek(-1, os.SEEK_CUR)　　　　　　　　　　#此时再去匹配每行的开头是否等于时间戳，如果等于的话，那么就把指针从文件内容开头移动到匹配到时间戳的这行，标记这行为要查找的时间戳出现的第一行。
                self.Search_LineHead()　　　　　　　　　　　　　　　　　　
                point = self.fp.tell()　　　　　　　　　　　　　　　　　　　　　　
                line = self.fp.readline()
                print line
                if line.startswith(timestamp) == True:
                    self.fp.seek(point, os.SEEK_SET)
                    continue
                first_point = self.fp.tell()
                break

        return first_point

    def Search_EndStamp(self, timestamp, start_p = 0):

        last_point = -1
        val = self.Search_TimeStamp(timestamp, start_p)
        if val == True:　　　　　　　　　　　　　　　　　　　　　　　　#道理同上，不过这个的就反转过来，用来查找要查找的时间戳的最后一行。
            last_point = self.fp.tell()
            while last_point < self.size:
                line = self.fp.readline()
                if line.startswith(timestamp) == True:
                    last_point = self.fp.tell()
                    continue
                print line
                self.fp.seek(last_point, os.SEEK_SET)
                break

            return last_point

    def Search_Dump(self, s_p, e_p):　　　　　　　　　　　　　　　　　　#把查找到的内容写入日志
        self.fp.seek(s_p, os.SEEK_SET)
        f_log=open('search_err.log','wb')
        while self.fp.tell() < e_p:
            print self.fp.readline()
            f_log.write(self.fp.readline())
        f_log.close()

    def Search_DeInit(self):
        self.fp.close()

if __name__ == '__main__':
    time_s = time.time()
    obj = SearchLog(sys.argv[3])
    s_point = obj.Search_FirstStamp(sys.argv[1])
    if s_point > -1:
        e_point = obj.Search_EndStamp(sys.argv[2], s_point)
    if s_point > -1:
        print s_point , e_point
        obj.Search_Dump(s_point, e_point)
    else:
        print "not find"
    print time_s, time.time()

    obj.Search_DeInit()