worker: add -config flag

Useful when invoking "go run" from the parent dir, e.g. so that
the worker can access the local images/ dir.
This commit is contained in:
Simon Ser 2022-07-08 19:38:56 +00:00 committed by Drew DeVault
parent 7f1998a01a
commit 189cebe60b
1 changed files with 4 additions and 1 deletions

View File

@ -28,12 +28,15 @@ var (
)
func main() {
var configPath string
flag.IntVar(&workers, "workers", runtime.NumCPU(),
"configure number of workers")
flag.StringVar(&configPath, "config", "../config.ini",
"path to config.ini file")
flag.Parse()
var err error
for _, path := range []string{"../config.ini", "/etc/sr.ht/config.ini"} {
for _, path := range []string{configPath, "/etc/sr.ht/config.ini"} {
config, err = ini.LoadFile(path)
if err == nil {
break