refactor(main) flag parsing and validation in init

This commit is contained in:
Guilherme Salazar
2019-12-12 13:35:43 -08:00
parent 8daa26cfda
commit 77057d1f51

14
main.go
View File

@@ -14,6 +14,15 @@ import (
var socket = flag.String("socket", "", "Socket to listen into")
func init() {
flag.Parse()
if *socket == "" {
flag.Usage()
os.Exit(2)
}
}
func runServer(listener net.Listener) {
var handle codec.MsgpackHandle
handle.ReaderBufferSize = 4096
@@ -37,10 +46,6 @@ func runServer(listener net.Listener) {
}
func main() {
flag.Parse()
if *socket != "" {
err := os.Remove(*socket)
if err != nil && !os.IsNotExist(err) {
log.Printf(`removing "%s": %s`, socket, err)
@@ -56,5 +61,4 @@ func main() {
rpc.RegisterName("plugin", newServer())
runServer(listener)
}
}