server.go

@@ -30,6 +30,14 @@ func Listen(socketPath string, onNewClient func(client net.Conn)) (*Server, erro
 		return nil, err
 	}
 
+	if onNewClient == nil {
+		onNewClient = func(client net.Conn) {
+			data := make([]byte, 1024)
+			for _, err := client.Read(data); err == nil; _, err = client.Read(data) {
+			}
+		}
+	}
+
 	s := &Server{
 		socketPath:  absPath,
 		server:      server,
@@ -62,14 +70,7 @@ func (s *Server) handleClients() {
 func (s *Server) handleClient(client net.Conn) {
 	s.clients.Put(client, struct{}{})
 	defer s.clients.Delete(client)
-
+	s.onNewClient(client)
-	if s.onNewClient != nil {
-		s.onNewClient(client)
-	}
-
-	data := make([]byte, 1024)
-	for _, err := client.Read(data); err == nil; _, err = client.Read(data) {
-	}
 }
 
 func (s *Server) Close() error {