Update execute.ts

issue-:graphql#4001

Author: xonx4l

Diff for: src/execution/execute.ts

@@ -1781,18 +1781,48 @@ export function createSourceEventStream(
 function createSourceEventStreamImpl(
   exeContext: ExecutionContext,
 ): PromiseOrValue<AsyncIterable<unknown> | ExecutionResult> {
-  try {
-    const eventStream = executeSubscription(exeContext);
-    if (isPromise(eventStream)) {
-      return eventStream.then(undefined, (error) => ({ errors: [error] }));
+try {
+
+  const eventStream = executeSubscription(exeContext);
+
+  if (isPromise(eventStream)) {
+    // Handle promise errors
+    return eventStream.then(
+      data => ({ data }), 
+      error => ({ errors: [error] })
+    );
+  }
+
+  const wrappedStream = {
+    async *[Symbol.asyncIterator]() {
+      try {
+        yield* eventStream;
+      } catch (error) {
+        yield { errors: [error] };
+      }
     }
+  };
 
-    return eventStream;
-  } catch (error) {
-    return { errors: [error] };
+  return wrappedStream;
+
+} catch (error) {
+  // Handle sync errors
+  return { errors: [error] }; 
+}
+
+function resolve(payload) {
+  if (payload.errors) {
+    throw payload.errors[0]; 
+  } else {
+    return payload.data;
   }
 }
 
+const stream = executeSubscription();
+mapAsyncIterable(stream, resolve); 
+
+}
+
 function executeSubscription(
   exeContext: ExecutionContext,
 ): PromiseOrValue<AsyncIterable<unknown>> {