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/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 {