Fix go routine leak

googleforgames · Jun 28, 2024 · 1b0822e · 1b0822e
1 parent 262b274
commit 1b0822e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/frontend/frontend.go b/frontend/frontend.go
@@ -119,14 +119,19 @@ func streamAssignments(ctx context.Context, assignments chan *pb.Assignment, err
 		errs <- fmt.Errorf("error getting assignment stream: %w", err)
 		return
 	}
-	for {
+
+	var assignment *pb.Assignment
+	for assignment.GetConnection() == "" {
 		resp, err := stream.Recv()
 		if err != nil {
 			errs <- fmt.Errorf("error streaming assignment: %w", err)
 			return
 		}
-		assignments <- resp.Assignment
+		assignment = resp.Assignment
 	}
+
+	assignments <- assignment
+	log.Printf("Got assignment: %v", assignment)
 }
 
 func connectFrontendServer() (*grpc.ClientConn, error) {