Add limit to rows for insert command

Signed-off-by: Shaad7 <[email protected]>

pkg/data/const.go

@@ -25,3 +25,7 @@ const (
 const (
 	actor = "kubedb-cli"
 )
+
+const (
+	maxRows = 100000
+)

pkg/data/elasticsearch.go

@@ -90,10 +90,9 @@ func InsertElasticsearchDataCMD(f cmdutil.Factory) *cobra.Command {
 				log.Fatalln(err)
 			}
 
-			if rows <= 0 {
-				log.Fatal("rows need to be greater than 0")
+			if rows < 1 || rows > maxRows {
+				log.Fatalf("rows need to be between 1 and %d", maxRows)
 			}
-
 			err = opts.insertDataInDatabase(rows)
 			if err != nil {
 				log.Fatal(err)

pkg/data/mariadb.go

@@ -71,17 +71,16 @@ func InsertMariaDBDataCMD(f cmdutil.Factory) *cobra.Command {
 				log.Fatalln(err)
 			}
 
-			if rows <= 0 {
-				log.Fatal("Inserted rows must be greater than 0")
-			}
-
 			tunnel, err := lib.TunnelToDBService(opts.config, dbName, namespace, api.MySQLDatabasePort)
 			if err != nil {
 				log.Fatal("couldn't create tunnel, error: ", err)
 			}
 
 			defer tunnel.Close()
 
+			if rows < 1 || rows > maxRows {
+				log.Fatalf("rows need to be between 1 and %d", maxRows)
+			}
 			err = opts.insertDataExecCmd(tunnel, rows)
 			if err != nil {
 				log.Fatal(err)

pkg/data/mongodb.go

@@ -76,8 +76,8 @@ func InsertMongoDBDataCMD(f cmdutil.Factory) *cobra.Command {
 				log.Fatalln(err)
 			}
 
-			if rows <= 0 {
-				log.Fatal("rows need to be greater than 0")
+			if rows < 1 || rows > maxRows {
+				log.Fatalf("rows need to be between 1 and %d", maxRows)
 			}
 
 			command := fmt.Sprintf("for(var i=1;i<=%d;i++){db[\"%s\"].insert({_id:\"doc\"+i,actor:\"%s\"})}", rows, mgCollectionName, actor)

pkg/data/mysql.go

@@ -81,7 +81,9 @@ func InsertMySQLDataCMD(f cmdutil.Factory) *cobra.Command {
 			}
 
 			defer tunnel.Close()
-
+			if rows < 1 || rows > maxRows {
+				log.Fatalf("rows need to be between 1 and %d", maxRows)
+			}
 			err = opts.insertDataExecCmd(tunnel, rows)
 			if err != nil {
 				log.Fatal(err)

pkg/data/postgres.go

@@ -84,6 +84,9 @@ func InsertPostgresDataCMD(f cmdutil.Factory) *cobra.Command {
 				log.Fatal("couldn't create tunnel, error: ", err)
 			}
 			defer tunnel.Close()
+			if rows < 1 || rows > maxRows {
+				log.Fatalf("rows need to be between 1 and %d", maxRows)
+			}
 			err = opts.insertDataExecCmd(tunnel, rows)
 			if err != nil {
 				log.Fatal(err)
@@ -97,10 +100,6 @@ func InsertPostgresDataCMD(f cmdutil.Factory) *cobra.Command {
 }
 
 func (opts *postgresOpts) insertDataExecCmd(tunnel *portforward.Tunnel, rows int) error {
-	if rows <= 0 {
-		return fmt.Errorf("rows need to be greater than 0")
-	}
-
 	command := `
 	DO $$ BEGIN
 		IF NOT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_schema = 'public' AND table_name = 'appscode_kubedb_postgres_test_table') THEN

pkg/data/redis.go

@@ -115,6 +115,9 @@ func InsertRedisDataCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			if rows < 1 || rows > maxRows {
+				log.Fatalf("rows need to be between 1 and %d", maxRows)
+			}
 
 			err = opts.insertDataInDatabase(rows)
 			if err != nil {