废话不多说,直接上代码吧!
import threading import os class Find(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 def run(self): self.findlist=[] for i in range(self.startIndex,self.endIndex): line=self.datalist[i].decode("gbk","ignore") #读取一行 if line.find(self.seachstr)!=-1: print(self.getName(),line,end="") #搜索数据 self.findlist.append(line) global mutex #多线程共享全局变量(全局锁) with mutex: #获取锁(自动释放锁) for line in self.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=[] #线程列表 # 97 9 0-1000000 1000000-2000000 2000000-3000000 for i in range(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) #添加到线程列表 for thd in threadlist: #遍历线程列表 thd.join() print("finish")
以上这篇Python 多线程搜索txt文件的内容,并写入搜到的内容(Lock)方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。