快捷導(dǎo)航

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

更新時(shí)間：2015年02月26日 15:32:23 作者：不是JS

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

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

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

先大概說(shuō)下什么是共享內(nèi)存。我們知道不同進(jìn)程見(jiàn)的內(nèi)存是互相獨(dú)立的，沒(méi)辦法直接互相操作對(duì)方內(nèi)的數(shù)據(jù)，而共享內(nèi)存則是靠操作系統(tǒng)提供的內(nèi)存映射機(jī)制，讓不同進(jìn)程的一塊地址空間映射到同一個(gè)虛擬內(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ù)，它可以將一個(gè)文件映射到虛擬內(nèi)存的一個(gè)區(qū)域中，程序使用指針引用這個(gè)區(qū)域，對(duì)這個(gè)內(nèi)存區(qū)域的操作會(huì)被回寫(xiě)到文件上，Go內(nèi)置的syscall包中有mmap函數(shù)，但是它是經(jīng)過(guò)封裝的，返回的是[]byte，沒(méi)辦法做我需求的指針運(yùn)算，所以我還是用cgo來(lái)調(diào)用原生的mmap。

實(shí)驗(yàn)分為讀和寫(xiě)兩個(gè)程序，這樣我們可以觀察到讀進(jìn)程可以讀到寫(xiě)進(jìn)程寫(xiě)入共享內(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)

}