chenshijun
2020-04-15 e485095690a71af1a9bc8e811eb4ea64a95508aa
shmwrap.go
@@ -24,6 +24,16 @@
    return data, id, nil
}
// NewBlockOnly shm block with size, no attach
func NewBlockOnly(size int, key int) (int, error) {
    id, err := shm.Get(key, size, shm.IPC_CREAT|0666)
    if err != nil || id == -1 {
        fmt.Printf("NewBlock Get:%x, %v\n", key, err)
        return -1, err
    }
    return id, nil
}
// AttachBlock attach exist shm
func AttachBlock(key int) ([]byte, int, error) {
    id, err := shm.Get(key, 0, 0)
@@ -91,6 +101,28 @@
    return data, id
}
// CreateShmOnly create shm block with size, only space, no padding, return id(int)
// context for quit
func CreateShmOnly(ctx context.Context, size int, key int) int {
    id, err := NewBlockOnly(size, key)
    if err != nil {
    loopB:
        for {
            select {
            case <-ctx.Done():
                return -1
            default:
                if err == nil {
                    break loopB
                }
                time.Sleep(time.Millisecond)
                id, err = NewBlockOnly(size, key)
            }
        }
    }
    return id
}
// CreatePaddingShm create padding shm block with size, return data-with-padding([]byte), id(int)
// context for quit, padding for fill raw shm block
func CreatePaddingShm(ctx context.Context, size int, key int, padding []byte) ([]byte, int) {
@@ -117,6 +149,18 @@
    return data, id
}
// CheckShmExist check if shm is exist or not
// if exist, return id and true, else return 0 and false
func CheckShmExist(key int) (int, bool) {
    id, err := shm.Get(key, 0, 0)
    if err != nil || id == -1 { //no exist
        fmt.Printf("AttachBlock Get:%x, %v\n", key, err)
        return 0, false
    }
    return id, true
}
// DetachShm destroy
func DetachShm(data []byte) {
    ReleaseBlock(data)
@@ -139,10 +183,11 @@
// RemoveShmKey Remove shmkey
func RemoveShmKey(shmkey int) error {
    _, id, err := AttachBlock(shmkey)
    d, id, err := AttachBlock(shmkey)
    if err != nil {
        return err
    } else {
        Detach(d)
        return shm.Rm(id)
    }
}
@@ -153,6 +198,7 @@
    if err != nil {
        return err
    } else {
        Detach(d)
        return Detach(d)
    }
}
@@ -167,3 +213,25 @@
        return shm.Rm(id)
    }
}
func ShmAttachNum(shmId int) (int,error) {
    var idDs shm.IdDs
    _, err := shm.Ctl(shmId, shm.IPC_STAT, &idDs)
    if err != nil {
        return 0, err
    }
    return int(idDs.Nattch), nil
}
func GetKeyById(shmId int) (int,error) {
    var idDs shm.IdDs
    _, err := shm.Ctl(shmId, shm.IPC_STAT, &idDs)
    if err != nil {
        return 0, err
    }
    return int(idDs.Perm.Key), nil
}