- Interface change.

Author: general-kroll-4-life

.vscode/launch.json

@@ -182,6 +182,7 @@
         "replace /*+ AWAIT */ google.compute.firewalls set data__disabled = 'true'  where project = 'mutable-project'  and firewall = 'replacable-firewall'  returning *;",
         "delete from aws.transfer.servers  where  data__ServerId = 's-0000000001'  and region = 'ap-southeast-2';",
         "replace google.compute.firewalls set data__disabled = 'true'  where project = 'mutable-project'  and firewall = 'replacable-firewall'  returning *;",
+        "replace /*+ AWAIT */ google.compute.firewalls set data__disabled = 'true'  where project = 'mutable-project'  and firewall = 'replacable-firewall'  returning *;",
       ], 
       "default": "show providers;"
     },

internal/stackql/dependencyplanner/dependencyplanner.go

@@ -408,8 +408,7 @@ func (dp *standardDependencyPlanner) processOrphan(
 	} else {
 		// Persist SQL mirror table here prior to generating insert DML
 		drmCfg := dp.handlerCtx.GetDrmConfig()
-		// TODO: extract isAwait somehow
-		ddl, ddlErr := drmCfg.GenerateDDL(anTab, nil, nil, nil, opStore, false, 0, false, false)
+		ddl, ddlErr := drmCfg.GenerateDDL(anTab, nil, nil, nil, opStore, annotationCtx.IsAwait(), 0, false, false)
 		if ddlErr != nil {
 			return nil, nil, ddlErr
 		}
@@ -418,7 +417,13 @@ func (dp *standardDependencyPlanner) processOrphan(
 			return nil, nil, err
 		}
 	}
-	insPsc, err := dp.primitiveComposer.GetDRMConfig().GenerateInsertDML(anTab, opStore, tcc, false)
+	insPsc, err := dp.primitiveComposer.GetDRMConfig().GenerateInsertDML(
+		anTab,
+		opStore,
+		tcc,
+		false,
+		annotationCtx.IsAwait(),
+	)
 	return insPsc, tcc, err
 }
 

internal/stackql/drm/drm_cfg.go

@@ -62,12 +62,14 @@ type Config interface {
 		anysdk.OperationStore,
 		internaldto.TxnControlCounters,
 		bool,
+		bool,
 	) (PreparedStatementCtx, error)
 	GenerateSelectDML(
 		util.AnnotatedTabulation,
 		internaldto.TxnControlCounters,
 		string,
 		string,
+		bool,
 	) (PreparedStatementCtx, error)
 	ExecuteInsertDML(sqlengine.SQLEngine, PreparedStatementCtx, map[string]interface{}, string) (sql.Result, error)
 	OpenapiColumnsToRelationalColumns(cols []anysdk.ColumnDescriptor) []typing.RelationalColumn
@@ -542,6 +544,7 @@ func (dc *staticDRMConfig) GenerateInsertDML(
 	method anysdk.OperationStore,
 	tcc internaldto.TxnControlCounters,
 	isNilResponseAlloed bool,
+	isAsync bool,
 ) (PreparedStatementCtx, error) {
 	var columns []typing.ColumnMetadata
 	_, isSQLDataSource := tabAnnotated.GetSQLDataSource()
@@ -636,6 +639,7 @@ func (dc *staticDRMConfig) GenerateSelectDML(
 	txnCtrlCtrs internaldto.TxnControlCounters,
 	selectSuffix,
 	rewrittenWhere string,
+	isAwait bool,
 ) (PreparedStatementCtx, error) {
 	var quotedColNames []string
 	var columns []typing.ColumnMetadata

internal/stackql/primitivegenerator/unary_selection.go

@@ -73,7 +73,12 @@ func (pb *standardPrimitiveGenerator) assembleUnarySelectionBuilder(
 	}
 	ctrs := pbi.GetTxnCtrlCtrs()
 	insPsc, err := pb.PrimitiveComposer.GetDRMConfig().GenerateInsertDML(
-		annotatedInsertTabulation, method, ctrs, methodAnalysisOutput.IsNilResponseAllowed())
+		annotatedInsertTabulation,
+		method,
+		ctrs,
+		methodAnalysisOutput.IsNilResponseAllowed(),
+		methodAnalysisOutput.IsAwait(),
+	)
 	if err != nil {
 		return err
 	}
@@ -117,6 +122,7 @@ func (pb *standardPrimitiveGenerator) assembleUnarySelectionBuilder(
 			handlerCtx.GetSQLSystem(),
 			handlerCtx.GetASTFormatter(),
 			handlerCtx.GetNamespaceCollection()),
+		methodAnalysisOutput.IsAwait(),
 	)
 	if err != nil {
 		return err
@@ -233,13 +239,6 @@ func (pb *standardPrimitiveGenerator) analyzeUnaryAction(
 	hIDs := internaldto.NewHeirarchyIdentifiers(
 		rawhIDs.GetProviderStr(), rawhIDs.GetServiceStr(), itemSchemaName, strings.ToLower(publishedMethodKey))
 
-	// annotatedInsertTabulation := util.NewAnnotatedTabulation(insertTabulation, hIDs, inputTableName, "")
-
-	// ctrs := pbi.GetTxnCtrlCtrs()
-	// insPsc, err := pb.PrimitiveComposer.GetDRMConfig().GenerateInsertDML(annotatedInsertTabulation, method, ctrs)
-	// if err != nil {
-	// 	return err
-	// }
 	schema, _, err := tbl.GetResponseSchemaAndMediaType()
 	if err != nil && !methodAnalysisOutput.IsNilResponseAllowed() {
 		return err