首页 > 解决方案 > 获取切片中值的指针

问题描述

有没有办法获得指向切片中实际值的指针?我在地图上有一个报价树:

map[uint64]map[uint16][]offer

为了通过 id 快速访问(从 1 到 n),我需要一片指针:

[]*offer

实际报价位于第一棵树中,切片中应该是指向实际报价的指针。

我无法遍历树并收集实际值的指针,因为 with rangeof this slice 给了我一个副本,但我总共有超过数十亿个这样的结构,副本会导致我的内存浪费:一个结构 40 个字节导致100+ GB 和更多的未来。我还需要在树中将值的切片存储为连续的,以便对报价进行事实查找。

可能有一些方法可以使用unsafereflect打包来获取这些指针?

指针索引只构建一次,并且两个结构都是不可变的,仅用于查找报价。

更新: 我完全错了。获取指向切片中元素的指针没有问题。我的代码是:

        var offers []offer // actually there was about 50gb of offers
        for i := range offers {
            currentOffer := offers[i]
            s.Relations[currentOffer.Id] = &currentOffer
        }

在此片段之后,应用程序消耗的 ram 总量变为 100+gb。我立即认为从值切片中获取元素会给我该值的副本,但这是错误的。我只是自己将原始值复制到currentOffer变量中。小错误带走了 50gb 的内存。

实际上这个片段可以正常工作:

    for i := range offers {
        s.Relations[offers[i].Id] = &offers[i]
    }

标签: go

解决方案


经过一番调查,我得出以下代码:

package main

import (
    "fmt"
    "reflect"
    "unsafe"
)

type offer struct {
    id uint64
}

func main() {

    sl := []offer{{1}, {id: 2}}

    size := unsafe.Sizeof(offer{})
    header := (*reflect.SliceHeader)(unsafe.Pointer(&sl))
    for i := 0; i < len(sl); i++ {
        offset := uintptr(i) * size
        ptr := header.Data + offset

        o := (*offer)(unsafe.Pointer(ptr))
        o.id = 5
        fmt.Println(o)
    }

    for _, o := range sl {
        fmt.Println(o.id)
    }
}

游乐场: https: //play.golang.org/p/OM3i84cKAB_7

输出:

&{5}
&{5}
5
5

这段代码说明了对切片中实际数据的访问,而不是复制的。

更新: 通过不安全指针获取指针与通过索引从切片获取常规指针的效果相同。


推荐阅读