mirror of
https://github.com/Kong/go-pluginserver.git
synced 2025-10-16 05:00:44 +08:00
refactor(main) flag parsing and validation in init
This commit is contained in:
14
main.go
14
main.go
@@ -14,6 +14,15 @@ import (
|
|||||||
|
|
||||||
var socket = flag.String("socket", "", "Socket to listen into")
|
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) {
|
func runServer(listener net.Listener) {
|
||||||
var handle codec.MsgpackHandle
|
var handle codec.MsgpackHandle
|
||||||
handle.ReaderBufferSize = 4096
|
handle.ReaderBufferSize = 4096
|
||||||
@@ -37,10 +46,6 @@ func runServer(listener net.Listener) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
if *socket != "" {
|
|
||||||
|
|
||||||
err := os.Remove(*socket)
|
err := os.Remove(*socket)
|
||||||
if err != nil && !os.IsNotExist(err) {
|
if err != nil && !os.IsNotExist(err) {
|
||||||
log.Printf(`removing "%s": %s`, socket, err)
|
log.Printf(`removing "%s": %s`, socket, err)
|
||||||
@@ -57,4 +62,3 @@ func main() {
|
|||||||
|
|
||||||
runServer(listener)
|
runServer(listener)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
Reference in New Issue
Block a user