scope collections to the workspace

rockset · Oct 16, 2023 · d4d06c8 · d4d06c8
1 parent 8f5f011
commit d4d06c8
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmd/completions.go b/cmd/completions.go
@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"github.com/rockset/rockset-go-client/option"
 	"github.com/spf13/cobra"
 )
 
@@ -10,7 +11,12 @@ func collectionCompletion(cmd *cobra.Command, args []string, toComplete string)
 		return nil, cobra.ShellCompDirectiveError
 	}
 
-	collections, err := rs.ListCollections(cmd.Context())
+	var options []option.ListCollectionOption
+	if ws, _ := cmd.Flags().GetString(WorkspaceFlag); ws != "" {
+		options = append(options, option.WithWorkspace(ws))
+	}
+
+	collections, err := rs.ListCollections(cmd.Context(), options...)
 	if err != nil {
 		return nil, cobra.ShellCompDirectiveError
 	}