Improved cassandra checks

Author: iluhinsky

.github/workflows/go.yml

@@ -111,19 +111,28 @@ jobs:
         sudo apt-get update
         sudo apt-get install -y postgresql-client
         
-        # Install Cassandra client
-        sudo apt-get install -y curl
+        # Install Cassandra client and netcat
+        sudo apt-get install -y curl netcat
         curl -fsSL https://www.apache.org/dist/cassandra/KEYS | sudo gpg --dearmor -o /usr/share/keyrings/cassandra-archive-keyring.gpg
         echo "deb [signed-by=/usr/share/keyrings/cassandra-archive-keyring.gpg] https://apache.jfrog.io/artifactory/cassandra-deb/ 40x main" | sudo tee -a /etc/apt/sources.list.d/cassandra.sources.list
         sudo apt-get update
         sudo apt-get install -y cassandra-tools
       
     - name: Create Cassandra keyspace
       run: |
+        # First wait for the port to be open
+        until nc -z localhost 9042; do
+          echo "Waiting for Cassandra port to be ready..."
+          sleep 5
+        done
+        
+        # Then wait for Cassandra to be fully initialized
         until cqlsh -u cassandra -p cassandra -e 'describe keyspaces' localhost 9042 > /dev/null 2>&1; do
           echo "Waiting for Cassandra to be ready..."
           sleep 5
         done
+        
+        echo "Creating Cassandra keyspace..."
         cqlsh -u cassandra -p cassandra -e "CREATE KEYSPACE IF NOT EXISTS perfkit_db_ci WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1};" localhost 9042
 
     - name: Create vector extension