Go語言共享內(nèi)存讀寫實(shí)例分析

2020-04-01 19:19:04

字體：大中小

供稿：網(wǎng)友

這篇文章主要介紹了Go語言共享內(nèi)存讀寫方法,實(shí)例分析了共享內(nèi)存的原理與讀寫技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下

本文實(shí)例分析了Go語言共享內(nèi)存讀寫的方法。分享給大家供大家參考。具體分析如下：

前面分析了Go語言指針運(yùn)算和內(nèi)嵌C代碼的方法，做了一個Go語言共享內(nèi)存讀寫的實(shí)驗(yàn)。

先大概說下什么是共享內(nèi)存。我們知道不同進(jìn)程見的內(nèi)存是互相獨(dú)立的，沒辦法直接互相操作對方內(nèi)的數(shù)據(jù)，而共享內(nèi)存則是靠操作系統(tǒng)提供的內(nèi)存映射機(jī)制，讓不同進(jìn)程的一塊地址空間映射到同一個虛擬內(nèi)存區(qū)域上，使不同的進(jìn)程可以操作到一塊共用的內(nèi)存塊。共享內(nèi)存是效率最高的進(jìn)程間通訊機(jī)制，因?yàn)閿?shù)據(jù)不需要在內(nèi)核和程序之間復(fù)制。

共享內(nèi)存用到的是系統(tǒng)提供的mmap函數(shù)，它可以將一個文件映射到虛擬內(nèi)存的一個區(qū)域中，程序使用指針引用這個區(qū)域，對這個內(nèi)存區(qū)域的操作會被回寫到文件上，Go內(nèi)置的syscall包中有mmap函數(shù)，但是它是經(jīng)過封裝的，返回的是[]byte，沒辦法做我需求的指針運(yùn)算，所以我還是用cgo來調(diào)用原生的mmap。

實(shí)驗(yàn)分為讀和寫兩個程序，這樣我們可以觀察到讀進(jìn)程可以讀到寫進(jìn)程寫入共享內(nèi)存的信息。

下面是shm_writer.go的代碼：

復(fù)制代碼代碼如下:

		package main
		/*
		#cgo linux LDFLAGS: -lrt
		#include <fcntl.h>
		#include <unistd.h>
		#include <sys/mman.h>
		#define FILE_MODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)
		int my_shm_new(char *name) {
		    shm_unlink(name);
		    return shm_open(name, O_RDWR|O_CREAT|O_EXCL, FILE_MODE);
		}
		*/
		import "C"
		import (
		    "fmt"
		    "unsafe"
		)
		const SHM_NAME = "my_shm"
		const SHM_SIZE = 4 * 1000 * 1000 * 1000
		type MyData struct {
		    Col1 int
		    Col2 int
		    Col3 int
		}
		func main() {
		    fd, err := C.my_shm_new(C.CString(SHM_NAME))
		    if err != nil {
		        fmt.Println(err)
		        return
		    }
		    C.ftruncate(fd, SHM_SIZE)
		    ptr, err := C.mmap(nil, SHM_SIZE, C.PROT_READ|C.PROT_WRITE, C.MAP_SHARED, fd, 0)
		    if err != nil {
		        fmt.Println(err)
		        return
		    }
		    C.close(fd)
		    data := (*MyData)(unsafe.Pointer(ptr))
		    data.Col1 = 100
		    data.Col2 = 876
		    data.Col3 = 8021
		}

下面是shm_reader.go的代碼：

復(fù)制代碼代碼如下:

		package main
		/*
		#cgo linux LDFLAGS: -lrt
		#include <fcntl.h>
		#include <unistd.h>
		#include <sys/mman.h>
		#define FILE_MODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)
		int my_shm_open(char *name) {
		    return shm_open(name, O_RDWR);
		}
		*/
		import "C"
		import (
		    "fmt"
		    "unsafe"
		)
		const SHM_NAME = "my_shm"
		const SHM_SIZE = 4 * 1000 * 1000 * 1000
		type MyData struct {
		    Col1 int
		    Col2 int
		    Col3 int
		}
		func main() {
		    fd, err := C.my_shm_open(C.CString(SHM_NAME))
		    if err != nil {
		        fmt.Println(err)
		        return
		    }
		    ptr, err := C.mmap(nil, SHM_SIZE, C.PROT_READ|C.PROT_WRITE, C.MAP_SHARED, fd, 0)
		    if err != nil {
		        fmt.Println(err)
		        return
		    }
		    C.close(fd)
		    data := (*MyData)(unsafe.Pointer(ptr))
		    fmt.Println(data)
		}