Просмотр исходного кода

Add Flag and Config Parameter Handling

pull/294/head
stefanbenten 4 лет назад
Родитель
Сommit
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


Загрузка…
Отмена
Сохранить