demo fudanet kolloquium

cmd/get.go

@@ -46,7 +46,14 @@ Cobra is a CLI library for Go that empowers applications.
 This application is a tool to generate the needed files
 to quickly create a Cobra application.`,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		return gnmidemo.Get(args[0])
+		flags := cmd.Flags()
+		u, err := flags.GetString("username")
+		p, err := flags.GetString("password")
+		a, err := flags.GetString("address")
+		if err != nil {
+			return err
+		}
+		return gnmidemo.Get(args[0], a, u, p)
 	},
 }
 

cmd/root.go

@@ -69,7 +69,7 @@ func init() {
 	cobra.OnInitialize(initConfig)
 
 	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.gnmi-demo.yaml)")
-	rootCmd.PersistentFlags().StringP("address", "a", "arista.kieweg.dev", "address of the endpiont")
+	rootCmd.PersistentFlags().StringP("address", "a", "arista.kieweg.dev:6030", "address of the endpiont")
 	rootCmd.PersistentFlags().StringP("username", "u", "admin", "username for the user")
 	rootCmd.PersistentFlags().StringP("password", "p", "arista", "password of the user")
 }

cmd/set.go

@@ -31,8 +31,7 @@ POSSIBILITY OF SUCH DAMAGE.
 package cmd
 
 import (
-	"fmt"
-
+	gnmidemo "code.fbi.h-da.de/danet/gnmi-demo"
 	"github.com/spf13/cobra"
 )
 
@@ -46,8 +45,15 @@ and usage of using your command. For example:
 Cobra is a CLI library for Go that empowers applications.
 This application is a tool to generate the needed files
 to quickly create a Cobra application.`,
-	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("set called")
+	RunE: func(cmd *cobra.Command, args []string) error {
+		flags := cmd.Flags()
+		u, err := flags.GetString("username")
+		p, err := flags.GetString("password")
+		a, err := flags.GetString("address")
+		if err != nil {
+			return err
+		}
+		return gnmidemo.Set(args[0], a, u, p)
 	},
 }
 

cmd/subscribe.go

@@ -31,8 +31,7 @@ POSSIBILITY OF SUCH DAMAGE.
 package cmd
 
 import (
-	"fmt"
-
+	gnmidemo "code.fbi.h-da.de/danet/gnmi-demo"
 	"github.com/spf13/cobra"
 )
 
@@ -46,8 +45,15 @@ and usage of using your command. For example:
 Cobra is a CLI library for Go that empowers applications.
 This application is a tool to generate the needed files
 to quickly create a Cobra application.`,
-	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("subscribe called")
+	RunE: func(cmd *cobra.Command, args []string) error {
+		flags := cmd.Flags()
+		u, err := flags.GetString("username")
+		p, err := flags.GetString("password")
+		a, err := flags.GetString("address")
+		if err != nil {
+			return err
+		}
+		return gnmidemo.Subscribe(args[0], a, u, p)
 	},
 }
 

demo.go

@@ -66,11 +66,20 @@ func Get(params ...string) error {
 		if err := util.Write(resp.(proto.Message), "device"); err != nil {
 			return err
 		}
+		log.Info("wrote root resource to file")
 	default:
 		device := &openconfig.Device{}
+		blank := &gpb.GetResponse{}
+		if err := util.Read("device", blank); err != nil {
+			return err
+		}
+		if err := transport.ProcessResponse(blank, device, southbound.Schema()); err != nil {
+			return err
+		}
 		if err := transport.ProcessResponse(resp, device, southbound.Schema()); err != nil {
 			return err
 		}
+		log.Infof("response: %v", resp)
 	}
 	return nil
 }