锁(LOCK)
当有两个或跟多个线程或进程需要操作一个变量或进程时,会出现意想不到的结果,这是因为线程或进程时迸发进行的,对同意变量或文件操作时,会出现同时对其操作,从到导致逻辑错误。
#!/bin/usr/env python #coding:utf-8 import multiprocessing import time class multF(multiprocessing.Process): """docstring for multF""" def __init__(self,filename,proname,stime): multiprocessing.Process.__init__(self) self.filename = filename self.proname = proname self.stime = stime def run(self): with open(self.filename,‘a‘) as f: for x in xrange(1,5): f.write("进程编号:"+str(self.proname)+"---"+str(x)+‘\n‘) time.sleep(self.stime) f.close() print u"文件写入完成." def main(): filename = ‘./test.txt‘ t1 = multF(filename,‘01‘,2) t2 = multF(filename,‘02‘,1) t3 = multF(filename,‘03‘,2) t1.start() t2.start() t3.start() t1.join() t2.join() t3.join() print u"结束" if __name__ == ‘__main__‘: main()
为解决以上问题我们需要引入锁。
锁有两种状态:被锁(locked)和没有被锁(unlocked)。拥有acquire()和release()两种方法,并且遵循以下的规则:
如果一个锁的状态是unlocked,调用acquire()方法改变它的状态为locked; 如果一个锁的状态是locked,acquire()将会阻塞,直到另一个线程或进程调用release()方法释放它; 如果一个锁的状态是unlocked,调用release()会抛出RuntimeError异常; 如果一个锁的状态是locked,调用release()方法改变它的状态Wieunlocked。
解决上面两个进程或线程同时写一个文件的问题的方法就是:我们给写文件的类的构造器中传入一个锁(lock),使用这个锁来保护文件操作,实现在给定的时间只有一个线程写文件。
#!/bin/usr/env python #coding:utf-8 import multiprocessing import time class multF(multiprocessing.Process): """docstring for multF""" def __init__(self,filename,proname,stime,lock): multiprocessing.Process.__init__(self) self.filename = filename self.proname = proname self.stime = stime self.lock = lock def run(self): with open(self.filename,‘a‘) as f: self.lock.acquire() for x in xrange(1,5): f.write("进程编号:"+str(self.proname)+"---"+str(x)+‘\n‘) time.sleep(self.stime) f.close() self.lock.release() print u"文件写入完成." def main(): filename = ‘./test2.txt‘ lock = multiprocessing.Lock() t1 = multF(filename,‘01‘,2, lock) t2 = multF(filename,‘02‘,1,lock) t3 = multF(filename,‘03‘,2,lock) t1.start() t2.start() t3.start() t1.join() t2.join() t3.join() print u"结束" if __name__ == ‘__main__‘: main()
时间: 2024-10-05 15:29:57