python 锁机制

发布时间:2019-09-21 11:14:23编辑:auto阅读(1639)

    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()




关键字

上一篇: check_memcache with

下一篇: 用python写exploit