upstream-gitea/modules/queue/base.go

// Copyright 2023 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package queue

import (
	"context"
	"time"
)

var pushBlockTime = 5 * time.Second

type baseQueue interface {
	PushItem(ctx context.Context, data []byte) error
	PopItem(ctx context.Context) ([]byte, error)
	HasItem(ctx context.Context, data []byte) (bool, error)
	Len(ctx context.Context) (int, error)
	Close() error
	RemoveAll(ctx context.Context) error
}

func popItemByChan(ctx context.Context, popItemFn func(ctx context.Context) ([]byte, error)) (chanItem chan []byte, chanErr chan error) {
	chanItem = make(chan []byte)
	chanErr = make(chan error)
	go func() {
		for {
			it, err := popItemFn(ctx)
			if err != nil {
				close(chanItem)
				chanErr <- err
				return
			}
			if it == nil {
				close(chanItem)
				close(chanErr)
				return
			}
			chanItem <- it
		}
	}()
	return chanItem, chanErr
}

type baseQueueNotifiableInterface interface {
	getNotifySignalChan() chan struct{}
}

type baseQueueNotifiable struct {
	notifySignal chan struct{}
}

var _ baseQueueNotifiableInterface = (*baseQueueNotifiable)(nil)

func (n *baseQueueNotifiable) notifyPushItem() {
	select {
	case n.notifySignal <- struct{}{}:
	default:
	}
}

func (n *baseQueueNotifiable) getNotifySignalChan() chan struct{} {
	return n.notifySignal
}

func newBaseQueueNotifiable() *baseQueueNotifiable {
	return &baseQueueNotifiable{notifySignal: make(chan struct{}, 1)}
}