added config flag
This commit is contained in:
9
main.go
9
main.go
@@ -35,6 +35,7 @@ type Conf struct {
|
||||
var (
|
||||
conf *Conf
|
||||
journalLog = flag.Bool("journal", false, "Log to systemd journal instead of stdout")
|
||||
configFlag = flag.String("config", "config.yaml", "Set config file to use")
|
||||
reBandwidth = regexp.MustCompile(`(?m)bandwidth\s(\d+)Mbit`)
|
||||
)
|
||||
|
||||
@@ -164,12 +165,12 @@ func (m UploadManager) pingWorker() error {
|
||||
}
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
|
||||
killSignals := make(chan os.Signal, 1)
|
||||
signal.Notify(killSignals, syscall.SIGINT, syscall.SIGTERM)
|
||||
reloadSignals := make(chan os.Signal, 1)
|
||||
signal.Notify(reloadSignals, syscall.SIGUSR1)
|
||||
flag.Parse()
|
||||
confStr, err := os.ReadFile("config.yaml")
|
||||
|
||||
confStr, err := os.ReadFile(*configFlag)
|
||||
if err != nil {
|
||||
log.Fatalf("Error reading config file: %v", err)
|
||||
}
|
||||
|
Reference in New Issue
Block a user