- Refine await.

Author: general-kroll-4-life

internal/stackql/execution/mono_valent_execution.go

@@ -78,6 +78,7 @@ type monoValentExecution struct {
 	isReadOnly                 bool //nolint:unused // TODO: build out
 	isSkipResponse             bool
 	isMutation                 bool
+	isAwait                    bool
 }
 
 func NewMonoValentExecutorFactory(
@@ -90,6 +91,7 @@ func NewMonoValentExecutorFactory(
 	stream streaming.MapStream,
 	isSkipResponse bool,
 	isMutation bool,
+	isAwait bool,
 ) MonoValentExecutorFactory {
 	var tcc internaldto.TxnControlCounters
 	if insertCtx != nil {
@@ -110,6 +112,7 @@ func NewMonoValentExecutorFactory(
 		stream:                     stream,
 		isSkipResponse:             isSkipResponse,
 		isMutation:                 isMutation,
+		isAwait:                    isAwait,
 	}
 }
 
@@ -586,6 +589,7 @@ type AgnosticatePayload interface {
 	GetInsertPreparator() InsertPreparator
 	IsSkipResponse() bool
 	IsMutation() bool
+	IsAwait() bool
 }
 
 type httpAgnosticatePayload struct {
@@ -604,6 +608,7 @@ type httpAgnosticatePayload struct {
 	insertPreparator        InsertPreparator
 	isSkipResponse          bool
 	isMutation              bool
+	isAwait                 bool
 }
 
 func newHTTPAgnosticatePayload(
@@ -622,6 +627,7 @@ func newHTTPAgnosticatePayload(
 	insertPreparator InsertPreparator,
 	isSkipResponse bool,
 	isMutation bool,
+	isAwait bool,
 ) AgnosticatePayload {
 	return &httpAgnosticatePayload{
 		tableMeta:               tableMeta,
@@ -639,13 +645,18 @@ func newHTTPAgnosticatePayload(
 		insertPreparator:        insertPreparator,
 		isSkipResponse:          isSkipResponse,
 		isMutation:              isMutation,
+		isAwait:                 isAwait,
 	}
 }
 
 func (ap *httpAgnosticatePayload) GetPolyHandler() PolyHandler {
 	return ap.polyHandler
 }
 
+func (ap *httpAgnosticatePayload) IsAwait() bool {
+	return ap.isAwait
+}
+
 func (ap *httpAgnosticatePayload) GetInsertPreparator() InsertPreparator {
 	return ap.insertPreparator
 }
@@ -748,6 +759,7 @@ func agnosticate(
 	insertPreparator := agPayload.GetInsertPreparator()
 	isSkipResponse := agPayload.IsNilResponseAcceptable()
 	isMutation := agPayload.IsMutation()
+	isAwait := agPayload.IsAwait()
 	// TODO: TCC setup
 	armoury, armouryErr := agPayload.GetArmoury()
 	if armouryErr != nil {
@@ -797,7 +809,7 @@ func agnosticate(
 				insertPreparator,
 				isSkipResponse,
 				false,
-				false,
+				isAwait,
 				false,
 				isMutation,
 				"",
@@ -1193,7 +1205,7 @@ func (sp *standardProcessor) Process() ProcessorResponse {
 		)
 		housekeepingDone = insertPrepResult.IsHousekeepingDone()
 		insertPrepErr, hasInsertPrepErr := insertPrepResult.GetError()
-		if isSkipResponse && isMutation && httpResponse.StatusCode < 300 {
+		if !isAwait && isSkipResponse && isMutation && httpResponse.StatusCode < 300 {
 			return newHTTPProcessorResponse(
 				nil, reversalStream, false, nil,
 			).WithSuccessMessages([]string{"The operation was despatched successfully"})
@@ -1388,6 +1400,7 @@ func (mv *monoValentExecution) GetExecutor() (func(pc primitive.IPrimitiveCtx) i
 				mv,
 				mv.isSkipResponse,
 				mv.isMutation,
+				mv.isAwait,
 			)
 			processorResponse, agnosticErr := agnosticate(agnosticatePayload)
 			if agnosticErr != nil {

internal/stackql/primitivebuilder/delete.go

@@ -80,6 +80,7 @@ func (ss *Delete) Build() error {
 		streaming.NewNopMapStream(),
 		!ss.isAwait,
 		true,
+		ss.isAwait,
 	)
 	ex, exErr := mvb.GetExecutor()
 	if exErr != nil {

internal/stackql/primitivebuilder/mono_valent_builder.go

@@ -72,6 +72,7 @@ func newMonoValentBuilder(
 			stream,
 			isSkipResponse,
 			isMutation,
+			false,
 		),
 	}
 }