瀏覽代碼

Add Flag and Config Parameter Handling

pull/294/head
stefanbenten 4 年之前
父節點
當前提交
7f4a7870a0
共有 2 個文件被更改,包括 16 次插入0 次删除
  1. +10
    -0
      cmd/cmd.go
  2. +6
    -0
      server/server.go

+ 10
- 0
cmd/cmd.go 查看文件

@@ -91,6 +91,11 @@ var globalFlags = []cli.Flag{
Usage: "key for user voice (front end)",
Value: "",
},
cli.IntFlag{
Name: "lifetime",
Usage: "default file lifetime",
Value: 14,
},
cli.StringFlag{
Name: "provider",
Usage: "s3|gdrive|local",
@@ -334,6 +339,11 @@ func New() *Cmd {
if applyIPFilter {
options = append(options, server.FilterOptions(ipFilterOptions))
}
if lifetime := c.Int("lifetime"); lifetime > 0 {
server.LifeTime(lifetime)
} else {
panic("lifetime not greater than 0")
}

switch provider := c.String("provider"); provider {
case "s3":


+ 6
- 0
server/server.go 查看文件

@@ -177,6 +177,12 @@ func EnableProfiler() OptionFn {
}
}

func LifeTime(lifetime int) OptionFn {
return func(srvr *Server) {
srvr.lifetime = time.Hour * 24 * time.Duration(lifetime)
}
}

func UseStorage(s Storage) OptionFn {
return func(srvr *Server) {
srvr.storage = s


Loading…
取消
儲存