Skip to content

Commit fc9fc7a

Browse files
wxiaoguangGiteaBot
authored andcommitted
Calculate MAX_WORKERS default value by CPU number (go-gitea#26177)
To avoid consuming user's 100% CPU, limit the default value of MAX_WORKERS Fix go-gitea#26063 (the CPU 100% problem mentioned in it)
1 parent 666038a commit fc9fc7a

File tree

4 files changed

+25
-18
lines changed

4 files changed

+25
-18
lines changed

custom/conf/app.example.ini

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1414,8 +1414,8 @@ LEVEL = Info
14141414
;; Provides the suffix of the default redis/disk unique queue set name - specific queues can be overridden within in their [queue.name] sections.
14151415
;SET_NAME = "_unique"
14161416
;;
1417-
;; Dynamically scale the worker pool to at this many workers
1418-
;MAX_WORKERS = 10
1417+
;; Maximum number of worker go-routines for the queue. Default value is "CpuNum/2" clipped to between 1 and 10.
1418+
;MAX_WORKERS = ; (dynamic)
14191419

14201420
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
14211421
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

docs/content/administration/config-cheat-sheet.en-us.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -485,7 +485,7 @@ Configuration at `[queue]` will set defaults for queues with overrides for indiv
485485
- `CONN_STR`: **redis://127.0.0.1:6379/0**: Connection string for the redis queue type. For `redis-cluster` use `redis+cluster://127.0.0.1:6379/0`. Options can be set using query params. Similarly, LevelDB options can also be set using: **leveldb://relative/path?option=value** or **leveldb:///absolute/path?option=value**, and will override `DATADIR`
486486
- `QUEUE_NAME`: **_queue**: The suffix for default redis and disk queue name. Individual queues will default to **`name`**`QUEUE_NAME` but can be overridden in the specific `queue.name` section.
487487
- `SET_NAME`: **_unique**: The suffix that will be added to the default redis and disk queue `set` name for unique queues. Individual queues will default to **`name`**`QUEUE_NAME`_`SET_NAME`_ but can be overridden in the specific `queue.name` section.
488-
- `MAX_WORKERS`: **10**: Maximum number of worker go-routines for the queue.
488+
- `MAX_WORKERS`: **(dynamic)**: Maximum number of worker go-routines for the queue. Default value is "CpuNum/2" clipped to between 1 and 10.
489489

490490
Gitea creates the following non-unique queues:
491491

modules/queue/manager_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ CONN_STR = redis://
5151
assert.Equal(t, "", q.baseConfig.ConnStr)
5252
assert.Equal(t, "default_queue", q.baseConfig.QueueFullName)
5353
assert.Equal(t, "default_queue_unique", q.baseConfig.SetFullName)
54-
assert.Equal(t, 10, q.GetWorkerMaxNumber())
54+
assert.NotZero(t, q.GetWorkerMaxNumber())
5555
assert.Equal(t, 0, q.GetWorkerNumber())
5656
assert.Equal(t, 0, q.GetWorkerActiveNumber())
5757
assert.Equal(t, 0, q.GetQueueItemNumber())
@@ -75,7 +75,7 @@ BATCH_LENGTH = 22
7575
CONN_STR =
7676
QUEUE_NAME = _q2
7777
SET_NAME = _u2
78-
MAX_WORKERS = 2
78+
MAX_WORKERS = 123
7979
`)
8080

8181
assert.NoError(t, err)
@@ -89,7 +89,7 @@ MAX_WORKERS = 2
8989
assert.Equal(t, "addrs=127.0.0.1:6379 db=0", q1.baseConfig.ConnStr)
9090
assert.Equal(t, "no-such_queue1", q1.baseConfig.QueueFullName)
9191
assert.Equal(t, "no-such_queue1_unique", q1.baseConfig.SetFullName)
92-
assert.Equal(t, 10, q1.GetWorkerMaxNumber())
92+
assert.NotZero(t, q1.GetWorkerMaxNumber())
9393
assert.Equal(t, 0, q1.GetWorkerNumber())
9494
assert.Equal(t, 0, q1.GetWorkerActiveNumber())
9595
assert.Equal(t, 0, q1.GetQueueItemNumber())
@@ -105,7 +105,7 @@ MAX_WORKERS = 2
105105
assert.Equal(t, "", q2.baseConfig.ConnStr)
106106
assert.Equal(t, "sub_q2", q2.baseConfig.QueueFullName)
107107
assert.Equal(t, "sub_q2_u2", q2.baseConfig.SetFullName)
108-
assert.Equal(t, 2, q2.GetWorkerMaxNumber())
108+
assert.Equal(t, 123, q2.GetWorkerMaxNumber())
109109
assert.Equal(t, 0, q2.GetWorkerNumber())
110110
assert.Equal(t, 0, q2.GetWorkerActiveNumber())
111111
assert.Equal(t, 0, q2.GetQueueItemNumber())

modules/setting/queue.go

Lines changed: 18 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ package setting
55

66
import (
77
"path/filepath"
8+
"runtime"
89

910
"code.gitea.io/gitea/modules/json"
1011
"code.gitea.io/gitea/modules/log"
@@ -25,18 +26,24 @@ type QueueSettings struct {
2526
MaxWorkers int
2627
}
2728

28-
var queueSettingsDefault = QueueSettings{
29-
Type: "level", // dummy, channel, level, redis
30-
Datadir: "queues/common", // relative to AppDataPath
31-
Length: 100, // queue length before a channel queue will block
32-
33-
QueueName: "_queue",
34-
SetName: "_unique",
35-
BatchLength: 20,
36-
MaxWorkers: 10,
37-
}
38-
3929
func GetQueueSettings(rootCfg ConfigProvider, name string) (QueueSettings, error) {
30+
queueSettingsDefault := QueueSettings{
31+
Type: "level", // dummy, channel, level, redis
32+
Datadir: "queues/common", // relative to AppDataPath
33+
Length: 100, // queue length before a channel queue will block
34+
35+
QueueName: "_queue",
36+
SetName: "_unique",
37+
BatchLength: 20,
38+
MaxWorkers: runtime.NumCPU() / 2,
39+
}
40+
if queueSettingsDefault.MaxWorkers < 1 {
41+
queueSettingsDefault.MaxWorkers = 1
42+
}
43+
if queueSettingsDefault.MaxWorkers > 10 {
44+
queueSettingsDefault.MaxWorkers = 10
45+
}
46+
4047
// deep copy default settings
4148
cfg := QueueSettings{}
4249
if cfgBs, err := json.Marshal(queueSettingsDefault); err != nil {

0 commit comments

Comments
 (0)