fumiama/orbyte
1
0
Fork 0
mirror of https://github.com/fumiama/orbyte.git synced 2026-06-05 02:00:30 +08:00
Code Activity

feat: no panic on same goroutine

Browse Source
This commit is contained in:
源文雨
2025-04-14 23:01:07 +09:00
parent 162407d64e
commit b1efb0e49f
2 changed files with 46 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
item.go
View File

@@ -18,6 +18,10 @@ type Item[T any] struct {
cfg any
// align 64
// id only take effect on issync = True
id int64
// align 64
val T
}
@@ -36,7 +40,9 @@ func (b *Item[T]) Trans() T {
panic("use after destroy")
}
if b.pool.issync {
b.stat.setinsyncop(true)
if !b.stat.setinsyncop(true) {
panic("non-unique op")
}
}
val := b.val
atomic.StoreUintptr(
@@ -51,7 +57,10 @@ func (b *Item[T]) Trans() T {
// and will be Reset on putting back.
func (b *Item[T]) HasInvolved() bool {
if b.pool.issync {
b.stat.setinsyncop(true)
if !b.stat.setinsyncop(true) && getGoroutineID() != b.id {
panic("non-unique op")
}
atomic.StoreInt64(&b.id, getGoroutineID())
defer b.stat.setinsyncop(false)
}
return b.stat.isbuffered()
@@ -65,7 +74,10 @@ func (b *Item[T]) V(f func(T)) {
panic("use after destroy")
}
if b.pool.issync {
b.stat.setinsyncop(true)
if !b.stat.setinsyncop(true) && getGoroutineID() != b.id {
panic("non-unique op")
}
atomic.StoreInt64(&b.id, getGoroutineID())
defer b.stat.setinsyncop(false)
}
f(b.val)
@@ -80,7 +92,10 @@ func (b *Item[T]) P(f func(*T)) {
panic("use after destroy")
}
if b.pool.issync {
b.stat.setinsyncop(true)
if !b.stat.setinsyncop(true) && getGoroutineID() != b.id {
panic("non-unique op")
}
atomic.StoreInt64(&b.id, getGoroutineID())
defer b.stat.setinsyncop(false)
}
f(&b.val)
@@ -93,7 +108,10 @@ func (b *Item[T]) Copy() (cb *Item[T]) {
panic("use after destroy")
}
if b.pool.issync {
b.stat.setinsyncop(true)
if !b.stat.setinsyncop(true) && getGoroutineID() != b.id {
panic("non-unique op")
}
atomic.StoreInt64(&b.id, getGoroutineID())
defer b.stat.setinsyncop(false)
}
cb = b.pool.New(b.cfg)