Method expandInsertStmt

internal/x/expander/expander.go:208–237

expandInsertStmt expands * in an INSERT statement's RETURNING clause

(ctx context.Context, stmt *ast.InsertStmt)

// expandInsertStmt expands * in an INSERT statement's RETURNING clause
func (e *Expander) expandInsertStmt(ctx context.Context, stmt *ast.InsertStmt) error {
	// Expand CTEs first
	if stmt.WithClause != nil && stmt.WithClause.Ctes != nil {
		for _, cte := range stmt.WithClause.Ctes.Items {
			if err := e.expandNode(ctx, cte); err != nil {
				return err
			}
		}
	}

	// Expand the SELECT part if present
	if stmt.SelectStmt != nil {
		if err := e.expandNode(ctx, stmt.SelectStmt); err != nil {
			return err
		}
	}

	// Expand RETURNING clause
	if hasStarInList(stmt.ReturningList) {
		tempRaw := &ast.RawStmt{Stmt: stmt}
		tempQuery := ast.Format(tempRaw, e.dialect)
		columns, err := e.getColumnNames(ctx, tempQuery)
		if err != nil {
			return fmt.Errorf("failed to get column names: %w", err)
		}
		stmt.ReturningList = rewriteTargetList(stmt.ReturningList, columns)
	}

	return nil
}

// expandUpdateStmt expands * in an UPDATE statement's RETURNING clause
func (e *Expander) expandUpdateStmt(ctx context.Context, stmt *ast.UpdateStmt) error {