mirror of
https://gitee.com/johng/gf
synced 2026-06-26 09:27:31 +08:00
126 lines
3.0 KiB
Go
126 lines
3.0 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 Array struct {
|
|
mu sync.RWMutex // 互斥锁
|
|
cap int // 初始化设置的数组容量
|
|
size int // 初始化设置的数组大小
|
|
array []interface{} // 底层数组
|
|
}
|
|
|
|
func NewArray(size int, cap ... int) *Array {
|
|
a := &Array{}
|
|
a.size = size
|
|
if len(cap) > 0 {
|
|
a.cap = cap[0]
|
|
a.array = make([]interface{}, size, cap[0])
|
|
} else {
|
|
a.array = make([]interface{}, size)
|
|
}
|
|
return a
|
|
}
|
|
|
|
// 获取指定索引的数据项, 调用方注意判断数组边界
|
|
func (a *Array) Get(index int) interface{} {
|
|
a.mu.RLock()
|
|
defer a.mu.RUnlock()
|
|
value := a.array[index]
|
|
return value
|
|
}
|
|
|
|
// 设置指定索引的数据项, 调用方注意判断数组边界
|
|
func (a *Array) Set(index int, value interface{}) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
a.array[index] = value
|
|
}
|
|
|
|
// 在当前索引位置前插入一个数据项, 调用方注意判断数组边界
|
|
func (a *Array) Insert(index int, value interface{}) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
rear := append([]interface{}{}, a.array[index : ]...)
|
|
a.array = append(a.array[0 : index], value)
|
|
a.array = append(a.array, rear...)
|
|
}
|
|
|
|
// 删除指定索引的数据项, 调用方注意判断数组边界
|
|
func (a *Array) Remove(index int) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
a.array = append(a.array[ : index], a.array[index + 1 : ]...)
|
|
}
|
|
|
|
// 追加数据项
|
|
func (a *Array) Append(value interface{}) {
|
|
a.mu.Lock()
|
|
a.array = append(a.array, value)
|
|
a.mu.Unlock()
|
|
}
|
|
|
|
// 数组长度
|
|
func (a *Array) Len() int {
|
|
a.mu.RLock()
|
|
length := len(a.array)
|
|
a.mu.RUnlock()
|
|
return length
|
|
}
|
|
|
|
// 返回原始数据数组
|
|
func (a *Array) Slice() []interface{} {
|
|
a.mu.RLock()
|
|
array := a.array
|
|
a.mu.RUnlock()
|
|
return array
|
|
}
|
|
|
|
// 清空数据数组
|
|
func (a *Array) Clear() {
|
|
a.mu.Lock()
|
|
if a.cap > 0 {
|
|
a.array = make([]interface{}, a.size, a.cap)
|
|
} else {
|
|
a.array = make([]interface{}, a.size)
|
|
}
|
|
a.mu.Unlock()
|
|
}
|
|
|
|
// 查找指定数值的索引位置，返回索引位置，如果查找不到则返回-1
|
|
func (a *Array) Search(value interface{}) 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 *Array) LockFunc(f func(array []interface{})) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
f(a.array)
|
|
}
|
|
|
|
// 使用自定义方法执行加锁读取操作
|
|
func (a *Array) RLockFunc(f func(array []interface{})) {
|
|
a.mu.RLock()
|
|
defer a.mu.RUnlock()
|
|
f(a.array)
|
|
}
|