Python 多线程搜索txt文件的内容,并写入搜到的内容(Lock)方法

2023-08-16 17:16:05 74

importthreading importos classFind(threading.Thread):#搜索数据的线程类 def__init__(self,datalist,startIndex,endIndex,searchstr,savefile):#datalist要搜索的内容列表，startIndex列表搜索范围的开始下标，searchstr要搜索的内容 threading.Thread.__init__(self) self.datalist=datalist#要搜索的数据的内存地址 self.startIndex=startIndex#开始的索引 self.endIndex=endIndex#结束的索引 self.seachstr=searchstr#需要搜索的数据 self.savefile=savefile defrun(self): self.findlist=[] foriinrange(self.startIndex,self.endIndex): line=self.datalist[i].decode("gbk","ignore")#读取一行 ifline.find(self.seachstr)!=-1: print(self.getName(),line,end="")#搜索数据 self.findlist.append(line) globalmutex#多线程共享全局变量(全局锁) withmutex:#获取锁(自动释放锁) forlineinself.findlist: self.savefile.write(line.encode("gbk")) mutex=threading.Lock()#创建一个锁 savefile=open("c:\\zhaodao.txt","wb")#搜索到的内容写入该文件 path="C:\\data1.txt"#要搜索的文件 file=open(path,"rb") datalist=file.readlines()#全部读入内存 lines=len(datalist)#所有的行数 searchstr=input("输入要查询的数据") N=10#开启10个线程 threadlist=[]#线程列表 #9790-10000001000000-20000002000000-3000000 foriinrange(0,N-1):#0,1,2,3,4,5,6,7,8数据切割 mythd=Find(datalist,i*(lines//(N-1)),(i+1)*(lines//(N-1)),searchstr,savefile)#//表示整除 mythd.start() threadlist.append(mythd)#添加到线程列表 #97=97//10*10=90 mylastthd=Find(datalist,lines//(N-1)*(N-1),lines,searchstr,savefile)#最后的线程搜索剩下的尾数 mylastthd.start() threadlist.append(mylastthd)#添加到线程列表 forthdinthreadlist:#遍历线程列表 thd.join() print("finish")

Python 多线程搜索txt文件的内容,并写入搜到的内容(Lock)方法

热门推荐

随机推荐