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