wKiom1e7lSzSPeudAAAMUzyp5Ds993.png-wh_50锁(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()