Skip to content

Commit cc6473f

Browse files
committed
优化代码
1 parent 5efc92f commit cc6473f

File tree

7 files changed

+18
-8
lines changed

7 files changed

+18
-8
lines changed

load.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ func LoadArgsMain() (unit Unit, ok bool, err error) {
8484
unit = Unit{
8585
Name: "arg-main",
8686
Group: DefaultGroup,
87-
Kind: "daemon",
87+
Kind: KindDaemon,
8888
ExecuteOptions: ExecuteOptions{
8989
Command: args,
9090
},
@@ -106,9 +106,9 @@ func LoadEnvMain() (unit Unit, ok bool, err error) {
106106
if group == "" {
107107
group = DefaultGroup
108108
}
109-
kind := "daemon"
109+
kind := KindDaemon
110110
if once, _ := strconv.ParseBool(strings.TrimSpace(os.Getenv("MINIT_MAIN_ONCE"))); once {
111-
kind = "once"
111+
kind = KindOnce
112112
return
113113
}
114114
var command []string

runner.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,31 +20,31 @@ type RunnerFactory struct {
2020

2121
var (
2222
RunnerFactories = map[string]*RunnerFactory{
23-
"render": {
23+
KindRender: {
2424
Level: RunnerL1,
2525
Create: func(unit Unit, logger *mlog.Logger) (Runner, error) {
2626
return NewRenderRunner(unit, logger)
2727
},
2828
},
29-
"once": {
29+
KindOnce: {
3030
Level: RunnerL2,
3131
Create: func(unit Unit, logger *mlog.Logger) (Runner, error) {
3232
return NewOnceRunner(unit, logger)
3333
},
3434
},
35-
"daemon": {
35+
KindDaemon: {
3636
Level: RunnerL3,
3737
Create: func(unit Unit, logger *mlog.Logger) (Runner, error) {
3838
return NewDaemonRunner(unit, logger)
3939
},
4040
},
41-
"cron": {
41+
KindCron: {
4242
Level: RunnerL3,
4343
Create: func(unit Unit, logger *mlog.Logger) (Runner, error) {
4444
return NewCronRunner(unit, logger)
4545
},
4646
},
47-
"logrotate": {
47+
KindLogrotate: {
4848
Level: RunnerL3,
4949
Create: func(unit Unit, logger *mlog.Logger) (Runner, error) {
5050
return NewLogrotateRunner(unit, logger)

runner_cron.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ import (
77
"github.com/robfig/cron/v3"
88
)
99

10+
const KindCron = "cron"
11+
1012
type CronRunner struct {
1113
Unit
1214
logger *mlog.Logger

runner_daemon.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ import (
77
"time"
88
)
99

10+
const KindDaemon = "daemon"
11+
1012
type DaemonRunner struct {
1113
Unit
1214
logger *mlog.Logger

runner_logrotate.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ import (
1414
"time"
1515
)
1616

17+
const KindLogrotate = "logrotate"
18+
1719
// filename mark
1820
// daily: FILENAME.ROT2020-06-02.EXT
1921
// filesize: FILENAME.ROT000000000001.EXT (%012d)

runner_once.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@ import (
66
"github.com/acicn/minit/pkg/mlog"
77
)
88

9+
const KindOnce = "once"
10+
911
type OnceRunner struct {
1012
Unit
1113
logger *mlog.Logger

runner_render.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ import (
1313
"text/template"
1414
)
1515

16+
const KindRender = "render"
17+
1618
type RenderRunner struct {
1719
Unit
1820
logger *mlog.Logger

0 commit comments

Comments
 (0)