python的進程間通信--共享內存

最近看進程間通信方式，剛好自己也在深入學習python，看到python支持共享內存。所以寫下這篇筆記。

python中，有一個 multiprocessing.shared_memory.SharedMemory類，用來創建和訪問共享內存。

以下是一個例子：

創建共享內存部分，我們使用C代碼，此處參考 大大飛魚老哥的代碼： https://blog.csdn.net/ababab12345/article/details/102931841

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <sys/mman.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>

#define MMAP_DATA_SIZE 1024

int main(int argc,char * argv[])
{
    char * data;
    int fd = shm_open("/shm-file0001", O_CREAT|O_RDWR, 0777);
    
    if (fd < 0) {
        printf("shm_open failed!\n");
        return -1;
    }
    
    ftruncate(fd, MMAP_DATA_SIZE);
    
    data = (char*)mmap(NULL, 1024, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0);
    if (!data) {
        printf("mmap failed\n");
        close(fd);
    }
    
    
    sprintf(data,"This is a share memory! %d\n",fd);
    
    munmap(data,MMAP_DATA_SIZE);
    
    
    close(fd);
    getchar();
    
    shm_unlink("/shm-file0001");
    
    return 0;
}

 

通過xcode編譯運行這段C代碼：

 

 python代碼部分我們只寫一個共享內存的讀取：

from multiprocessing import shared_memory

if __name__ == '__main__':
    existing_shm = shared_memory.SharedMemory(name='shm-file0001')
    content = existing_shm.buf.tobytes()
    print(content)
    existing_shm.close()

 

最終python部分代碼運行結果：

 

進一步，class multiprocessing.managers.SharedMemoryManager提供了一種比較方便的管理共享內存的方式。由SharedMemoryManager返回的sharedMemory,在sharedMemoryManager的shutdown()方法被調用后，都會自動清理掉，不用程序員再額外費心了。