mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-11-03 20:36:07 +01:00 
			
		
		
		
	Multiple Queue improvements: LevelDB Wait on empty, shutdown empty shadow level queue, reduce goroutines etc (#15693)
* move shutdownfns, terminatefns and hammerfns out of separate goroutines Coalesce the shutdownfns etc into a list of functions that get run at shutdown rather then have them run at goroutines blocked on selects. This may help reduce the background select/poll load in certain configurations. * The LevelDB queues can actually wait on empty instead of polling Slight refactor to cause leveldb queues to wait on empty instead of polling. * Shutdown the shadow level queue once it is empty * Remove bytefifo additional goroutine for readToChan as it can just be run in run * Remove additional removeWorkers goroutine for workers * Simplify the AtShutdown and AtTerminate functions and add Channel Flusher * Add shutdown flusher to CUQ * move persistable channel shutdown stuff to Shutdown Fn * Ensure that UPCQ has the correct config * handle shutdown during the flushing * reduce risk of race between zeroBoost and addWorkers * prevent double shutdown Signed-off-by: Andrew Thornton <art27@cantab.net>
This commit is contained in:
		@@ -5,6 +5,8 @@
 | 
			
		||||
package queue
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/modules/nosql"
 | 
			
		||||
 | 
			
		||||
	"gitea.com/lunny/levelqueue"
 | 
			
		||||
@@ -37,6 +39,7 @@ func NewLevelQueue(handle HandlerFunc, cfg, exemplar interface{}) (Queue, error)
 | 
			
		||||
	if len(config.ConnectionString) == 0 {
 | 
			
		||||
		config.ConnectionString = config.DataDir
 | 
			
		||||
	}
 | 
			
		||||
	config.WaitOnEmpty = true
 | 
			
		||||
 | 
			
		||||
	byteFIFO, err := NewLevelQueueByteFIFO(config.ConnectionString, config.QueueName)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -82,7 +85,7 @@ func NewLevelQueueByteFIFO(connection, prefix string) (*LevelQueueByteFIFO, erro
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// PushFunc will push data into the fifo
 | 
			
		||||
func (fifo *LevelQueueByteFIFO) PushFunc(data []byte, fn func() error) error {
 | 
			
		||||
func (fifo *LevelQueueByteFIFO) PushFunc(ctx context.Context, data []byte, fn func() error) error {
 | 
			
		||||
	if fn != nil {
 | 
			
		||||
		if err := fn(); err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
@@ -92,7 +95,7 @@ func (fifo *LevelQueueByteFIFO) PushFunc(data []byte, fn func() error) error {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Pop pops data from the start of the fifo
 | 
			
		||||
func (fifo *LevelQueueByteFIFO) Pop() ([]byte, error) {
 | 
			
		||||
func (fifo *LevelQueueByteFIFO) Pop(ctx context.Context) ([]byte, error) {
 | 
			
		||||
	data, err := fifo.internal.RPop()
 | 
			
		||||
	if err != nil && err != levelqueue.ErrNotFound {
 | 
			
		||||
		return nil, err
 | 
			
		||||
@@ -108,7 +111,7 @@ func (fifo *LevelQueueByteFIFO) Close() error {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Len returns the length of the fifo
 | 
			
		||||
func (fifo *LevelQueueByteFIFO) Len() int64 {
 | 
			
		||||
func (fifo *LevelQueueByteFIFO) Len(ctx context.Context) int64 {
 | 
			
		||||
	return fifo.internal.Len()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user