Skip to content

Commit

Permalink
extract setupBackend()
Browse files Browse the repository at this point in the history
  • Loading branch information
thraxil committed Jul 28, 2017
1 parent bcb6b27 commit dc836a2
Showing 1 changed file with 23 additions and 16 deletions.
39 changes: 23 additions & 16 deletions cask.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,22 +56,9 @@ func main() {
}
log.SetPrefix(c.UUID[:8] + " ")
n := newNode(c.UUID, c.BaseURL, c.Writeable)
var backend backend
if c.Backend == "disk" {
backend = newDiskBackend(c.DiskBackendRoot)
} else if c.Backend == "s3" {
if c.S3AccessKey == "" || c.S3SecretKey == "" || c.S3Bucket == "" {
log.Fatal("need S3 ACCESS_KEY, SECRET_KEY, and bucket all configured")
} else {
backend = newS3Backend(c.S3AccessKey, c.S3SecretKey, c.S3Bucket)
}
} else if c.Backend == "dropbox" {
if c.DBAccessKey == "" || c.DBSecretKey == "" {
log.Fatal("need dropbox ACCESS_KEY and SECRET_KEY")
} else {
backend = newDropboxBackend(c.DBAccessKey, c.DBSecretKey, c.DBToken)
}
}

backend := setupBackend(c)

if c.MaxProcs > 0 {
log.Printf("max procs: %d\n", c.MaxProcs)
runtime.GOMAXPROCS(c.MaxProcs)
Expand Down Expand Up @@ -126,3 +113,23 @@ func main() {
log.Fatal(server.ListenAndServe())
}
}

func setupBackend(c config) backend {
var backend backend
if c.Backend == "disk" {
backend = newDiskBackend(c.DiskBackendRoot)
} else if c.Backend == "s3" {
if c.S3AccessKey == "" || c.S3SecretKey == "" || c.S3Bucket == "" {
log.Fatal("need S3 ACCESS_KEY, SECRET_KEY, and bucket all configured")
} else {
backend = newS3Backend(c.S3AccessKey, c.S3SecretKey, c.S3Bucket)
}
} else if c.Backend == "dropbox" {
if c.DBAccessKey == "" || c.DBSecretKey == "" {
log.Fatal("need dropbox ACCESS_KEY and SECRET_KEY")
} else {
backend = newDropboxBackend(c.DBAccessKey, c.DBSecretKey, c.DBToken)
}
}
return backend
}

0 comments on commit dc836a2

Please sign in to comment.