Revert all changes to ExtendedQueryStateMachine

Author: MahdiBM

Sources/PostgresNIO/New/Connection State Machine/ExtendedQueryStateMachine.swift

@@ -29,8 +29,7 @@ struct ExtendedQueryStateMachine {
         case sendParseDescribeBindExecuteSync(PostgresQuery)
         case sendParseDescribeSync(name: String, query: String, bindingDataTypes: [PostgresDataType])
         case sendBindExecuteSync(PSQLExecuteStatement)
-        case sendQuery(String)
-
+        
         // --- general actions
         case failQuery(EventLoopPromise<PSQLRowStream>, with: PSQLError)
         case succeedQuery(EventLoopPromise<PSQLRowStream>, with: QueryResult)
@@ -176,12 +175,7 @@ struct ExtendedQueryStateMachine {
     }
 
     mutating func rowDescriptionReceived(_ rowDescription: RowDescription) -> Action {
-        let queryContext: ExtendedQueryContext
-        switch self.state {
-        case .messagesSent(let extendedQueryContext),
-             .parameterDescriptionReceived(let extendedQueryContext):
-            queryContext = extendedQueryContext
-        default:
+        guard case .parameterDescriptionReceived(let queryContext) = self.state else {
             return self.setAndFireError(.unexpectedBackendMessage(.rowDescription(rowDescription)))
         }
 
@@ -270,7 +264,7 @@ struct ExtendedQueryStateMachine {
             }
             // we ignore all rows and wait for readyForQuery
             return .wait
-
+            
         case .initialized,
              .messagesSent,
              .parseCompleteReceived,
@@ -301,7 +295,7 @@ struct ExtendedQueryStateMachine {
             case .prepareStatement:
                 preconditionFailure("Invalid state: \(self.state)")
             }
-
+            
         case .streaming(_, var demandStateMachine):
             return self.avoidingStateMachineCoW { state -> Action in
                 state = .commandComplete(commandTag: commandTag)
@@ -312,14 +306,14 @@ struct ExtendedQueryStateMachine {
             precondition(self.isCancelled)
             self.state = .commandComplete(commandTag: commandTag)
             return .wait
-
+        
         case .initialized,
+             .messagesSent,
              .parseCompleteReceived,
              .parameterDescriptionReceived,
              .noDataMessageReceived,
              .emptyQueryResponseReceived,
              .rowDescriptionReceived,
-             .messagesSent,
              .commandComplete,
              .error:
             return self.setAndFireError(.unexpectedBackendMessage(.commandComplete(commandTag)))
@@ -329,21 +323,20 @@ struct ExtendedQueryStateMachine {
     }
 
     mutating func emptyQueryResponseReceived() -> Action {
-        switch self.state {
-        case .bindCompleteReceived(let queryContext):
-            switch queryContext.query {
-            case .unnamed(_, let eventLoopPromise),
-                 .executeStatement(_, let eventLoopPromise):
-                return self.avoidingStateMachineCoW { state -> Action in
-                    state = .emptyQueryResponseReceived
-                    let result = QueryResult(value: .noRows(.emptyResponse), logger: queryContext.logger)
-                    return .succeedQuery(eventLoopPromise, with: result)
-                }
+        guard case .bindCompleteReceived(let queryContext) = self.state else {
+            return self.setAndFireError(.unexpectedBackendMessage(.emptyQueryResponse))
+        }
 
-            case .prepareStatement:
-                return self.setAndFireError(.unexpectedBackendMessage(.emptyQueryResponse))
+        switch queryContext.query {
+        case .unnamed(_, let eventLoopPromise),
+             .executeStatement(_, let eventLoopPromise):
+            return self.avoidingStateMachineCoW { state -> Action in
+                state = .emptyQueryResponseReceived
+                let result = QueryResult(value: .noRows(.emptyResponse), logger: queryContext.logger)
+                return .succeedQuery(eventLoopPromise, with: result)
             }
-        default:
+
+        case .prepareStatement(_, _, _, _):
             return self.setAndFireError(.unexpectedBackendMessage(.emptyQueryResponse))
         }
     }