feat(cmd): add a flag to the root command called limit as an option f…

…or the user to set up the limit value 86
danvergara · Dec 22, 2021 · dd2c852 · dd2c852
1 parent 0b183c4
commit dd2c852
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/.dblab.yaml b/.dblab.yaml
@@ -5,3 +5,4 @@ database:
   password: "password"
   user: "postgres"
   driver: "postgres"
+limit: 50
diff --git a/cmd/root.go b/cmd/root.go
@@ -21,6 +21,7 @@ var (
 	pass   string
 	db     string
 	ssl    string
+	limit  int
 )
 
 // NewRootCmd returns the root command.
@@ -48,6 +49,7 @@ func NewRootCmd() *cobra.Command {
 					Pass:   pass,
 					DBName: db,
 					SSL:    ssl,
+					Limit:  limit,
 				}
 
 				if form.IsEmpty(opts) {
@@ -106,4 +108,5 @@ func init() {
 	rootCmd.Flags().StringVarP(&pass, "pass", "", "", "Password for user")
 	rootCmd.Flags().StringVarP(&db, "db", "", "", "Database name")
 	rootCmd.Flags().StringVarP(&ssl, "ssl", "", "", "SSL mode")
+	rootCmd.Flags().IntVarP(&limit, "limit", "", 100, "Size of the result set from the table content query")
 }