remove membucket query

klothoplatform · Feb 10, 2023 · 21909d4 · 21909d4
1 parent 3113a41
commit 21909d4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 20 deletions.
diff --git a/pkg/lang/golang/plugin_fs.go b/pkg/lang/golang/plugin_fs.go
@@ -99,10 +99,16 @@ func (p *PersistFsPlugin) transformFS(f *core.SourceFile, cap *core.Annotation,
 
 	newNodeContent := strings.Replace(result.args.Content(), result.args.Content(), ArgumentListToString(args), 1)
 
-	f.ReplaceNodeContent(result.args, newNodeContent)
-	f.ReplaceNodeContent(result.operator, "blob")
+	err := f.ReplaceNodeContent(result.args, newNodeContent)
+	if err != nil {
+		return nil, err
+	}
+	err = f.ReplaceNodeContent(result.operator, "blob")
+	if err != nil {
+		return nil, err
+	}
 
-	err := UpdateImportsInFile(f, p.runtime.GetFsImports(), []Import{{Package: "gocloud.dev/blob/fileblob"}})
+	err = UpdateImportsInFile(f, p.runtime.GetFsImports(), []Import{{Package: "gocloud.dev/blob/fileblob"}})
 	if err != nil {
 		return nil, err
 	}

diff --git a/pkg/lang/golang/queries/gocloud/mem_bucket.scm b/pkg/lang/golang/queries/gocloud/mem_bucket.scm