remove unnecessary sleeps

pkg/lock/lock_table.go

@@ -29,14 +29,12 @@ func (t *Table) Lock(ctx context.Context, name string, clientId uint64) bool {
 	result := false
 
 	wg := &sync.WaitGroup{}
-	isRunning := false
+
+	wg.Add(1)
 
 	go func() {
-		wg.Add(1)
 		defer wg.Done()
 
-		isRunning = true
-
 		for {
 			select {
 			case <-ctx.Done():
@@ -52,10 +50,6 @@ func (t *Table) Lock(ctx context.Context, name string, clientId uint64) bool {
 		}
 	}()
 
-	for !isRunning {
-		<-time.After(time.Millisecond)
-	}
-
 	wg.Wait()
 
 	return result

pkg/server/server.go

@@ -7,7 +7,6 @@ import (
 	"log"
 	"net"
 	"sync"
-	"time"
 )
 
 type Server struct {
@@ -54,7 +53,6 @@ func New(ctx context.Context, opts ...Option) (*Server, error) {
 }
 
 func (s *Server) acceptor() {
-	s.wg.Add(1)
 	defer s.wg.Done()
 
 	for {
@@ -104,16 +102,17 @@ func (s *Server) wait() {
 }
 
 func (s *Server) Run() {
+	s.wg.Add(1)
+
 	go s.startCommandListener()
 
 	for i := 0; i < s.options.acceptorCount; i++ {
+		s.wg.Add(1)
 		go s.acceptor()
 	}
 
 	log.Println("Ready to serve connections")
 
-	<-time.After(time.Millisecond)
-
 	s.wait()
 }
 

pkg/server/server_command.go

@@ -58,7 +58,6 @@ func (s *Server) handleCommand(cmd *client.Command) {
 }
 
 func (s *Server) startCommandListener() {
-	s.wg.Add(1)
 	defer s.wg.Done()
 
 	for {

test/integration/server_lock_test.go

@@ -210,9 +210,9 @@ func TestLockWithTimeoutAndConcurrency(t *testing.T) {
 
 	wg2 := &sync.WaitGroup{}
 
-	go func() {
-		wg2.Add(1)
+	wg2.Add(1)
 
+	go func() {
 		conn1 := connect(t, serverAddr)
 
 		defer func(conn *net.TCPConn) {
@@ -247,9 +247,9 @@ func TestLockWithTimeoutAndConcurrency(t *testing.T) {
 		time.Sleep(time.Millisecond * 250)
 	}()
 
-	go func() {
-		wg2.Add(1)
+	wg2.Add(1)
 
+	go func() {
 		conn1 := connect(t, serverAddr)
 
 		defer func(conn *net.TCPConn) {
@@ -282,7 +282,5 @@ func TestLockWithTimeoutAndConcurrency(t *testing.T) {
 		}
 	}()
 
-	time.Sleep(time.Millisecond)
-
 	wg2.Wait()
 }