Kaynağa Gözat

Add Flag and Config Parameter Handling

pull/294/head
stefanbenten 4 yıl önce
ebeveyn
işleme
7f4a7870a0
2 değiştirilmiş dosya ile 16 ekleme ve 0 silme
  1. +10
    -0
      cmd/cmd.go
  2. +6
    -0
      server/server.go

+ 10
- 0
cmd/cmd.go Dosyayı Görüntüle

@@ -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 Dosyayı Görüntüle

@@ -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


Yükleniyor…
İptal
Kaydet