add data stream status and object type (#55)

Author: rsulli-rai

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## v0.1.6-alpha
+* Add Snowflake data stream status support
+* Modify Snowflake data stream create for view support
+
 ## v0.1.5-alpha
 
 * Add Snowflake data stream commands

go.mod

@@ -4,7 +4,7 @@ go 1.18
 
 require (
 	github.com/pkg/errors v0.9.1
-	github.com/relationalai/rai-sdk-go v0.4.5-alpha
+	github.com/relationalai/rai-sdk-go v0.4.7-alpha
 	github.com/spf13/cobra v1.5.0
 )
 

go.sum

@@ -277,8 +277,8 @@ github.com/prometheus/procfs v0.0.0-20190117184657-bf6a532e95b1/go.mod h1:c3At6R
 github.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=
 github.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=
 github.com/rcrowley/go-metrics v0.0.0-20181016184325-3113b8401b8a/go.mod h1:bCqnVzQkZxMG4s8nGwiZ5l3QUCyqpo9Y+/ZMZ9VjZe4=
-github.com/relationalai/rai-sdk-go v0.4.5-alpha h1:+F7Z2K7H2q8S1x4Y0P7baz9GyCGOlTIVRmnacwmo228=
-github.com/relationalai/rai-sdk-go v0.4.5-alpha/go.mod h1:qfIZ7OiUpM+ZLPB8m358DFQPRPFzMjM4ghX4iKR8UHo=
+github.com/relationalai/rai-sdk-go v0.4.7-alpha h1:9Jbl+01nLEwar7eJGfZOkuKEpJaO7oRLCYHSwijl1z4=
+github.com/relationalai/rai-sdk-go v0.4.7-alpha/go.mod h1:qfIZ7OiUpM+ZLPB8m358DFQPRPFzMjM4ghX4iKR8UHo=
 github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=
 github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=
 github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=

rai/cmds.go

@@ -895,13 +895,13 @@ func createSnowflakeDataStream(cmd *cobra.Command, args []string) {
 	warehouse := action.getStringEnv("warehouse", "SNOWSQL_WAREHOUSE")
 	username := action.getStringEnv("username", "SNOWSQL_USER")
 	password := action.getStringEnv("password", "SNOWSQL_PWD")
-	isView := action.getBool("is-view")
+	objectType := action.getString("object-type")
 	raiDatabase := action.getString("rai-database")
 	relation := action.getString("rai-relation")
 	creds := &rai.SnowflakeCredentials{Username: username, Password: password}
 
 	opts := &rai.DataStreamOpts{
-		IsView:      isView,
+		ObjectType:  objectType,
 		RaiDatabase: raiDatabase,
 		Relation:    relation,
 		ObjectName:  dataStream,
@@ -948,6 +948,16 @@ func listSnowflakeDataStreams(cmd *cobra.Command, args []string) {
 	action.Exit(rsp, err)
 }
 
+func getSnowflakeDataStreamStatus(cmd *cobra.Command, args []string) {
+	action := newAction(cmd)
+	integration := args[0]
+	dbLink := args[1]
+	dataStream := args[2]
+	action.Start("Get Snowflake data stream status %s (%s)", dataStream, integration)
+	rsp, err := action.Client().GetSnowflakeDataStreamStatus(integration, dbLink, dataStream)
+	action.Exit(rsp, err)
+}
+
 //
 // Misc
 //

rai/main.go

@@ -379,7 +379,7 @@ func addCommands(root *cobra.Command) {
 	cmd.Flags().String("warehouse", "", "Snowflake warehouse (default: SNOWSQL_WAREHOUSE env var)")
 	cmd.Flags().String("username", "", "Snowflake username (default: SNOWSQL_USER env var)")
 	cmd.Flags().String("password", "", "Snowflake password (default: SNOWSQL_PWD env var)")
-	cmd.Flags().Bool("is-view", false, "Snowflake object is a view (default: false)")
+	cmd.Flags().String("object-type", "", "Snowflake object type (default: table)")
 	cmd.Flags().String("rai-database", "", "RelationalAI target database name")
 	cmd.MarkFlagRequired("rai-database")
 	cmd.Flags().String("rai-relation", "", "RelationalAI target relation")
@@ -411,6 +411,13 @@ func addCommands(root *cobra.Command) {
 		Run:   listSnowflakeDataStreams}
 	root.AddCommand(cmd)
 
+	cmd = &cobra.Command{
+		Use:   "get-snowflake-data-stream-status integration database-link objectName",
+		Short: "Get status information about a Snowflake data stream associated with an integration and database-link",
+		Args:  cobra.ExactArgs(3),
+		Run:   getSnowflakeDataStreamStatus}
+	root.AddCommand(cmd)
+
 	// Misc
 	cmd = &cobra.Command{
 		Use:   "get-access-token",

vendor/github.com/relationalai/rai-sdk-go/rai/client.go

@@ -49,7 +49,7 @@ github.com/pierrec/lz4/v4/internal/xxh32
 # github.com/pkg/errors v0.9.1
 ## explicit
 github.com/pkg/errors
-# github.com/relationalai/rai-sdk-go v0.4.5-alpha
+# github.com/relationalai/rai-sdk-go v0.4.7-alpha
 ## explicit; go 1.18
 github.com/relationalai/rai-sdk-go/rai
 github.com/relationalai/rai-sdk-go/rai/pb