mirror of
https://gitee.com/johng/gf
synced 2026-07-03 11:51:04 +08:00
127 lines
2.7 KiB
Go
127 lines
2.7 KiB
Go
// Copyright 2018 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://gitee.com/johng/gf.
|
|
|
|
package garray
|
|
|
|
import "sync"
|
|
|
|
type IntArray struct {
|
|
mu sync.RWMutex // 互斥锁
|
|
cap int // 初始化设置的数组容量
|
|
size int // 初始化设置的数组大小
|
|
array []int // 底层数组
|
|
}
|
|
|
|
func NewIntArray(size int, cap ... int) *IntArray {
|
|
a := &IntArray{}
|
|
a.size = size
|
|
if len(cap) > 0 {
|
|
a.cap = cap[0]
|
|
a.array = make([]int, size, cap[0])
|
|
} else {
|
|
a.array = make([]int, size)
|
|
}
|
|
return a
|
|
}
|
|
|
|
// 获取指定索引的数据项, 调用方注意判断数组边界
|
|
func (a *IntArray) Get(index int) int {
|
|
a.mu.RLock()
|
|
defer a.mu.RUnlock()
|
|
value := a.array[index]
|
|
return value
|
|
}
|
|
|
|
// 设置指定索引的数据项, 调用方注意判断数组边界
|
|
func (a *IntArray) Set(index int, value int) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
a.array[index] = value
|
|
}
|
|
|
|
// 在当前索引位置前插入一个数据项, 调用方注意判断数组边界
|
|
func (a *IntArray) Insert(index int, value int) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
rear := append([]int{}, a.array[index:]...)
|
|
a.array = append(a.array[0:index], value)
|
|
a.array = append(a.array, rear...)
|
|
|
|
}
|
|
|
|
// 删除指定索引的数据项, 调用方注意判断数组边界
|
|
func (a *IntArray) Remove(index int) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
a.array = append(a.array[:index], a.array[index+1:]...)
|
|
}
|
|
|
|
// 追加数据项
|
|
func (a *IntArray) Append(value int) {
|
|
a.mu.Lock()
|
|
a.array = append(a.array, value)
|
|
a.mu.Unlock()
|
|
}
|
|
|
|
// 数组长度
|
|
func (a *IntArray) Len() int {
|
|
a.mu.RLock()
|
|
length := len(a.array)
|
|
a.mu.RUnlock()
|
|
return length
|
|
}
|
|
|
|
// 返回原始数据数组
|
|
func (a *IntArray) Slice() []int {
|
|
a.mu.RLock()
|
|
array := a.array
|
|
a.mu.RUnlock()
|
|
return array
|
|
}
|
|
|
|
// 清空数据数组
|
|
func (a *IntArray) Clear() {
|
|
a.mu.Lock()
|
|
if a.cap > 0 {
|
|
a.array = make([]int, a.size, a.cap)
|
|
} else {
|
|
a.array = make([]int, a.size)
|
|
}
|
|
a.mu.Unlock()
|
|
}
|
|
|
|
// 查找指定数值的索引位置，返回索引位置，如果查找不到则返回-1
|
|
func (a *IntArray) Search(value int) int {
|
|
if len(a.array) == 0 {
|
|
return -1
|
|
}
|
|
a.mu.RLock()
|
|
result := -1
|
|
for index, v := range a.array {
|
|
if v == value {
|
|
result = index
|
|
break
|
|
}
|
|
}
|
|
a.mu.RUnlock()
|
|
|
|
return result
|
|
}
|
|
|
|
// 使用自定义方法执行加锁修改操作
|
|
func (a *IntArray) LockFunc(f func(array []int)) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
f(a.array)
|
|
}
|
|
|
|
// 使用自定义方法执行加锁读取操作
|
|
func (a *IntArray) RLockFunc(f func(array []int)) {
|
|
a.mu.RLock()
|
|
defer a.mu.RUnlock()
|
|
f(a.array)
|
|
}
|