Change package path
This commit is contained in:
@@ -1,7 +1,11 @@
|
||||
# queue
|
||||
|
||||
[](https://pkg.go.dev/git.belvedersky.ru/queue)
|
||||
|
||||
|
||||
Simple and lightweight queue implementation in Go with optional Dead Letter Queue (DLQ) support.
|
||||
|
||||
|
||||
Repository:
|
||||
https://git.belvedersky.ru/belvedersky/queue
|
||||
|
||||
|
||||
@@ -6,7 +6,7 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"git.belvedersky.ru/belvedersky/queue"
|
||||
"git.belvedersky.ru/queue"
|
||||
)
|
||||
|
||||
type benchTask struct {
|
||||
|
||||
@@ -5,7 +5,7 @@ import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"git.belvedersky.ru/belvedersky/queue"
|
||||
"git.belvedersky.ru/queue"
|
||||
)
|
||||
|
||||
// Example_basic демонстрирует базовое использование очереди.
|
||||
|
||||
@@ -6,7 +6,7 @@ import (
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"git.belvedersky.ru/belvedersky/queue"
|
||||
"git.belvedersky.ru/queue"
|
||||
)
|
||||
|
||||
// Example_DLQ показывает, как использовать Dead Letter Queue
|
||||
|
||||
@@ -6,7 +6,7 @@ import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"git.belvedersky.ru/belvedersky/queue"
|
||||
"git.belvedersky.ru/queue"
|
||||
)
|
||||
|
||||
// Example_parallel демонстрирует работу нескольких воркеров.
|
||||
|
||||
2
go.mod
2
go.mod
@@ -1,3 +1,3 @@
|
||||
module git.belvedersky.ru/belvedersky/queue
|
||||
module git.belvedersky.ru/queue
|
||||
|
||||
go 1.25.1
|
||||
|
||||
25
queue.go
25
queue.go
@@ -9,19 +9,19 @@ import (
|
||||
|
||||
// QueueHooks — набор хуков для метрик и мониторинга.
|
||||
type QueueHooks struct {
|
||||
OnProduced func() // вызывается при успешном добавлении элемента
|
||||
OnHandled func() // вызывается при успешной обработке
|
||||
OnDropped func(interface{}) // вызывается при пропуске (cancel/timeout)
|
||||
OnProduced func() // вызывается при успешном добавлении элемента
|
||||
OnHandled func() // вызывается при успешной обработке
|
||||
OnDropped func(any) // вызывается при пропуске (cancel/timeout)
|
||||
}
|
||||
|
||||
// Queue — потокобезопасная обобщённая FIFO очередь с DLQ, параллельной обработкой
|
||||
// и корректным завершением без data race (даже под -race).
|
||||
// и корректным завершением без data race
|
||||
type Queue[T any] struct {
|
||||
ch chan T
|
||||
capacity int
|
||||
|
||||
mu sync.RWMutex // защищает: handlers/errorFunc/dlq/hooks/closed
|
||||
sendMu sync.RWMutex // синхронизирует ВСЕ записи в канал и его закрытие
|
||||
sendMu sync.RWMutex // синхронизирует записи в канал и его закрытие
|
||||
handlers []func(T)
|
||||
|
||||
errorFunc func(error)
|
||||
@@ -47,7 +47,6 @@ func NewQueue[T any](capacity int) *Queue[T] {
|
||||
}
|
||||
|
||||
// ---------- Конфигурация ----------
|
||||
|
||||
func (q *Queue[T]) Register(handler func(T)) {
|
||||
if handler == nil {
|
||||
panic("handler cannot be nil")
|
||||
@@ -57,24 +56,28 @@ func (q *Queue[T]) Register(handler func(T)) {
|
||||
q.mu.Unlock()
|
||||
}
|
||||
|
||||
// ---------- При ошибке ----------
|
||||
func (q *Queue[T]) OnError(fn func(error)) {
|
||||
q.mu.Lock()
|
||||
q.errorFunc = fn
|
||||
q.mu.Unlock()
|
||||
}
|
||||
|
||||
// ---------- Установка DLQ ----------
|
||||
func (q *Queue[T]) SetDLQ(dlq *Queue[T]) {
|
||||
q.mu.Lock()
|
||||
q.dlq = dlq
|
||||
q.mu.Unlock()
|
||||
}
|
||||
|
||||
// ---------- Установка Хуков ----------
|
||||
func (q *Queue[T]) SetHooks(h QueueHooks) {
|
||||
q.mu.Lock()
|
||||
q.hooks = h
|
||||
q.mu.Unlock()
|
||||
}
|
||||
|
||||
// ---------- Установка RateLimit ----------
|
||||
func (q *Queue[T]) SetRateLimit(rps int) {
|
||||
if rps > 0 {
|
||||
q.rateLimit = time.Tick(time.Second / time.Duration(rps))
|
||||
@@ -82,7 +85,6 @@ func (q *Queue[T]) SetRateLimit(rps int) {
|
||||
}
|
||||
|
||||
// ---------- Запуск обработчиков ----------
|
||||
|
||||
func (q *Queue[T]) HandleParallel(ctx context.Context, workers int) {
|
||||
if workers <= 0 {
|
||||
workers = 1
|
||||
@@ -110,12 +112,10 @@ func (q *Queue[T]) HandleParallel(ctx context.Context, workers int) {
|
||||
}
|
||||
|
||||
// ---------- Отправка в канал (синхронизированная) ----------
|
||||
|
||||
// sendBlocking — блокирующая отправка (используется Produce).
|
||||
// Берём RLock на всё время операции записи, чтобы Close() (Lock) не мог выполнить close(q.ch)
|
||||
// одновременно с записью — это устраняет data race.
|
||||
func (q *Queue[T]) sendBlocking(item T) bool {
|
||||
// Быстрая проверка закрытия под RLock (необязательна, но экономит работу)
|
||||
// Быстрая проверка закрытия под RLock
|
||||
q.sendMu.RLock()
|
||||
q.mu.RLock()
|
||||
closed := q.closed
|
||||
@@ -167,7 +167,6 @@ func (q *Queue[T]) sendNonBlocking(item T) bool {
|
||||
}
|
||||
|
||||
// ---------- Паблик-API продюсеров ----------
|
||||
|
||||
// Produce — блокирующая постановка. Если очередь закрыта — элемент игнорируется.
|
||||
func (q *Queue[T]) Produce(item T) {
|
||||
_ = q.sendBlocking(item)
|
||||
@@ -212,7 +211,6 @@ func (q *Queue[T]) TryProduce(item T, timeout time.Duration) bool {
|
||||
}
|
||||
|
||||
// ---------- Завершение ----------
|
||||
|
||||
func (q *Queue[T]) Shutdown(ctx context.Context) error {
|
||||
q.Close()
|
||||
done := make(chan struct{})
|
||||
@@ -245,7 +243,6 @@ func (q *Queue[T]) Close() {
|
||||
func (q *Queue[T]) Wait() { q.wg.Wait() }
|
||||
|
||||
// ---------- Инфо-методы ----------
|
||||
|
||||
func (q *Queue[T]) Len() int { return len(q.ch) }
|
||||
func (q *Queue[T]) Cap() int { return q.capacity }
|
||||
func (q *Queue[T]) IsEmpty() bool { return len(q.ch) == 0 }
|
||||
@@ -261,7 +258,6 @@ func (q *Queue[T]) String() string {
|
||||
}
|
||||
|
||||
// ---------- Внутренняя обработка ----------
|
||||
|
||||
func (q *Queue[T]) processItem(item T) {
|
||||
q.mu.RLock()
|
||||
handlers := append([]func(T){}, q.handlers...)
|
||||
@@ -269,7 +265,6 @@ func (q *Queue[T]) processItem(item T) {
|
||||
dlq := q.dlq
|
||||
hooks := q.hooks
|
||||
q.mu.RUnlock()
|
||||
|
||||
for _, h := range handlers {
|
||||
func() {
|
||||
defer func() {
|
||||
|
||||
@@ -7,7 +7,7 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"git.belvedersky.ru/belvedersky/queue"
|
||||
"git.belvedersky.ru/queue"
|
||||
)
|
||||
|
||||
type Task struct {
|
||||
|
||||
Reference in New Issue
Block a user