eson/structure
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add GetOrSet

Browse Source
This commit is contained in:
eson
2019-04-17 03:22:07 +08:00
parent 3bc9a99ca9
commit 8d0df4bfed
4 changed files with 104 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
sparse_array/array2/array2.go
View File

@@ -108,6 +108,29 @@ func (arr *Array2) Get(idx int) (interface{}, bool) {
return v, v != nil
}
func (arr *Array2) GetOrSet(idx int, DoSetValue func([]interface{}, int)) (result interface{}, isSet bool) {
yindex := idx / arr.xsize
xindex := idx % arr.xsize
xdata := arr.data[yindex]
if xdata == nil {
xdata = make([]interface{}, arr.xsize, arr.xsize)
arr.data[yindex] = xdata
}
result = xdata[xindex]
if result == nil {
DoSetValue(xdata, xindex)
result = xdata[xindex]
if result == nil {
panic("DoSetValue Not Set <nil> Value")
}
arr.sizes[yindex]++
return result, true
}
return result, false
}
func (arr *Array2) Del(idx int) (interface{}, bool) {
yindex := idx / arr.xsize
xindex := idx % arr.xsize