acepanel/panel
2
0
Fork 0
mirror of https://github.com/acepanel/panel.git synced 2026-02-05 03:22:32 +08:00
Code Issues Packages Releases Activity

feat: 优化队列实现

Browse Source
This commit is contained in:
耗子
2024-10-12 00:58:55 +08:00
parent 3c39027120
commit 0c4d97a512
6 changed files with 82 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
pkg/queue/queue_test.go
View File

@@ -1,6 +1,7 @@
package queue
import (
"context"
"errors"
"testing"
"time"
@@ -17,33 +18,21 @@ func TestQueueTestSuite(t *testing.T) {
}
func (suite *QueueTestSuite) TestQueueInitialization() {
queue := New()
queue := New(10)
suite.NotNil(queue)
suite.NotNil(queue.jobs)
suite.NotNil(queue.isShutdown)
suite.NotNil(queue.done)
}
func (suite *QueueTestSuite) TestPushJobToQueue() {
queue := New()
queue := New(10)
job := &MockJob{}
err := queue.Push(job, []any{"arg1", "arg2"})
suite.NoError(err)
}
func (suite *QueueTestSuite) TestPushJobToShutdownQueue() {
queue := New()
queue.Run()
suite.NoError(queue.Shutdown())
job := &MockJob{}
err := queue.Push(job, []any{"arg1", "arg2"})
suite.Error(err)
suite.EqualError(err, "queue is shutdown, cannot add new jobs")
}
func (suite *QueueTestSuite) TestBulkJobsToQueue() {
queue := New()
jobs := []Jobs{
queue := New(10)
jobs := []JobItem{
{Job: &MockJob{}, Args: []any{"arg1"}},
{Job: &MockJob{}, Args: []any{"arg2"}},
}
@@ -51,80 +40,51 @@ func (suite *QueueTestSuite) TestBulkJobsToQueue() {
suite.NoError(err)
}
func (suite *QueueTestSuite) TestBulkJobsToShutdownQueue() {
queue := New()
queue.Run()
suite.NoError(queue.Shutdown())
jobs := []Jobs{
{Job: &MockJob{}, Args: []any{"arg1"}},
{Job: &MockJob{}, Args: []any{"arg2"}},
}
err := queue.Bulk(jobs)
suite.Error(err)
suite.EqualError(err, "queue is shutdown, cannot add new jobs")
}
func (suite *QueueTestSuite) TestLaterJobExecution() {
queue := New()
job := &MockJob{}
err := queue.Later(1, job, []any{"arg1"})
suite.NoError(err)
}
func (suite *QueueTestSuite) TestLaterJobExecutionOnShutdownQueue() {
queue := New()
queue.Run()
suite.NoError(queue.Shutdown())
queue := New(10)
job := &MockJob{}
err := queue.Later(1, job, []any{"arg1"})
suite.NoError(err)
}
func (suite *QueueTestSuite) TestRunQueue() {
queue := New()
queue := New(10)
job := &MockJob{}
suite.NoError(queue.Push(job, []any{"arg1"}))
queue.Run()
queue.Run(context.Background())
time.Sleep(1 * time.Second)
suite.True(job.Executed)
}
func (suite *QueueTestSuite) TestRunQueueWithLaterJob() {
queue := New()
queue := New(10)
job := &MockJob{}
suite.NoError(queue.Later(1, job, []any{"arg1"}))
queue.Run()
queue.Run(context.Background())
time.Sleep(2 * time.Second)
suite.True(job.Executed)
}
func (suite *QueueTestSuite) TestRunQueueWithBulkJobs() {
queue := New()
jobs := []Jobs{
queue := New(10)
jobs := []JobItem{
{Job: &MockJob{}, Args: []any{"arg1"}},
{Job: &MockJob{}, Args: []any{"arg2"}},
}
suite.NoError(queue.Bulk(jobs))
queue.Run()
queue.Run(context.Background())
time.Sleep(1 * time.Second)
}
func (suite *QueueTestSuite) TestRunQueueWithErrHandle() {
queue := New()
queue := New(10)
job := &MockJob{}
suite.NoError(queue.Push(job, []any{"arg1"}))
queue.Run()
queue.Run(context.Background())
time.Sleep(1 * time.Second)
suite.Error(job.Err)
}
func (suite *QueueTestSuite) TestShutdownQueue() {
queue := New()
queue.Run()
err := queue.Shutdown()
suite.NoError(err)
}
type MockJob struct {
Executed bool
Err error